初始化
This commit is contained in:
3
.gitignore
vendored
Normal file
3
.gitignore
vendored
Normal file
@@ -0,0 +1,3 @@
|
||||
composer.lock
|
||||
vendor/
|
||||
.idea/
|
||||
201
LICENSE
Normal file
201
LICENSE
Normal file
@@ -0,0 +1,201 @@
|
||||
Apache License
|
||||
Version 2.0, January 2004
|
||||
http://www.apache.org/licenses/
|
||||
|
||||
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
|
||||
|
||||
1. Definitions.
|
||||
|
||||
"License" shall mean the terms and conditions for use, reproduction,
|
||||
and distribution as defined by Sections 1 through 9 of this document.
|
||||
|
||||
"Licensor" shall mean the copyright owner or entity authorized by
|
||||
the copyright owner that is granting the License.
|
||||
|
||||
"Legal Entity" shall mean the union of the acting entity and all
|
||||
other entities that control, are controlled by, or are under common
|
||||
control with that entity. For the purposes of this definition,
|
||||
"control" means (i) the power, direct or indirect, to cause the
|
||||
direction or management of such entity, whether by contract or
|
||||
otherwise, or (ii) ownership of fifty percent (50%) or more of the
|
||||
outstanding shares, or (iii) beneficial ownership of such entity.
|
||||
|
||||
"You" (or "Your") shall mean an individual or Legal Entity
|
||||
exercising permissions granted by this License.
|
||||
|
||||
"Source" form shall mean the preferred form for making modifications,
|
||||
including but not limited to software source code, documentation
|
||||
source, and configuration files.
|
||||
|
||||
"Object" form shall mean any form resulting from mechanical
|
||||
transformation or translation of a Source form, including but
|
||||
not limited to compiled object code, generated documentation,
|
||||
and conversions to other media types.
|
||||
|
||||
"Work" shall mean the work of authorship, whether in Source or
|
||||
Object form, made available under the License, as indicated by a
|
||||
copyright notice that is included in or attached to the work
|
||||
(an example is provided in the Appendix below).
|
||||
|
||||
"Derivative Works" shall mean any work, whether in Source or Object
|
||||
form, that is based on (or derived from) the Work and for which the
|
||||
editorial revisions, annotations, elaborations, or other modifications
|
||||
represent, as a whole, an original work of authorship. For the purposes
|
||||
of this License, Derivative Works shall not include works that remain
|
||||
separable from, or merely link (or bind by name) to the interfaces of,
|
||||
the Work and Derivative Works thereof.
|
||||
|
||||
"Contribution" shall mean any work of authorship, including
|
||||
the original version of the Work and any modifications or additions
|
||||
to that Work or Derivative Works thereof, that is intentionally
|
||||
submitted to Licensor for inclusion in the Work by the copyright owner
|
||||
or by an individual or Legal Entity authorized to submit on behalf of
|
||||
the copyright owner. For the purposes of this definition, "submitted"
|
||||
means any form of electronic, verbal, or written communication sent
|
||||
to the Licensor or its representatives, including but not limited to
|
||||
communication on electronic mailing lists, source code control systems,
|
||||
and issue tracking systems that are managed by, or on behalf of, the
|
||||
Licensor for the purpose of discussing and improving the Work, but
|
||||
excluding communication that is conspicuously marked or otherwise
|
||||
designated in writing by the copyright owner as "Not a Contribution."
|
||||
|
||||
"Contributor" shall mean Licensor and any individual or Legal Entity
|
||||
on behalf of whom a Contribution has been received by Licensor and
|
||||
subsequently incorporated within the Work.
|
||||
|
||||
2. Grant of Copyright License. Subject to the terms and conditions of
|
||||
this License, each Contributor hereby grants to You a perpetual,
|
||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||
copyright license to reproduce, prepare Derivative Works of,
|
||||
publicly display, publicly perform, sublicense, and distribute the
|
||||
Work and such Derivative Works in Source or Object form.
|
||||
|
||||
3. Grant of Patent License. Subject to the terms and conditions of
|
||||
this License, each Contributor hereby grants to You a perpetual,
|
||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||
(except as stated in this section) patent license to make, have made,
|
||||
use, offer to sell, sell, import, and otherwise transfer the Work,
|
||||
where such license applies only to those patent claims licensable
|
||||
by such Contributor that are necessarily infringed by their
|
||||
Contribution(s) alone or by combination of their Contribution(s)
|
||||
with the Work to which such Contribution(s) was submitted. If You
|
||||
institute patent litigation against any entity (including a
|
||||
cross-claim or counterclaim in a lawsuit) alleging that the Work
|
||||
or a Contribution incorporated within the Work constitutes direct
|
||||
or contributory patent infringement, then any patent licenses
|
||||
granted to You under this License for that Work shall terminate
|
||||
as of the date such litigation is filed.
|
||||
|
||||
4. Redistribution. You may reproduce and distribute copies of the
|
||||
Work or Derivative Works thereof in any medium, with or without
|
||||
modifications, and in Source or Object form, provided that You
|
||||
meet the following conditions:
|
||||
|
||||
(a) You must give any other recipients of the Work or
|
||||
Derivative Works a copy of this License; and
|
||||
|
||||
(b) You must cause any modified files to carry prominent notices
|
||||
stating that You changed the files; and
|
||||
|
||||
(c) You must retain, in the Source form of any Derivative Works
|
||||
that You distribute, all copyright, patent, trademark, and
|
||||
attribution notices from the Source form of the Work,
|
||||
excluding those notices that do not pertain to any part of
|
||||
the Derivative Works; and
|
||||
|
||||
(d) If the Work includes a "NOTICE" text file as part of its
|
||||
distribution, then any Derivative Works that You distribute must
|
||||
include a readable copy of the attribution notices contained
|
||||
within such NOTICE file, excluding those notices that do not
|
||||
pertain to any part of the Derivative Works, in at least one
|
||||
of the following places: within a NOTICE text file distributed
|
||||
as part of the Derivative Works; within the Source form or
|
||||
documentation, if provided along with the Derivative Works; or,
|
||||
within a display generated by the Derivative Works, if and
|
||||
wherever such third-party notices normally appear. The contents
|
||||
of the NOTICE file are for informational purposes only and
|
||||
do not modify the License. You may add Your own attribution
|
||||
notices within Derivative Works that You distribute, alongside
|
||||
or as an addendum to the NOTICE text from the Work, provided
|
||||
that such additional attribution notices cannot be construed
|
||||
as modifying the License.
|
||||
|
||||
You may add Your own copyright statement to Your modifications and
|
||||
may provide additional or different license terms and conditions
|
||||
for use, reproduction, or distribution of Your modifications, or
|
||||
for any such Derivative Works as a whole, provided Your use,
|
||||
reproduction, and distribution of the Work otherwise complies with
|
||||
the conditions stated in this License.
|
||||
|
||||
5. Submission of Contributions. Unless You explicitly state otherwise,
|
||||
any Contribution intentionally submitted for inclusion in the Work
|
||||
by You to the Licensor shall be under the terms and conditions of
|
||||
this License, without any additional terms or conditions.
|
||||
Notwithstanding the above, nothing herein shall supersede or modify
|
||||
the terms of any separate license agreement you may have executed
|
||||
with Licensor regarding such Contributions.
|
||||
|
||||
6. Trademarks. This License does not grant permission to use the trade
|
||||
names, trademarks, service marks, or product names of the Licensor,
|
||||
except as required for reasonable and customary use in describing the
|
||||
origin of the Work and reproducing the content of the NOTICE file.
|
||||
|
||||
7. Disclaimer of Warranty. Unless required by applicable law or
|
||||
agreed to in writing, Licensor provides the Work (and each
|
||||
Contributor provides its Contributions) on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
||||
implied, including, without limitation, any warranties or conditions
|
||||
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
|
||||
PARTICULAR PURPOSE. You are solely responsible for determining the
|
||||
appropriateness of using or redistributing the Work and assume any
|
||||
risks associated with Your exercise of permissions under this License.
|
||||
|
||||
8. Limitation of Liability. In no event and under no legal theory,
|
||||
whether in tort (including negligence), contract, or otherwise,
|
||||
unless required by applicable law (such as deliberate and grossly
|
||||
negligent acts) or agreed to in writing, shall any Contributor be
|
||||
liable to You for damages, including any direct, indirect, special,
|
||||
incidental, or consequential damages of any character arising as a
|
||||
result of this License or out of the use or inability to use the
|
||||
Work (including but not limited to damages for loss of goodwill,
|
||||
work stoppage, computer failure or malfunction, or any and all
|
||||
other commercial damages or losses), even if such Contributor
|
||||
has been advised of the possibility of such damages.
|
||||
|
||||
9. Accepting Warranty or Additional Liability. While redistributing
|
||||
the Work or Derivative Works thereof, You may choose to offer,
|
||||
and charge a fee for, acceptance of support, warranty, indemnity,
|
||||
or other liability obligations and/or rights consistent with this
|
||||
License. However, in accepting such obligations, You may act only
|
||||
on Your own behalf and on Your sole responsibility, not on behalf
|
||||
of any other Contributor, and only if You agree to indemnify,
|
||||
defend, and hold each Contributor harmless for any liability
|
||||
incurred by, or claims asserted against, such Contributor by reason
|
||||
of your accepting any such warranty or additional liability.
|
||||
|
||||
END OF TERMS AND CONDITIONS
|
||||
|
||||
APPENDIX: How to apply the Apache License to your work.
|
||||
|
||||
To apply the Apache License to your work, attach the following
|
||||
boilerplate notice, with the fields enclosed by brackets "{}"
|
||||
replaced with your own identifying information. (Don't include
|
||||
the brackets!) The text should be enclosed in the appropriate
|
||||
comment syntax for the file format. We also recommend that a
|
||||
file or class name and description of purpose be included on the
|
||||
same "printed page" as the copyright notice for easier
|
||||
identification within third-party archives.
|
||||
|
||||
Copyright {yyyy} {name of copyright owner}
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
Unless required by applicable law or agreed to in writing, software
|
||||
distributed under the License is distributed on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
139
README.md
139
README.md
@@ -1,2 +1,139 @@
|
||||
# think-worker
|
||||
ThinkPHP Workerman 扩展
|
||||
===============
|
||||
|
||||
交流群：981069000 [](https://qm.qq.com/q/A8YNpzrzC8)
|
||||
|
||||
## 安装
|
||||
```
|
||||
composer require topthink/think-worker
|
||||
```
|
||||
|
||||
## 说明
|
||||
> 由于windows下无法在一个文件里启动多个worker，所以本扩展不支持windows平台
|
||||
|
||||
## 使用方法
|
||||
|
||||
### HttpServer
|
||||
|
||||
在命令行启动服务端
|
||||
~~~
|
||||
php think worker
|
||||
~~~
|
||||
|
||||
然后就可以通过浏览器直接访问当前应用
|
||||
|
||||
~~~
|
||||
http://localhost:8080
|
||||
~~~
|
||||
|
||||
如果需要使用守护进程方式运行，建议使用supervisor来管理进程
|
||||
|
||||
## 访问静态文件
|
||||
> 建议使用nginx来支持静态文件访问，也可使用路由输出文件内容，下面是示例，可参照修改
|
||||
1. 添加静态文件路由：
|
||||
|
||||
```php
|
||||
Route::get('static/:path', function (string $path) {
|
||||
$filename = public_path() . $path;
|
||||
return new \think\worker\response\File($filename);
|
||||
})->pattern(['path' => '.*\.\w+$']);
|
||||
```
|
||||
|
||||
2. 访问路由 `http://localhost/static/文件路径`
|
||||
|
||||
## 队列支持
|
||||
|
||||
使用方法见 [think-queue](https://github.com/top-think/think-queue)
|
||||
|
||||
以下配置代替think-queue里的最后一步:`监听任务并执行`,无需另外起进程执行队列
|
||||
|
||||
```php
|
||||
return [
|
||||
// ...
|
||||
'queue' => [
|
||||
'enable' => true,
|
||||
//键名是队列名称
|
||||
'workers' => [
|
||||
//下面参数是不设置时的默认配置
|
||||
'default' => [
|
||||
'delay' => 0,
|
||||
'sleep' => 3,
|
||||
'tries' => 0,
|
||||
'timeout' => 60,
|
||||
'worker_num' => 1,
|
||||
],
|
||||
//使用@符号后面可指定队列使用驱动
|
||||
'default@connection' => [
|
||||
//此处可不设置任何参数，使用上面的默认配置
|
||||
],
|
||||
],
|
||||
],
|
||||
// ...
|
||||
];
|
||||
|
||||
```
|
||||
|
||||
### websocket
|
||||
|
||||
> 使用路由调度的方式，可以让不同路径的websocket服务响应不同的事件
|
||||
|
||||
#### 配置
|
||||
|
||||
```
|
||||
worker.websocket = true 时开启
|
||||
```
|
||||
|
||||
#### 路由定义
|
||||
```php
|
||||
Route::get('path1','controller/action1');
|
||||
Route::get('path2','controller/action2');
|
||||
```
|
||||
|
||||
#### 控制器
|
||||
|
||||
```php
|
||||
use \think\worker\Websocket;
|
||||
use \think\worker\websocket\Frame;
|
||||
|
||||
class Controller {
|
||||
|
||||
public function action1(){
|
||||
|
||||
return (new \think\worker\response\Websocket())
|
||||
->onOpen(...)
|
||||
->onMessage(function(Websocket $websocket, Frame $frame){
|
||||
...
|
||||
})
|
||||
->onClose(...);
|
||||
}
|
||||
|
||||
public function action2(){
|
||||
|
||||
return (new \think\worker\response\Websocket())
|
||||
->onOpen(...)
|
||||
->onMessage(function(Websocket $websocket, Frame $frame){
|
||||
...
|
||||
})
|
||||
->onClose(...);
|
||||
}
|
||||
}
|
||||
```
|
||||
|
||||
|
||||
## 自定义worker
|
||||
监听`worker.init`事件 注入`Manager`对象，调用addWorker方法添加
|
||||
~~~php
|
||||
use think\worker\Manager;
|
||||
use \think\worker\Worker;
|
||||
|
||||
//...
|
||||
|
||||
public function handle(Manager $manager){
|
||||
$worker = $manager->addWorker(function(Worker $worker){
|
||||
//..其他回调或处理
|
||||
//动态添加监听可参考 https://www.workerman.net/doc/workerman/worker/listen.html
|
||||
});
|
||||
}
|
||||
|
||||
//...
|
||||
~~~
|
||||
|
||||
55
composer.json
Normal file
55
composer.json
Normal file
@@ -0,0 +1,55 @@
|
||||
{
|
||||
"name": "topthink/think-worker",
|
||||
"description": "workerman extend for thinkphp",
|
||||
"license": "Apache-2.0",
|
||||
"authors": [
|
||||
{
|
||||
"name": "liu21st",
|
||||
"email": "liu21st@gmail.com"
|
||||
}
|
||||
],
|
||||
"require": {
|
||||
"php": ">=8.2",
|
||||
"workerman/workerman": "~5.0.0",
|
||||
"topthink/framework": "^8.0",
|
||||
"revolt/event-loop": "^1.0",
|
||||
"workerman/redis": "^2.0",
|
||||
"symfony/finder": ">=4.3"
|
||||
},
|
||||
"autoload": {
|
||||
"psr-4": {
|
||||
"think\\worker\\": "src"
|
||||
}
|
||||
},
|
||||
"extra": {
|
||||
"think": {
|
||||
"services": [
|
||||
"think\\worker\\Service"
|
||||
],
|
||||
"config": {
|
||||
"worker": "src/config/worker.php"
|
||||
}
|
||||
}
|
||||
},
|
||||
"require-dev": {
|
||||
"pestphp/pest": "^3.7",
|
||||
"guzzlehttp/guzzle": "^7.0",
|
||||
"topthink/think-queue": "^3.0",
|
||||
"phpstan/phpstan": "^2.0",
|
||||
"ratchet/pawl": "^0.4.1"
|
||||
},
|
||||
"autoload-dev": {
|
||||
"psr-4": {
|
||||
"app\\": "tests/stub/app"
|
||||
}
|
||||
},
|
||||
"config": {
|
||||
"allow-plugins": {
|
||||
"pestphp/pest-plugin": true
|
||||
}
|
||||
},
|
||||
"scripts": {
|
||||
"analyze": "phpstan --memory-limit=1G",
|
||||
"test": "pest --colors=always"
|
||||
}
|
||||
}
|
||||
16
phpstan.neon
Normal file
16
phpstan.neon
Normal file
@@ -0,0 +1,16 @@
|
||||
parameters:
|
||||
level: 5
|
||||
paths:
|
||||
- src
|
||||
- tests
|
||||
universalObjectCratesClasses:
|
||||
- PHPUnit\Framework\TestCase
|
||||
ignoreErrors:
|
||||
- '#Function root_path not found.#'
|
||||
- '#Function env not found.#'
|
||||
- '#Function app_path not found.#'
|
||||
- '#Function config_path not found.#'
|
||||
- '#Function public_path not found.#'
|
||||
- '#Function json not found.#'
|
||||
- '#Function runtime_path not found.#'
|
||||
- '#Constant STUB_DIR not found.#'
|
||||
18
phpunit.xml
Normal file
18
phpunit.xml
Normal file
@@ -0,0 +1,18 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<phpunit xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
|
||||
xsi:noNamespaceSchemaLocation="https://schema.phpunit.de/10.1/phpunit.xsd"
|
||||
bootstrap="vendor/autoload.php"
|
||||
colors="true"
|
||||
>
|
||||
<testsuites>
|
||||
<testsuite name="Test Suite">
|
||||
<directory suffix="Test.php">./tests</directory>
|
||||
</testsuite>
|
||||
</testsuites>
|
||||
<coverage/>
|
||||
<source>
|
||||
<include>
|
||||
<directory suffix=".php">./src</directory>
|
||||
</include>
|
||||
</source>
|
||||
</phpunit>
|
||||
23
src/App.php
Normal file
23
src/App.php
Normal file
@@ -0,0 +1,23 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker;
|
||||
|
||||
class App extends \think\App
|
||||
{
|
||||
protected $inConsole = true;
|
||||
|
||||
public function setInConsole($inConsole = true)
|
||||
{
|
||||
$this->inConsole = $inConsole;
|
||||
}
|
||||
|
||||
public function runningInConsole(): bool
|
||||
{
|
||||
return $this->inConsole;
|
||||
}
|
||||
|
||||
public function clearInstances()
|
||||
{
|
||||
$this->instances = [];
|
||||
}
|
||||
}
|
||||
22
src/Conduit.php
Normal file
22
src/Conduit.php
Normal file
@@ -0,0 +1,22 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker;
|
||||
|
||||
/**
|
||||
* @mixin \think\worker\conduit\Driver
|
||||
*/
|
||||
class Conduit extends \think\Manager
|
||||
{
|
||||
|
||||
protected $namespace = "\\think\\worker\\conduit\\driver\\";
|
||||
|
||||
protected function resolveConfig(string $name)
|
||||
{
|
||||
return $this->app->config->get("worker.conduit.{$name}", []);
|
||||
}
|
||||
|
||||
public function getDefaultDriver()
|
||||
{
|
||||
return $this->app->config->get('worker.conduit.type', 'socket');
|
||||
}
|
||||
}
|
||||
17
src/Http.php
Normal file
17
src/Http.php
Normal file
@@ -0,0 +1,17 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker;
|
||||
|
||||
class Http extends \think\Http
|
||||
{
|
||||
|
||||
protected function loadMiddleware(): void
|
||||
{
|
||||
|
||||
}
|
||||
|
||||
protected function loadRoutes(): void
|
||||
{
|
||||
|
||||
}
|
||||
}
|
||||
41
src/Ipc.php
Normal file
41
src/Ipc.php
Normal file
@@ -0,0 +1,41 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker;
|
||||
|
||||
class Ipc
|
||||
{
|
||||
protected $workerId;
|
||||
|
||||
public function __construct(protected Manager $manager, protected Conduit $conduit)
|
||||
{
|
||||
|
||||
}
|
||||
|
||||
public function listenMessage()
|
||||
{
|
||||
$this->subscribe();
|
||||
return $this->workerId;
|
||||
}
|
||||
|
||||
public function sendMessage($workerId, $message)
|
||||
{
|
||||
if ($workerId === $this->workerId) {
|
||||
$this->manager->triggerEvent('message', $message);
|
||||
} else {
|
||||
$this->publish($workerId, $message);
|
||||
}
|
||||
}
|
||||
|
||||
public function subscribe()
|
||||
{
|
||||
$this->workerId = $this->conduit->inc('ipc:worker');
|
||||
$this->conduit->subscribe("ipc:message:{$this->workerId}", function ($message) {
|
||||
$this->manager->triggerEvent('message', unserialize($message));
|
||||
});
|
||||
}
|
||||
|
||||
public function publish($workerId, $message)
|
||||
{
|
||||
$this->conduit->publish("ipc:message:{$workerId}", serialize($message));
|
||||
}
|
||||
}
|
||||
27
src/Manager.php
Normal file
27
src/Manager.php
Normal file
@@ -0,0 +1,27 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker;
|
||||
|
||||
use think\worker\concerns\InteractsWithConduit;
|
||||
use think\worker\concerns\InteractsWithHttp;
|
||||
use think\worker\concerns\InteractsWithQueue;
|
||||
use think\worker\concerns\InteractsWithServer;
|
||||
use think\worker\concerns\WithApplication;
|
||||
use think\worker\concerns\WithContainer;
|
||||
|
||||
class Manager
|
||||
{
|
||||
use InteractsWithServer,
|
||||
InteractsWithHttp,
|
||||
InteractsWithQueue,
|
||||
InteractsWithConduit,
|
||||
WithApplication,
|
||||
WithContainer;
|
||||
|
||||
protected function initialize(): void
|
||||
{
|
||||
$this->prepareHttp();
|
||||
$this->prepareQueue();
|
||||
$this->prepareConduit();
|
||||
}
|
||||
}
|
||||
210
src/Sandbox.php
Normal file
210
src/Sandbox.php
Normal file
@@ -0,0 +1,210 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker;
|
||||
|
||||
use Closure;
|
||||
use InvalidArgumentException;
|
||||
use ReflectionObject;
|
||||
use RuntimeException;
|
||||
use think\Config;
|
||||
use think\Container;
|
||||
use think\Event;
|
||||
use think\exception\Handle;
|
||||
use think\worker\App;
|
||||
use think\worker\concerns\ModifyProperty;
|
||||
use think\worker\contract\ResetterInterface;
|
||||
use think\worker\resetters\ClearInstances;
|
||||
use think\worker\resetters\ResetConfig;
|
||||
use think\worker\resetters\ResetEvent;
|
||||
use think\worker\resetters\ResetModel;
|
||||
use think\worker\resetters\ResetPaginator;
|
||||
use think\worker\resetters\ResetService;
|
||||
use Throwable;
|
||||
use WeakMap;
|
||||
|
||||
class Sandbox
|
||||
{
|
||||
use ModifyProperty;
|
||||
|
||||
/** @var App|null */
|
||||
protected $snapshot;
|
||||
|
||||
/** @var WeakMap */
|
||||
protected $snapshots;
|
||||
|
||||
/** @var App */
|
||||
protected $app;
|
||||
|
||||
/** @var Config */
|
||||
protected $config;
|
||||
|
||||
/** @var Event */
|
||||
protected $event;
|
||||
|
||||
/** @var ResetterInterface[] */
|
||||
protected $resetters = [];
|
||||
protected $services = [];
|
||||
|
||||
public function __construct(App $app)
|
||||
{
|
||||
$this->app = $app;
|
||||
$this->snapshots = new WeakMap();
|
||||
$this->initialize();
|
||||
}
|
||||
|
||||
protected function initialize()
|
||||
{
|
||||
Container::setInstance(function () {
|
||||
return $this->getSnapshot();
|
||||
});
|
||||
|
||||
$this->setInitialConfig();
|
||||
$this->setInitialServices();
|
||||
$this->setInitialEvent();
|
||||
$this->setInitialResetters();
|
||||
}
|
||||
|
||||
public function run(Closure $callable, ?object $key = null)
|
||||
{
|
||||
$this->snapshot = $this->createApp($key);
|
||||
try {
|
||||
$this->snapshot->invoke($callable, [$this]);
|
||||
} catch (Throwable $e) {
|
||||
$this->snapshot->make(Handle::class)->report($e);
|
||||
} finally {
|
||||
if (empty($key)) {
|
||||
$this->snapshot->clearInstances();
|
||||
}
|
||||
$this->snapshot = null;
|
||||
$this->setInstance($this->app);
|
||||
}
|
||||
}
|
||||
|
||||
protected function createApp(?object $key = null)
|
||||
{
|
||||
if (!empty($key)) {
|
||||
if (isset($this->snapshots[$key])) {
|
||||
return $this->snapshots[$key]->app;
|
||||
}
|
||||
}
|
||||
|
||||
$app = clone $this->app;
|
||||
$this->setInstance($app);
|
||||
$this->resetApp($app);
|
||||
|
||||
if (!empty($key)) {
|
||||
$this->snapshots[$key] = new class($app) {
|
||||
public function __construct(public App $app)
|
||||
{
|
||||
}
|
||||
|
||||
public function __destruct()
|
||||
{
|
||||
$this->app->clearInstances();
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
return $app;
|
||||
}
|
||||
|
||||
protected function resetApp(App $app)
|
||||
{
|
||||
foreach ($this->resetters as $resetter) {
|
||||
$resetter->handle($app, $this);
|
||||
}
|
||||
}
|
||||
|
||||
protected function setInstance(App $app)
|
||||
{
|
||||
$app->instance('app', $app);
|
||||
$app->instance(Container::class, $app);
|
||||
|
||||
$reflectObject = new ReflectionObject($app);
|
||||
$reflectProperty = $reflectObject->getProperty('services');
|
||||
$services = $reflectProperty->getValue($app);
|
||||
|
||||
foreach ($services as $service) {
|
||||
$this->modifyProperty($service, $app);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Set initial config.
|
||||
*/
|
||||
protected function setInitialConfig()
|
||||
{
|
||||
$this->config = clone $this->app->config;
|
||||
}
|
||||
|
||||
protected function setInitialEvent()
|
||||
{
|
||||
$this->event = clone $this->app->event;
|
||||
}
|
||||
|
||||
protected function setInitialServices()
|
||||
{
|
||||
$services = $this->config->get('worker.services', []);
|
||||
|
||||
foreach ($services as $service) {
|
||||
if (class_exists($service) && !in_array($service, $this->services)) {
|
||||
$serviceObj = new $service($this->app);
|
||||
$this->services[$service] = $serviceObj;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Initialize resetters.
|
||||
*/
|
||||
protected function setInitialResetters()
|
||||
{
|
||||
$resetters = [
|
||||
ClearInstances::class,
|
||||
ResetConfig::class,
|
||||
ResetEvent::class,
|
||||
ResetService::class,
|
||||
ResetModel::class,
|
||||
ResetPaginator::class,
|
||||
];
|
||||
|
||||
$resetters = array_merge($resetters, $this->config->get('worker.resetters', []));
|
||||
|
||||
foreach ($resetters as $resetter) {
|
||||
$resetterClass = $this->app->make($resetter);
|
||||
if (!$resetterClass instanceof ResetterInterface) {
|
||||
throw new RuntimeException("{$resetter} must implement " . ResetterInterface::class);
|
||||
}
|
||||
$this->resetters[$resetter] = $resetterClass;
|
||||
}
|
||||
}
|
||||
|
||||
public function getSnapshot()
|
||||
{
|
||||
$snapshot = $this->snapshot;
|
||||
if ($snapshot instanceof App) {
|
||||
return $snapshot;
|
||||
}
|
||||
|
||||
throw new InvalidArgumentException('The app object has not been initialized');
|
||||
}
|
||||
|
||||
/**
|
||||
* Get config snapshot.
|
||||
*/
|
||||
public function getConfig()
|
||||
{
|
||||
return $this->config;
|
||||
}
|
||||
|
||||
public function getEvent()
|
||||
{
|
||||
return $this->event;
|
||||
}
|
||||
|
||||
public function getServices()
|
||||
{
|
||||
return $this->services;
|
||||
}
|
||||
|
||||
}
|
||||
23
src/Service.php
Normal file
23
src/Service.php
Normal file
@@ -0,0 +1,23 @@
|
||||
<?php
|
||||
// +----------------------------------------------------------------------
|
||||
// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2006-2018 http://thinkphp.cn All rights reserved.
|
||||
// +----------------------------------------------------------------------
|
||||
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
||||
// +----------------------------------------------------------------------
|
||||
// | Author: liu21st <liu21st@gmail.com>
|
||||
// +----------------------------------------------------------------------
|
||||
namespace think\worker;
|
||||
|
||||
use think\worker\command\Server;
|
||||
|
||||
class Service extends \think\Service
|
||||
{
|
||||
public function boot()
|
||||
{
|
||||
$this->commands([
|
||||
Server::class,
|
||||
]);
|
||||
}
|
||||
}
|
||||
33
src/Watcher.php
Normal file
33
src/Watcher.php
Normal file
@@ -0,0 +1,33 @@
|
||||
<?php
|
||||
namespace think\worker;
|
||||
|
||||
use think\worker\watcher\Driver;
|
||||
|
||||
/**
|
||||
* @mixin Driver
|
||||
*/
|
||||
class Watcher extends \think\Manager
|
||||
{
|
||||
protected $namespace = '\\think\\worker\\watcher\\';
|
||||
|
||||
protected function getConfig(string $name, $default = null)
|
||||
{
|
||||
return $this->app->config->get('worker.hot_update.' . $name, $default);
|
||||
}
|
||||
|
||||
protected function resolveParams($name): array
|
||||
{
|
||||
return [
|
||||
array_filter($this->getConfig('include', []), function ($dir) {
|
||||
return is_dir($dir);
|
||||
}),
|
||||
$this->getConfig('exclude', []),
|
||||
$this->getConfig('name', []),
|
||||
];
|
||||
}
|
||||
|
||||
public function getDefaultDriver()
|
||||
{
|
||||
return $this->getConfig('type', 'scan');
|
||||
}
|
||||
}
|
||||
121
src/Websocket.php
Normal file
121
src/Websocket.php
Normal file
@@ -0,0 +1,121 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker;
|
||||
|
||||
use RuntimeException;
|
||||
use think\Event;
|
||||
use think\worker\websocket\Pusher;
|
||||
use think\worker\websocket\Room;
|
||||
use Workerman\Connection\TcpConnection;
|
||||
|
||||
/**
|
||||
* Class Websocket
|
||||
*/
|
||||
class Websocket
|
||||
{
|
||||
/**
|
||||
* @var \think\App
|
||||
*/
|
||||
protected $app;
|
||||
|
||||
/**
|
||||
* @var Room
|
||||
*/
|
||||
protected $room;
|
||||
|
||||
/**
|
||||
* @var string
|
||||
*/
|
||||
protected $sender;
|
||||
|
||||
/** @var Event */
|
||||
protected $event;
|
||||
|
||||
/** @var TcpConnection|null */
|
||||
protected $connection;
|
||||
|
||||
protected $connected = true;
|
||||
|
||||
/**
|
||||
* Websocket constructor.
|
||||
*
|
||||
* @param \think\App $app
|
||||
* @param Room $room
|
||||
* @param Event $event
|
||||
*/
|
||||
public function __construct(\think\App $app, Room $room, Event $event, ?TcpConnection $connection = null)
|
||||
{
|
||||
$this->app = $app;
|
||||
$this->room = $room;
|
||||
$this->event = $event;
|
||||
$this->connection = $connection;
|
||||
}
|
||||
|
||||
/**
|
||||
* @return Pusher
|
||||
*/
|
||||
protected function makePusher()
|
||||
{
|
||||
return $this->app->invokeClass(Pusher::class);
|
||||
}
|
||||
|
||||
public function to(...$values)
|
||||
{
|
||||
return $this->makePusher()->to(...$values);
|
||||
}
|
||||
|
||||
public function push($data)
|
||||
{
|
||||
$this->makePusher()->to($this->getSender())->push($data);
|
||||
}
|
||||
|
||||
public function emit(string $event, ...$data)
|
||||
{
|
||||
$this->makePusher()->to($this->getSender())->emit($event, ...$data);
|
||||
}
|
||||
|
||||
public function join(...$rooms): self
|
||||
{
|
||||
$this->room->add($this->getSender(), ...$rooms);
|
||||
|
||||
return $this;
|
||||
}
|
||||
|
||||
public function leave(...$rooms): self
|
||||
{
|
||||
$this->room->delete($this->getSender(), ...$rooms);
|
||||
|
||||
return $this;
|
||||
}
|
||||
|
||||
public function setConnected($connected)
|
||||
{
|
||||
$this->connected = $connected;
|
||||
}
|
||||
|
||||
public function isEstablished()
|
||||
{
|
||||
return $this->connected;
|
||||
}
|
||||
|
||||
public function close()
|
||||
{
|
||||
if ($this->connection) {
|
||||
$this->connection->close();
|
||||
}
|
||||
}
|
||||
|
||||
public function setSender(string $fd)
|
||||
{
|
||||
$this->sender = $fd;
|
||||
return $this;
|
||||
}
|
||||
|
||||
public function getSender()
|
||||
{
|
||||
if (empty($this->sender)) {
|
||||
throw new RuntimeException('Cannot use websocket as current client before handshake!');
|
||||
}
|
||||
return $this->sender;
|
||||
}
|
||||
}
|
||||
19
src/Worker.php
Normal file
19
src/Worker.php
Normal file
@@ -0,0 +1,19 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker;
|
||||
|
||||
class Worker extends \Workerman\Worker
|
||||
{
|
||||
protected static function init(): void
|
||||
{
|
||||
static::$pidFile = runtime_path() . 'worker.pid';
|
||||
static::$statusFile = runtime_path() . 'worker.status';
|
||||
static::$logFile = runtime_path() . 'worker.log';
|
||||
parent::init();
|
||||
}
|
||||
|
||||
protected static function parseCommand(): void
|
||||
{
|
||||
|
||||
}
|
||||
}
|
||||
35
src/command/Server.php
Normal file
35
src/command/Server.php
Normal file
@@ -0,0 +1,35 @@
|
||||
<?php
|
||||
// +----------------------------------------------------------------------
|
||||
// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2006-2018 http://thinkphp.cn All rights reserved.
|
||||
// +----------------------------------------------------------------------
|
||||
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
||||
// +----------------------------------------------------------------------
|
||||
// | Author: liu21st <liu21st@gmail.com>
|
||||
// +----------------------------------------------------------------------
|
||||
|
||||
namespace think\worker\command;
|
||||
|
||||
use think\console\Command;
|
||||
use think\worker\Manager;
|
||||
|
||||
/**
|
||||
* Worker Server 命令行类
|
||||
*/
|
||||
class Server extends Command
|
||||
{
|
||||
protected $config = [];
|
||||
|
||||
public function configure()
|
||||
{
|
||||
$this->setName('worker')
|
||||
->setDescription('Workerman Server for ThinkPHP');
|
||||
}
|
||||
|
||||
public function handle(Manager $manager)
|
||||
{
|
||||
$manager->start();
|
||||
}
|
||||
|
||||
}
|
||||
20
src/concerns/InteractsWithConduit.php
Normal file
20
src/concerns/InteractsWithConduit.php
Normal file
@@ -0,0 +1,20 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\concerns;
|
||||
|
||||
use think\worker\Conduit;
|
||||
|
||||
trait InteractsWithConduit
|
||||
{
|
||||
/** @var Conduit */
|
||||
protected $conduit;
|
||||
|
||||
protected function prepareConduit()
|
||||
{
|
||||
$this->conduit = $this->container->make(Conduit::class);
|
||||
$this->conduit->prepare();
|
||||
$this->onEvent('workerStart', function () {
|
||||
$this->app->instance(Conduit::class, $this->conduit);
|
||||
});
|
||||
}
|
||||
}
|
||||
321
src/concerns/InteractsWithHttp.php
Normal file
321
src/concerns/InteractsWithHttp.php
Normal file
@@ -0,0 +1,321 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\concerns;
|
||||
|
||||
use think\App;
|
||||
use think\Cookie;
|
||||
use think\Event;
|
||||
use think\exception\Handle;
|
||||
use think\helper\Str;
|
||||
use think\Http;
|
||||
use think\response\View;
|
||||
use think\worker\App as WorkerApp;
|
||||
use think\worker\Http as WorkerHttp;
|
||||
use think\worker\response\File as FileResponse;
|
||||
use think\worker\response\Iterator as IteratorResponse;
|
||||
use think\worker\websocket\Frame;
|
||||
use think\worker\Worker;
|
||||
use Throwable;
|
||||
use Workerman\Connection\TcpConnection;
|
||||
use Workerman\Protocols\Http\Chunk;
|
||||
use Workerman\Protocols\Http\Request as WorkerRequest;
|
||||
use Workerman\Protocols\Http\Response;
|
||||
use function substr;
|
||||
|
||||
/**
|
||||
* Trait InteractsWithHttp
|
||||
* @property App $app
|
||||
* @property App $container
|
||||
*/
|
||||
trait InteractsWithHttp
|
||||
{
|
||||
use ModifyProperty, InteractsWithWebsocket;
|
||||
|
||||
protected $wsEnable = false;
|
||||
|
||||
protected function prepareHttp()
|
||||
{
|
||||
if ($this->getConfig('http.enable', true)) {
|
||||
|
||||
$this->wsEnable = $this->getConfig('websocket.enable', false);
|
||||
|
||||
if ($this->wsEnable) {
|
||||
$this->prepareWebsocket();
|
||||
}
|
||||
|
||||
$workerNum = $this->getConfig('http.worker_num', 4);
|
||||
$this->addWorker([$this, 'createHttpServer'], 'http server', $workerNum);
|
||||
}
|
||||
}
|
||||
|
||||
public function createHttpServer()
|
||||
{
|
||||
$this->preloadHttp();
|
||||
|
||||
$host = $this->getConfig('http.host');
|
||||
$port = $this->getConfig('http.port');
|
||||
$options = $this->getConfig('http.options', []);
|
||||
|
||||
$server = new Worker("\\think\\worker\\protocols\\FlexHttp://{$host}:{$port}", $options);
|
||||
|
||||
$server->reusePort = true;
|
||||
|
||||
$server->onMessage = function (TcpConnection $connection, $data) {
|
||||
if ($data instanceof WorkerRequest) {
|
||||
if ($this->wsEnable && $this->isWebsocketRequest($data)) {
|
||||
$this->onHandShake($connection, $data);
|
||||
} else {
|
||||
$this->onRequest($connection, $data);
|
||||
}
|
||||
} elseif ($data instanceof Frame) {
|
||||
$this->onMessage($connection, $data);
|
||||
}
|
||||
};
|
||||
|
||||
$server->onClose = function (TcpConnection $connection) {
|
||||
if ($this->wsEnable) {
|
||||
$this->onClose($connection);
|
||||
}
|
||||
};
|
||||
|
||||
$server->listen();
|
||||
}
|
||||
|
||||
protected function preloadHttp()
|
||||
{
|
||||
$http = $this->app->http;
|
||||
$this->app->invokeMethod([$http, 'loadMiddleware'], [], true);
|
||||
|
||||
if ($this->app->config->get('app.with_route', true)) {
|
||||
$this->app->invokeMethod([$http, 'loadRoutes'], [], true);
|
||||
$route = clone $this->app->route;
|
||||
unset($this->app->route);
|
||||
|
||||
$this->app->resolving(WorkerHttp::class, function ($http, App $app) use ($route) {
|
||||
$newRoute = clone $route;
|
||||
$this->modifyProperty($newRoute, $app);
|
||||
$app->instance('route', $newRoute);
|
||||
});
|
||||
}
|
||||
|
||||
$middleware = clone $this->app->middleware;
|
||||
unset($this->app->middleware);
|
||||
|
||||
$this->app->resolving(WorkerHttp::class, function ($http, App $app) use ($middleware) {
|
||||
$newMiddleware = clone $middleware;
|
||||
$this->modifyProperty($newMiddleware, $app);
|
||||
$app->instance('middleware', $newMiddleware);
|
||||
});
|
||||
|
||||
unset($this->app->http);
|
||||
$this->app->bind(Http::class, WorkerHttp::class);
|
||||
}
|
||||
|
||||
public function onRequest(TcpConnection $connection, WorkerRequest $wkRequest)
|
||||
{
|
||||
$this->runInSandbox(function (Http $http, Event $event, WorkerApp $app) use ($connection, $wkRequest) {
|
||||
|
||||
$app->setInConsole(false);
|
||||
|
||||
$request = $this->prepareRequest($wkRequest);
|
||||
|
||||
try {
|
||||
$response = $this->handleRequest($http, $request);
|
||||
$this->prepareResponse($response);
|
||||
} catch (Throwable $e) {
|
||||
$handle = $app->make(Handle::class);
|
||||
$handle->report($e);
|
||||
$response = $handle->render($request, $e);
|
||||
}
|
||||
|
||||
$this->sendResponse($connection, $request, $response, $app->cookie);
|
||||
|
||||
//关闭连接
|
||||
$connection->close();
|
||||
|
||||
$http->end($response);
|
||||
});
|
||||
}
|
||||
|
||||
protected function handleRequest(Http $http, $request)
|
||||
{
|
||||
$level = ob_get_level();
|
||||
ob_start();
|
||||
|
||||
$response = $http->run($request);
|
||||
|
||||
if (ob_get_length() > 0) {
|
||||
$content = $response->getContent();
|
||||
$response->content(ob_get_contents() . $content);
|
||||
}
|
||||
|
||||
while (ob_get_level() > $level) {
|
||||
ob_end_clean();
|
||||
}
|
||||
|
||||
return $response;
|
||||
}
|
||||
|
||||
protected function prepareRequest(WorkerRequest $wkRequest)
|
||||
{
|
||||
$header = $wkRequest->header();
|
||||
$server = [];
|
||||
|
||||
foreach ($header as $key => $value) {
|
||||
$server['http_' . str_replace('-', '_', $key)] = $value;
|
||||
}
|
||||
|
||||
// 重新实例化请求对象 处理请求数据
|
||||
/** @var \think\Request $request */
|
||||
$request = $this->app->make('request', [], true);;
|
||||
|
||||
$queryString = $wkRequest->queryString();
|
||||
|
||||
return $request
|
||||
->setMethod($wkRequest->method())
|
||||
->withHeader($header)
|
||||
->withServer($server)
|
||||
->withGet($wkRequest->get())
|
||||
->withPost($wkRequest->post())
|
||||
->withCookie($wkRequest->cookie())
|
||||
->withFiles($wkRequest->file())
|
||||
->withInput($wkRequest->rawBody())
|
||||
->setBaseUrl($wkRequest->uri())
|
||||
->setUrl($wkRequest->uri() . (!empty($queryString) ? '?' . $queryString : ''))
|
||||
->setPathinfo(ltrim($wkRequest->path(), '/'));
|
||||
}
|
||||
|
||||
protected function prepareResponse(\think\Response $response)
|
||||
{
|
||||
switch (true) {
|
||||
case $response instanceof View:
|
||||
$response->getContent();
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
protected function sendResponse(TcpConnection $connection, \think\Request $request, \think\Response $response, Cookie $cookie)
|
||||
{
|
||||
switch (true) {
|
||||
case $response instanceof IteratorResponse:
|
||||
$this->sendIterator($connection, $response, $cookie);
|
||||
break;
|
||||
case $response instanceof FileResponse:
|
||||
$this->sendFile($connection, $request, $response, $cookie);
|
||||
break;
|
||||
default:
|
||||
$this->sendContent($connection, $response, $cookie);
|
||||
}
|
||||
}
|
||||
|
||||
protected function sendIterator(TcpConnection $connection, IteratorResponse $response, Cookie $cookie)
|
||||
{
|
||||
$wkResponse = $this->createResponse($response, $cookie);
|
||||
$connection->send($wkResponse);
|
||||
|
||||
foreach ($response as $content) {
|
||||
$connection->send($content, true);
|
||||
}
|
||||
}
|
||||
|
||||
protected function sendFile(TcpConnection $connection, \think\Request $request, FileResponse $response, Cookie $cookie)
|
||||
{
|
||||
$ifNoneMatch = $request->header('If-None-Match');
|
||||
$ifRange = $request->header('If-Range');
|
||||
|
||||
$code = $response->getCode();
|
||||
$file = $response->getFile();
|
||||
$eTag = $response->getHeader('ETag');
|
||||
$lastModified = $response->getHeader('Last-Modified');
|
||||
|
||||
$fileSize = $file->getSize();
|
||||
$offset = 0;
|
||||
$length = -1;
|
||||
|
||||
if ($ifNoneMatch == $eTag) {
|
||||
$code = 304;
|
||||
} elseif (!$ifRange || $ifRange === $eTag || $ifRange === $lastModified) {
|
||||
$range = $request->header('Range', '');
|
||||
if (Str::startsWith($range, 'bytes=')) {
|
||||
[$start, $end] = explode('-', substr($range, 6), 2) + [0];
|
||||
|
||||
$end = ('' === $end) ? $fileSize - 1 : (int) $end;
|
||||
|
||||
if ('' === $start) {
|
||||
$start = $fileSize - $end;
|
||||
$end = $fileSize - 1;
|
||||
} else {
|
||||
$start = (int) $start;
|
||||
}
|
||||
|
||||
if ($start <= $end) {
|
||||
$end = min($end, $fileSize - 1);
|
||||
if ($start < 0 || $start > $end) {
|
||||
$code = 416;
|
||||
$response->header([
|
||||
'Content-Range' => sprintf('bytes */%s', $fileSize),
|
||||
]);
|
||||
} elseif ($end - $start < $fileSize - 1) {
|
||||
$length = $end < $fileSize ? $end - $start + 1 : -1;
|
||||
$offset = $start;
|
||||
$code = 206;
|
||||
$response->header([
|
||||
'Content-Range' => sprintf('bytes %s-%s/%s', $start, $end, $fileSize),
|
||||
'Content-Length' => $end - $start + 1,
|
||||
]);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
$wkResponse = $this->createResponse($response, $cookie);
|
||||
|
||||
if ($code >= 200 && $code < 300 && $length !== 0) {
|
||||
$wkResponse->withFile($file->getPathname(), $offset, $length);
|
||||
}
|
||||
|
||||
$connection->send($wkResponse);
|
||||
}
|
||||
|
||||
protected function sendContent(TcpConnection $connection, \think\Response $response, Cookie $cookie)
|
||||
{
|
||||
$response->header(['Transfer-Encoding' => 'chunked']);
|
||||
|
||||
$wkResponse = $this->createResponse($response, $cookie);
|
||||
|
||||
$connection->send($wkResponse);
|
||||
|
||||
$content = $response->getContent();
|
||||
if ($content) {
|
||||
$contentSize = strlen($content);
|
||||
$chunkSize = 8192;
|
||||
|
||||
if ($contentSize > $chunkSize) {
|
||||
$sendSize = 0;
|
||||
do {
|
||||
if (!$connection->send(new Chunk(substr($content, $sendSize, $chunkSize)))) {
|
||||
break;
|
||||
}
|
||||
} while (($sendSize += $chunkSize) < $contentSize);
|
||||
} else {
|
||||
$connection->send(new Chunk($content));
|
||||
}
|
||||
}
|
||||
$connection->send(new Chunk(''));
|
||||
}
|
||||
|
||||
protected function createResponse(\think\Response $response, Cookie $cookie, $body = '')
|
||||
{
|
||||
$code = $response->getCode();
|
||||
$header = $response->getHeader();
|
||||
|
||||
$wkResponse = new Response($code, $header, $body);
|
||||
|
||||
foreach ($cookie->getCookie() as $name => $val) {
|
||||
[$value, $expire, $option] = $val;
|
||||
$wkResponse->cookie($name, $value, $expire, $option['path'], $option['domain'], (bool) $option['secure'], (bool) $option['httponly'], $option['samesite']);
|
||||
}
|
||||
|
||||
return $wkResponse;
|
||||
}
|
||||
}
|
||||
97
src/concerns/InteractsWithQueue.php
Normal file
97
src/concerns/InteractsWithQueue.php
Normal file
@@ -0,0 +1,97 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\concerns;
|
||||
|
||||
use think\helper\Arr;
|
||||
use think\queue\event\JobFailed;
|
||||
use think\queue\Worker;
|
||||
use Workerman\Timer;
|
||||
|
||||
trait InteractsWithQueue
|
||||
{
|
||||
protected function createQueueWorkers()
|
||||
{
|
||||
$workers = $this->getConfig('queue.workers', []);
|
||||
|
||||
foreach ($workers as $queue => $options) {
|
||||
|
||||
if (strpos($queue, '@') !== false) {
|
||||
[$queue, $connection] = explode('@', $queue);
|
||||
} else {
|
||||
$connection = null;
|
||||
}
|
||||
|
||||
$workerNum = Arr::get($options, 'worker_num', 1);
|
||||
|
||||
$this->addWorker(function () use ($options, $connection, $queue) {
|
||||
$delay = Arr::get($options, 'delay', 0);
|
||||
$sleep = Arr::get($options, 'sleep', 3);
|
||||
$tries = Arr::get($options, 'tries', 0);
|
||||
$timeout = Arr::get($options, 'timeout', 60);
|
||||
$qWorker = $this->app->make(Worker::class);
|
||||
|
||||
if ($this->supportsAsyncSignals()) {
|
||||
pcntl_signal(SIGALRM, function () {
|
||||
\think\worker\Worker::stopAll();
|
||||
});
|
||||
|
||||
pcntl_alarm($timeout);
|
||||
}
|
||||
|
||||
$this->createRunTimer(function () use ($connection, $queue, $delay, $sleep, $tries, $qWorker) {
|
||||
$this->runInSandbox(function () use ($connection, $queue, $delay, $sleep, $tries, $qWorker) {
|
||||
$qWorker->runNextJob($connection, $queue, $delay, $sleep, $tries);
|
||||
});
|
||||
if ($this->supportsAsyncSignals()) {
|
||||
pcntl_alarm(0);
|
||||
}
|
||||
});
|
||||
}, "queue [$queue]", $workerNum);
|
||||
}
|
||||
}
|
||||
|
||||
protected function supportsAsyncSignals()
|
||||
{
|
||||
return extension_loaded('pcntl');
|
||||
}
|
||||
|
||||
protected function createRunTimer($func)
|
||||
{
|
||||
Timer::add(0.1, function () use ($func) {
|
||||
$func();
|
||||
$this->createRunTimer($func);
|
||||
}, [], false);
|
||||
}
|
||||
|
||||
protected function prepareQueue()
|
||||
{
|
||||
if ($this->getConfig('queue.enable', false)) {
|
||||
$this->listenForEvents();
|
||||
$this->createQueueWorkers();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 注册事件
|
||||
*/
|
||||
protected function listenForEvents()
|
||||
{
|
||||
$this->container->event->listen(JobFailed::class, function (JobFailed $event) {
|
||||
$this->logFailedJob($event);
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* 记录失败任务
|
||||
* @param JobFailed $event
|
||||
*/
|
||||
protected function logFailedJob(JobFailed $event)
|
||||
{
|
||||
$this->container['queue.failer']->log(
|
||||
$event->connection,
|
||||
$event->job->getQueue(),
|
||||
$event->job->getRawBody(),
|
||||
$event->exception
|
||||
);
|
||||
}
|
||||
}
|
||||
119
src/concerns/InteractsWithServer.php
Normal file
119
src/concerns/InteractsWithServer.php
Normal file
@@ -0,0 +1,119 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\concerns;
|
||||
|
||||
use think\App;
|
||||
use think\worker\Ipc;
|
||||
use think\worker\Watcher;
|
||||
use think\worker\Worker;
|
||||
use Workerman\Redis\Client;
|
||||
|
||||
/**
|
||||
* Trait InteractsWithServer
|
||||
* @property App $container
|
||||
*/
|
||||
trait InteractsWithServer
|
||||
{
|
||||
/** @var Ipc */
|
||||
protected $ipc;
|
||||
|
||||
protected $workerId = null;
|
||||
protected $stopping = false;
|
||||
|
||||
public function addWorker(callable $func, $name = 'none', int $count = 1): Worker
|
||||
{
|
||||
$worker = new Worker();
|
||||
|
||||
$worker->name = $name;
|
||||
$worker->count = $count;
|
||||
|
||||
$worker->onWorkerStart = function (Worker $worker) use ($func) {
|
||||
$this->clearCache();
|
||||
$this->prepareApplication();
|
||||
|
||||
$this->conduit->connect();
|
||||
|
||||
$this->workerId = $this->ipc->listenMessage();
|
||||
|
||||
$this->triggerEvent('workerStart', $worker);
|
||||
|
||||
$func($worker);
|
||||
};
|
||||
|
||||
$worker->onWorkerReload = function () {
|
||||
$this->stopping = true;
|
||||
};
|
||||
|
||||
return $worker;
|
||||
}
|
||||
|
||||
public function getWorkerId()
|
||||
{
|
||||
return $this->workerId;
|
||||
}
|
||||
|
||||
public function isStopping()
|
||||
{
|
||||
return $this->stopping;
|
||||
}
|
||||
|
||||
/**
|
||||
* 启动服务
|
||||
*/
|
||||
public function start(): void
|
||||
{
|
||||
$this->initialize();
|
||||
$this->prepareIpc();
|
||||
$this->triggerEvent('init');
|
||||
|
||||
//热更新
|
||||
if ($this->getConfig('hot_update.enable', false)) {
|
||||
$this->addHotUpdateWorker();
|
||||
}
|
||||
|
||||
Worker::runAll();
|
||||
}
|
||||
|
||||
protected function prepareIpc()
|
||||
{
|
||||
$this->ipc = $this->container->make(Ipc::class);
|
||||
}
|
||||
|
||||
public function sendMessage($workerId, $message)
|
||||
{
|
||||
$this->ipc->sendMessage($workerId, $message);
|
||||
}
|
||||
|
||||
/**
|
||||
* 热更新
|
||||
*/
|
||||
protected function addHotUpdateWorker()
|
||||
{
|
||||
$worker = new Worker();
|
||||
|
||||
$worker->name = 'hot update';
|
||||
$worker->reloadable = false;
|
||||
|
||||
$worker->onWorkerStart = function () {
|
||||
$watcher = $this->container->make(Watcher::class);
|
||||
$watcher->watch(function () {
|
||||
posix_kill(posix_getppid(), SIGUSR1);
|
||||
});
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* 清除apc、op缓存
|
||||
*/
|
||||
protected function clearCache()
|
||||
{
|
||||
if (extension_loaded('apc')) {
|
||||
apc_clear_cache();
|
||||
}
|
||||
|
||||
if (extension_loaded('Zend OPcache')) {
|
||||
opcache_reset();
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
145
src/concerns/InteractsWithWebsocket.php
Normal file
145
src/concerns/InteractsWithWebsocket.php
Normal file
@@ -0,0 +1,145 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\concerns;
|
||||
|
||||
use think\App;
|
||||
use think\Event;
|
||||
use think\helper\Arr;
|
||||
use think\Http;
|
||||
use think\worker\message\PushMessage;
|
||||
use think\worker\Websocket;
|
||||
use think\worker\contract\websocket\HandlerInterface;
|
||||
use think\worker\websocket\Frame;
|
||||
use think\worker\websocket\Handler;
|
||||
use Throwable;
|
||||
use Workerman\Connection\TcpConnection;
|
||||
use Workerman\Protocols\Http\Request as WorkerRequest;
|
||||
use Workerman\Protocols\Http\Response;
|
||||
|
||||
trait InteractsWithWebsocket
|
||||
{
|
||||
|
||||
protected $messageSender = [];
|
||||
|
||||
protected function prepareWebsocket()
|
||||
{
|
||||
$this->onEvent('workerStart', function () {
|
||||
$handlerClass = $this->getConfig('websocket.handler', Handler::class);
|
||||
$this->app->bind(HandlerInterface::class, $handlerClass);
|
||||
|
||||
$this->onEvent('message', function ($message) {
|
||||
if ($message instanceof PushMessage) {
|
||||
if (isset($this->messageSender[$message->to])) {
|
||||
$this->messageSender[$message->to]($message->data);
|
||||
}
|
||||
}
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
public function onHandShake(TcpConnection $connection, WorkerRequest $wkRequest)
|
||||
{
|
||||
$this->runInSandbox(function (App $app, Http $http, Event $event) use ($connection, $wkRequest) {
|
||||
$request = $this->prepareRequest($wkRequest);
|
||||
|
||||
$response = $http->run($request);
|
||||
if (!$response instanceof \think\worker\response\Websocket) {
|
||||
$connection->close();
|
||||
return;
|
||||
}
|
||||
|
||||
$event->subscribe([$response]);
|
||||
$this->upgrade($connection, $wkRequest);
|
||||
|
||||
$websocket = $app->make(Websocket::class, [$connection], true);
|
||||
$app->instance(Websocket::class, $websocket);
|
||||
|
||||
$id = "{$this->workerId}.{$connection->id}";
|
||||
|
||||
$websocket->setSender($id);
|
||||
$websocket->join($id);
|
||||
|
||||
$handler = $app->make(HandlerInterface::class);
|
||||
|
||||
$this->messageSender[$connection->id] = function ($data) use ($connection, $handler) {
|
||||
$connection->send($handler->encodeMessage($data));
|
||||
};
|
||||
|
||||
try {
|
||||
$handler->onOpen($request);
|
||||
} catch (Throwable $e) {
|
||||
$this->logServerError($e);
|
||||
}
|
||||
}, $connection);
|
||||
}
|
||||
|
||||
public function onMessage(TcpConnection $connection, Frame $frame)
|
||||
{
|
||||
$this->runInSandbox(function (App $app) use ($frame) {
|
||||
$handler = $app->make(HandlerInterface::class);
|
||||
try {
|
||||
$handler->onMessage($frame);
|
||||
} catch (Throwable $e) {
|
||||
$this->logServerError($e);
|
||||
}
|
||||
}, $connection);
|
||||
}
|
||||
|
||||
public function onClose(TcpConnection $connection)
|
||||
{
|
||||
$this->runInSandbox(function (App $app) use ($connection) {
|
||||
if ($app->exists(Websocket::class)) {
|
||||
$websocket = $app->make(Websocket::class);
|
||||
$handler = $app->make(HandlerInterface::class);
|
||||
try {
|
||||
$handler->onClose();
|
||||
} catch (Throwable $e) {
|
||||
$this->logServerError($e);
|
||||
}
|
||||
|
||||
// leave all rooms
|
||||
$websocket->leave();
|
||||
|
||||
unset($this->messageSender[$connection->id]);
|
||||
|
||||
$websocket->setConnected(false);
|
||||
}
|
||||
}, $connection);
|
||||
}
|
||||
|
||||
protected function isWebsocketRequest(WorkerRequest $request)
|
||||
{
|
||||
$header = $request->header();
|
||||
return strcasecmp(Arr::get($header, 'connection', ''), 'upgrade') === 0 &&
|
||||
strcasecmp(Arr::get($header, 'upgrade', ''), 'websocket') === 0;
|
||||
}
|
||||
|
||||
protected function upgrade(TcpConnection $connection, WorkerRequest $request)
|
||||
{
|
||||
$key = $request->header('Sec-WebSocket-Key');
|
||||
|
||||
$headers = [
|
||||
'Upgrade' => 'websocket',
|
||||
'Sec-WebSocket-Version' => '13',
|
||||
'Connection' => 'Upgrade',
|
||||
'Sec-WebSocket-Accept' => base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true)),
|
||||
];
|
||||
|
||||
if ($protocol = $request->header('Sec-Websocket-Protocol')) {
|
||||
$headers['Sec-WebSocket-Protocol'] = $protocol;
|
||||
}
|
||||
|
||||
$response = new Response(101, $headers);
|
||||
|
||||
$connection->send($response);
|
||||
|
||||
// Websocket data buffer.
|
||||
$connection->context->websocketDataBuffer = '';
|
||||
// Current websocket frame length.
|
||||
$connection->context->websocketCurrentFrameLength = 0;
|
||||
// Current websocket frame data.
|
||||
$connection->context->websocketCurrentFrameBuffer = '';
|
||||
|
||||
$connection->context->websocketHandshake = true;
|
||||
}
|
||||
}
|
||||
18
src/concerns/ModifyProperty.php
Normal file
18
src/concerns/ModifyProperty.php
Normal file
@@ -0,0 +1,18 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\concerns;
|
||||
|
||||
use ReflectionObject;
|
||||
|
||||
trait ModifyProperty
|
||||
{
|
||||
protected function modifyProperty($object, $value, $property = 'app')
|
||||
{
|
||||
$reflectObject = new ReflectionObject($object);
|
||||
if ($reflectObject->hasProperty($property)) {
|
||||
$reflectProperty = $reflectObject->getProperty($property);
|
||||
$reflectProperty->setAccessible(true);
|
||||
$reflectProperty->setValue($object, $value);
|
||||
}
|
||||
}
|
||||
}
|
||||
77
src/concerns/WithApplication.php
Normal file
77
src/concerns/WithApplication.php
Normal file
@@ -0,0 +1,77 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\concerns;
|
||||
|
||||
use Closure;
|
||||
use think\App;
|
||||
use think\worker\App as WorkerApp;
|
||||
use think\worker\Manager;
|
||||
use think\worker\Sandbox;
|
||||
use Throwable;
|
||||
|
||||
/**
|
||||
* Trait WithApplication
|
||||
* @property App $container
|
||||
*/
|
||||
trait WithApplication
|
||||
{
|
||||
/**
|
||||
* @var WorkerApp
|
||||
*/
|
||||
protected $app;
|
||||
|
||||
protected function prepareApplication()
|
||||
{
|
||||
if (!$this->app instanceof WorkerApp) {
|
||||
$this->app = new WorkerApp($this->container->getRootPath());
|
||||
|
||||
$this->app->bind(WorkerApp::class, App::class);
|
||||
$this->app->bind(Manager::class, $this);
|
||||
|
||||
$this->app->initialize();
|
||||
$this->app->instance('request', $this->container->request);
|
||||
$this->prepareConcretes();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 预加载
|
||||
*/
|
||||
protected function prepareConcretes()
|
||||
{
|
||||
$defaultConcretes = ['db', 'cache', 'event'];
|
||||
|
||||
$concretes = array_merge($defaultConcretes, $this->getConfig('concretes', []));
|
||||
|
||||
foreach ($concretes as $concrete) {
|
||||
$this->app->make($concrete);
|
||||
}
|
||||
}
|
||||
|
||||
public function getApp()
|
||||
{
|
||||
return $this->app;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取沙箱
|
||||
* @return Sandbox
|
||||
*/
|
||||
protected function getSandbox()
|
||||
{
|
||||
return $this->app->make(Sandbox::class);
|
||||
}
|
||||
|
||||
/**
|
||||
* 在沙箱中执行
|
||||
* @param Closure $callable
|
||||
*/
|
||||
public function runInSandbox(Closure $callable, ?object $key = null)
|
||||
{
|
||||
try {
|
||||
$this->getSandbox()->run($callable, $key);
|
||||
} catch (Throwable $e) {
|
||||
$this->logServerError($e);
|
||||
}
|
||||
}
|
||||
}
|
||||
75
src/concerns/WithContainer.php
Normal file
75
src/concerns/WithContainer.php
Normal file
@@ -0,0 +1,75 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\concerns;
|
||||
|
||||
use think\App;
|
||||
use think\exception\Handle;
|
||||
use Throwable;
|
||||
|
||||
trait WithContainer
|
||||
{
|
||||
|
||||
/**
|
||||
* @var App
|
||||
*/
|
||||
protected $container;
|
||||
|
||||
/**
|
||||
* Manager constructor.
|
||||
* @param App $container
|
||||
*/
|
||||
public function __construct(App $container)
|
||||
{
|
||||
$this->container = $container;
|
||||
}
|
||||
|
||||
protected function getContainer()
|
||||
{
|
||||
return $this->container;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取配置
|
||||
* @param string $name
|
||||
* @param mixed $default
|
||||
* @return mixed
|
||||
*/
|
||||
public function getConfig(string $name, $default = null)
|
||||
{
|
||||
return $this->container->config->get("worker.{$name}", $default);
|
||||
}
|
||||
|
||||
/**
|
||||
* 触发事件
|
||||
* @param string $event
|
||||
* @param mixed $params
|
||||
*/
|
||||
public function triggerEvent(string $event, $params = null): void
|
||||
{
|
||||
$this->container->event->trigger("worker.{$event}", $params);
|
||||
}
|
||||
|
||||
/**
|
||||
* 监听事件
|
||||
* @param string $event
|
||||
* @param $listener
|
||||
* @param bool $first
|
||||
*/
|
||||
public function onEvent(string $event, $listener, bool $first = false): void
|
||||
{
|
||||
$this->container->event->listen("worker.{$event}", $listener, $first);
|
||||
}
|
||||
|
||||
/**
|
||||
* Log server error.
|
||||
*
|
||||
* @param Throwable $e
|
||||
*/
|
||||
public function logServerError(Throwable $e)
|
||||
{
|
||||
/** @var Handle $handle */
|
||||
$handle = $this->container->make(Handle::class);
|
||||
|
||||
$handle->report($e);
|
||||
}
|
||||
}
|
||||
26
src/conduit/Driver.php
Normal file
26
src/conduit/Driver.php
Normal file
@@ -0,0 +1,26 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\conduit;
|
||||
|
||||
abstract class Driver
|
||||
{
|
||||
abstract public function prepare();
|
||||
|
||||
abstract public function connect();
|
||||
|
||||
abstract public function get(string $name);
|
||||
|
||||
abstract public function set(string $name, $value);
|
||||
|
||||
abstract public function inc(string $name, int $step = 1);
|
||||
|
||||
abstract public function sAdd(string $name, ...$value);
|
||||
|
||||
abstract public function sRem(string $name, $value);
|
||||
|
||||
abstract public function sMembers(string $name);
|
||||
|
||||
abstract public function publish(string $name, $value);
|
||||
|
||||
abstract public function subscribe(string $name, $callback);
|
||||
}
|
||||
186
src/conduit/driver/Socket.php
Normal file
186
src/conduit/driver/Socket.php
Normal file
@@ -0,0 +1,186 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\conduit\driver;
|
||||
|
||||
use Exception;
|
||||
use Revolt\EventLoop;
|
||||
use Revolt\EventLoop\Suspension;
|
||||
use think\worker\conduit\Driver;
|
||||
use think\worker\conduit\driver\socket\Command;
|
||||
use think\worker\conduit\driver\socket\Event;
|
||||
use think\worker\conduit\driver\socket\Result;
|
||||
use think\worker\conduit\driver\socket\Server;
|
||||
use think\worker\Manager;
|
||||
use Workerman\Connection\AsyncTcpConnection;
|
||||
use Workerman\Protocols\Frame;
|
||||
use Workerman\Timer;
|
||||
|
||||
class Socket extends Driver
|
||||
{
|
||||
protected $id = 0;
|
||||
protected $domain;
|
||||
|
||||
/** @var AsyncTcpConnection|null */
|
||||
protected $connection = null;
|
||||
protected $reconnectTimer;
|
||||
protected $pingInterval = 55;
|
||||
|
||||
/** @var array<int, array{0: Suspension, 1: int}> */
|
||||
protected $suspensions = [];
|
||||
protected $events = [];
|
||||
|
||||
public function __construct(protected Manager $manager)
|
||||
{
|
||||
$filename = runtime_path() . 'conduit.sock';
|
||||
@unlink($filename);
|
||||
$this->domain = "unix://{$filename}";
|
||||
}
|
||||
|
||||
public function prepare()
|
||||
{
|
||||
//启动服务端
|
||||
Server::run($this->domain);
|
||||
}
|
||||
|
||||
public function connect()
|
||||
{
|
||||
$suspension = EventLoop::getSuspension();
|
||||
$this->connection = $this->createConnection($suspension);
|
||||
$suspension->suspend();
|
||||
|
||||
Timer::add($this->pingInterval, function () {
|
||||
if ($this->connection) {
|
||||
$this->connection->send('');
|
||||
}
|
||||
});
|
||||
|
||||
Timer::add(1, function () {
|
||||
//检查是否超时
|
||||
foreach ($this->suspensions as $id => $suspension) {
|
||||
if (time() - $suspension[1] > 10) {
|
||||
$suspension[0]->throw(new Exception('conduit connection is timeout'));
|
||||
unset($this->suspensions[$id]);
|
||||
}
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
public function get(string $name)
|
||||
{
|
||||
return $this->sendAndRecv(Command::create('get', $name));
|
||||
}
|
||||
|
||||
public function set(string $name, $value)
|
||||
{
|
||||
$this->send(Command::create('set', $name, $value));
|
||||
}
|
||||
|
||||
public function inc(string $name, int $step = 1)
|
||||
{
|
||||
return $this->sendAndRecv(Command::create('inc', $name, $step));
|
||||
}
|
||||
|
||||
public function sAdd(string $name, ...$value)
|
||||
{
|
||||
$this->send(Command::create('sAdd', $name, $value));
|
||||
}
|
||||
|
||||
public function sRem(string $name, $value)
|
||||
{
|
||||
$this->send(Command::create('sRem', $name, $value));
|
||||
}
|
||||
|
||||
public function sMembers(string $name)
|
||||
{
|
||||
return $this->sendAndRecv(Command::create('sMembers', $name));
|
||||
}
|
||||
|
||||
public function publish(string $name, $value)
|
||||
{
|
||||
$this->send(Command::create('publish', $name, $value));
|
||||
}
|
||||
|
||||
public function subscribe(string $name, $callback)
|
||||
{
|
||||
$this->send(Command::create('subscribe', $name));
|
||||
$this->events[$name] = $callback;
|
||||
}
|
||||
|
||||
protected function sendAndRecv(Command $command)
|
||||
{
|
||||
$suspension = EventLoop::getSuspension();
|
||||
|
||||
$id = $this->id++;
|
||||
|
||||
$command->id = $id;
|
||||
|
||||
$this->suspensions[$id] = [$suspension, time()];
|
||||
|
||||
$this->send($command);
|
||||
|
||||
return $suspension->suspend();
|
||||
}
|
||||
|
||||
protected function send(Command $command)
|
||||
{
|
||||
if (!$this->connection) {
|
||||
throw new Exception('conduit connection is disconnected');
|
||||
}
|
||||
|
||||
$this->connection->send(serialize($command));
|
||||
}
|
||||
|
||||
protected function createConnection(?Suspension $suspension = null)
|
||||
{
|
||||
$connection = new AsyncTcpConnection($this->domain);
|
||||
|
||||
$connection->protocol = Frame::class;
|
||||
|
||||
$connection->onConnect = function () use ($suspension) {
|
||||
$this->clearTimer();
|
||||
if ($suspension) {
|
||||
$suspension->resume();
|
||||
}
|
||||
//补订阅
|
||||
foreach ($this->events as $name => $callback) {
|
||||
$this->send(Command::create('subscribe', $name));
|
||||
}
|
||||
};
|
||||
|
||||
$connection->onMessage = function ($connection, $buffer) {
|
||||
/** @var Result|Event $result */
|
||||
$result = unserialize($buffer);
|
||||
|
||||
if ($result instanceof Event) {
|
||||
if (isset($this->events[$result->name])) {
|
||||
$this->events[$result->name]($result->data);
|
||||
}
|
||||
} elseif (isset($result->id) && isset($this->suspensions[$result->id])) {
|
||||
[$suspension] = $this->suspensions[$result->id];
|
||||
$suspension->resume($result->data);
|
||||
unset($this->suspensions[$result->id]);
|
||||
}
|
||||
};
|
||||
|
||||
$connection->onClose = function () {
|
||||
$this->connection = null;
|
||||
//重连
|
||||
$this->clearTimer();
|
||||
$this->reconnectTimer = Timer::add(1, function () {
|
||||
$this->connection = $this->createConnection();
|
||||
});
|
||||
};
|
||||
|
||||
$connection->connect();
|
||||
|
||||
return $connection;
|
||||
}
|
||||
|
||||
protected function clearTimer()
|
||||
{
|
||||
if ($this->reconnectTimer) {
|
||||
Timer::del($this->reconnectTimer);
|
||||
$this->reconnectTimer = null;
|
||||
}
|
||||
}
|
||||
}
|
||||
22
src/conduit/driver/socket/Command.php
Normal file
22
src/conduit/driver/socket/Command.php
Normal file
@@ -0,0 +1,22 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\conduit\driver\socket;
|
||||
|
||||
class Command
|
||||
{
|
||||
public $id;
|
||||
public $name;
|
||||
public $key;
|
||||
public $data;
|
||||
|
||||
public static function create($name, $key, $data = null)
|
||||
{
|
||||
$packet = new self();
|
||||
|
||||
$packet->name = $name;
|
||||
$packet->key = $key;
|
||||
$packet->data = $data;
|
||||
|
||||
return $packet;
|
||||
}
|
||||
}
|
||||
15
src/conduit/driver/socket/Event.php
Normal file
15
src/conduit/driver/socket/Event.php
Normal file
@@ -0,0 +1,15 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\conduit\driver\socket;
|
||||
|
||||
class Event
|
||||
{
|
||||
public function __construct(public $name, public $data)
|
||||
{
|
||||
}
|
||||
|
||||
public static function create($name, $data)
|
||||
{
|
||||
return new self($name, $data);
|
||||
}
|
||||
}
|
||||
16
src/conduit/driver/socket/Result.php
Normal file
16
src/conduit/driver/socket/Result.php
Normal file
@@ -0,0 +1,16 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\conduit\driver\socket;
|
||||
|
||||
class Result
|
||||
{
|
||||
public function __construct(public $id = null, public $data = null)
|
||||
{
|
||||
|
||||
}
|
||||
|
||||
public static function create($id = null)
|
||||
{
|
||||
return new self($id);
|
||||
}
|
||||
}
|
||||
102
src/conduit/driver/socket/Server.php
Normal file
102
src/conduit/driver/socket/Server.php
Normal file
@@ -0,0 +1,102 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\conduit\driver\socket;
|
||||
|
||||
use think\worker\Worker;
|
||||
use Workerman\Connection\TcpConnection;
|
||||
use Workerman\Protocols\Frame;
|
||||
|
||||
class Server
|
||||
{
|
||||
protected $data = [];
|
||||
|
||||
/** @var array<string,TcpConnection[]> */
|
||||
protected $subscribers = [];
|
||||
|
||||
public function onMessage(TcpConnection $connection, $buffer)
|
||||
{
|
||||
if (empty($buffer)) {
|
||||
return;
|
||||
}
|
||||
/** @var Command $command */
|
||||
$command = unserialize($buffer);
|
||||
|
||||
$result = Result::create($command->id);
|
||||
|
||||
switch ($command->name) {
|
||||
case 'get':
|
||||
$result->data = $this->data[$command->key] ?? null;
|
||||
break;
|
||||
case 'set':
|
||||
$this->data[$command->key] = $command->data;
|
||||
break;
|
||||
case 'inc':
|
||||
if (!isset($this->data[$command->key]) || !is_integer($this->data[$command->key])) {
|
||||
$this->data[$command->key] = 0;
|
||||
}
|
||||
$result->data = $this->data[$command->key] += $command->data ?? 1;
|
||||
break;
|
||||
case 'sAdd':
|
||||
if (!isset($this->data[$command->key]) || !is_array($this->data[$command->key])) {
|
||||
$this->data[$command->key] = [];
|
||||
}
|
||||
$this->data[$command->key] = array_merge($this->data[$command->key], $command->data);
|
||||
break;
|
||||
case 'sRem':
|
||||
if (!isset($this->data[$command->key]) || !is_array($this->data[$command->key])) {
|
||||
$this->data[$command->key] = [];
|
||||
}
|
||||
$this->data[$command->key] = array_diff($this->data[$command->key], [$command->data]);
|
||||
break;
|
||||
case 'sMembers':
|
||||
if (!isset($this->data[$command->key]) || !is_array($this->data[$command->key])) {
|
||||
$this->data[$command->key] = [];
|
||||
}
|
||||
$result->data = $this->data[$command->key];
|
||||
break;
|
||||
case 'subscribe':
|
||||
if (!isset($this->subscribers[$command->key])) {
|
||||
$this->subscribers[$command->key] = [];
|
||||
}
|
||||
$this->subscribers[$command->key][] = $connection;
|
||||
break;
|
||||
case 'publish':
|
||||
if (!empty($this->subscribers[$command->key])) {
|
||||
foreach ($this->subscribers[$command->key] as $conn) {
|
||||
$conn->send(serialize(Event::create($command->key, $command->data)));
|
||||
}
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
if (isset($result->id)) {
|
||||
$connection->send(serialize($result));
|
||||
}
|
||||
}
|
||||
|
||||
public function onClose(TcpConnection $connection)
|
||||
{
|
||||
if (!empty($this->subscribers)) {
|
||||
foreach ($this->subscribers as $key => $connections) {
|
||||
$this->subscribers[$key] = array_udiff($connections, [$connection], function ($a, $b) {
|
||||
return $a <=> $b;
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public static function run($domain)
|
||||
{
|
||||
//启动服务端
|
||||
$server = new self();
|
||||
|
||||
$worker = new Worker($domain);
|
||||
|
||||
$worker->name = 'conduit';
|
||||
$worker->protocol = Frame::class;
|
||||
$worker->reloadable = false;
|
||||
|
||||
$worker->onMessage = [$server, 'onMessage'];
|
||||
$worker->onClose = [$server, 'onClose'];
|
||||
}
|
||||
}
|
||||
39
src/config/worker.php
Normal file
39
src/config/worker.php
Normal file
@@ -0,0 +1,39 @@
|
||||
<?php
|
||||
// +----------------------------------------------------------------------
|
||||
// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2006-2018 http://thinkphp.cn All rights reserved.
|
||||
// +----------------------------------------------------------------------
|
||||
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
||||
// +----------------------------------------------------------------------
|
||||
// | Author: liu21st <liu21st@gmail.com>
|
||||
// +----------------------------------------------------------------------
|
||||
|
||||
use think\worker\websocket\Handler;
|
||||
|
||||
return [
|
||||
'http' => [
|
||||
'enable' => true,
|
||||
'host' => '0.0.0.0',
|
||||
'port' => 8080,
|
||||
'worker_num' => 4,
|
||||
'options' => [],
|
||||
],
|
||||
'websocket' => [
|
||||
'enable' => false,
|
||||
'handler' => Handler::class,
|
||||
'ping_interval' => 25000,
|
||||
'ping_timeout' => 60000,
|
||||
],
|
||||
//队列
|
||||
'queue' => [
|
||||
'enable' => false,
|
||||
'workers' => [],
|
||||
],
|
||||
'hot_update' => [
|
||||
'enable' => env('APP_DEBUG', false),
|
||||
'name' => ['*.php'],
|
||||
'include' => [app_path(), config_path(), root_path('route')],
|
||||
'exclude' => [],
|
||||
],
|
||||
];
|
||||
17
src/contract/ResetterInterface.php
Normal file
17
src/contract/ResetterInterface.php
Normal file
@@ -0,0 +1,17 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\contract;
|
||||
|
||||
use think\App;
|
||||
use think\worker\Sandbox;
|
||||
|
||||
interface ResetterInterface
|
||||
{
|
||||
/**
|
||||
* "handle" function for resetting app.
|
||||
*
|
||||
* @param \think\App $app
|
||||
* @param Sandbox $sandbox
|
||||
*/
|
||||
public function handle(App $app, Sandbox $sandbox);
|
||||
}
|
||||
31
src/contract/websocket/HandlerInterface.php
Normal file
31
src/contract/websocket/HandlerInterface.php
Normal file
@@ -0,0 +1,31 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\contract\websocket;
|
||||
|
||||
use think\Request;
|
||||
use think\worker\websocket\Frame;
|
||||
|
||||
interface HandlerInterface
|
||||
{
|
||||
/**
|
||||
* "onOpen" listener.
|
||||
*
|
||||
* @param Request $request
|
||||
*/
|
||||
public function onOpen(Request $request);
|
||||
|
||||
/**
|
||||
* "onMessage" listener.
|
||||
*
|
||||
* @param Frame $frame
|
||||
*/
|
||||
public function onMessage(Frame $frame);
|
||||
|
||||
/**
|
||||
* "onClose" listener.
|
||||
*/
|
||||
public function onClose();
|
||||
|
||||
public function encodeMessage($message);
|
||||
|
||||
}
|
||||
15
src/message/PushMessage.php
Normal file
15
src/message/PushMessage.php
Normal file
@@ -0,0 +1,15 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\message;
|
||||
|
||||
class PushMessage
|
||||
{
|
||||
public $to;
|
||||
public $data;
|
||||
|
||||
public function __construct($to, $data)
|
||||
{
|
||||
$this->to = $to;
|
||||
$this->data = $data;
|
||||
}
|
||||
}
|
||||
39
src/protocols/FlexHttp.php
Normal file
39
src/protocols/FlexHttp.php
Normal file
@@ -0,0 +1,39 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\protocols;
|
||||
|
||||
use think\worker\websocket\Frame;
|
||||
use Workerman\Connection\TcpConnection;
|
||||
use Workerman\Protocols\Http;
|
||||
use Workerman\Protocols\Websocket;
|
||||
|
||||
class FlexHttp
|
||||
{
|
||||
public static function input(string $buffer, TcpConnection $connection)
|
||||
{
|
||||
if (empty($connection->context->websocketHandshake)) {
|
||||
return Http::input($buffer, $connection);
|
||||
} else {
|
||||
return Websocket::input($buffer, $connection);
|
||||
}
|
||||
}
|
||||
|
||||
public static function decode(string $buffer, TcpConnection $connection)
|
||||
{
|
||||
if (empty($connection->context->websocketHandshake)) {
|
||||
return Http::decode($buffer, $connection);
|
||||
} else {
|
||||
$data = Websocket::decode($buffer, $connection);
|
||||
return new Frame($connection->id, $data);
|
||||
}
|
||||
}
|
||||
|
||||
public static function encode(mixed $response, TcpConnection $connection): string
|
||||
{
|
||||
if (empty($connection->context->websocketHandshake)) {
|
||||
return Http::encode($response, $connection);
|
||||
} else {
|
||||
return Websocket::encode($response, $connection);
|
||||
}
|
||||
}
|
||||
}
|
||||
23
src/resetters/ClearInstances.php
Normal file
23
src/resetters/ClearInstances.php
Normal file
@@ -0,0 +1,23 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\resetters;
|
||||
|
||||
use think\App;
|
||||
use think\worker\contract\ResetterInterface;
|
||||
use think\worker\Sandbox;
|
||||
|
||||
class ClearInstances implements ResetterInterface
|
||||
{
|
||||
public function handle(App $app, Sandbox $sandbox)
|
||||
{
|
||||
$instances = ['log', 'session', 'view', 'response', 'cookie'];
|
||||
|
||||
$instances = array_merge($instances, $sandbox->getConfig()->get('worker.instances', []));
|
||||
|
||||
foreach ($instances as $instance) {
|
||||
$app->delete($instance);
|
||||
}
|
||||
|
||||
return $app;
|
||||
}
|
||||
}
|
||||
18
src/resetters/ResetConfig.php
Normal file
18
src/resetters/ResetConfig.php
Normal file
@@ -0,0 +1,18 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\resetters;
|
||||
|
||||
use think\App;
|
||||
use think\worker\contract\ResetterInterface;
|
||||
use think\worker\Sandbox;
|
||||
|
||||
class ResetConfig implements ResetterInterface
|
||||
{
|
||||
|
||||
public function handle(App $app, Sandbox $sandbox)
|
||||
{
|
||||
$app->instance('config', clone $sandbox->getConfig());
|
||||
|
||||
return $app;
|
||||
}
|
||||
}
|
||||
22
src/resetters/ResetEvent.php
Normal file
22
src/resetters/ResetEvent.php
Normal file
@@ -0,0 +1,22 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\resetters;
|
||||
|
||||
use think\App;
|
||||
use think\worker\concerns\ModifyProperty;
|
||||
use think\worker\contract\ResetterInterface;
|
||||
use think\worker\Sandbox;
|
||||
|
||||
class ResetEvent implements ResetterInterface
|
||||
{
|
||||
use ModifyProperty;
|
||||
|
||||
public function handle(App $app, Sandbox $sandbox)
|
||||
{
|
||||
$event = clone $sandbox->getEvent();
|
||||
$this->modifyProperty($event, $app);
|
||||
$app->instance('event', $event);
|
||||
|
||||
return $app;
|
||||
}
|
||||
}
|
||||
21
src/resetters/ResetModel.php
Normal file
21
src/resetters/ResetModel.php
Normal file
@@ -0,0 +1,21 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\resetters;
|
||||
|
||||
use think\App;
|
||||
use think\Model;
|
||||
use think\worker\contract\ResetterInterface;
|
||||
use think\worker\Sandbox;
|
||||
|
||||
class ResetModel implements ResetterInterface
|
||||
{
|
||||
|
||||
public function handle(App $app, Sandbox $sandbox)
|
||||
{
|
||||
if (class_exists(Model::class)) {
|
||||
Model::setInvoker(function (...$args) use ($sandbox) {
|
||||
return $sandbox->getSnapshot()->invoke(...$args);
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
30
src/resetters/ResetPaginator.php
Normal file
30
src/resetters/ResetPaginator.php
Normal file
@@ -0,0 +1,30 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\resetters;
|
||||
|
||||
use think\App;
|
||||
use think\Paginator;
|
||||
use think\worker\contract\ResetterInterface;
|
||||
use think\worker\Sandbox;
|
||||
|
||||
class ResetPaginator implements ResetterInterface
|
||||
{
|
||||
|
||||
public function handle(App $app, Sandbox $sandbox)
|
||||
{
|
||||
Paginator::currentPathResolver(function () use ($sandbox) {
|
||||
return $sandbox->getSnapshot()->request->baseUrl();
|
||||
});
|
||||
|
||||
Paginator::currentPageResolver(function ($varPage = 'page') use ($sandbox) {
|
||||
|
||||
$page = $sandbox->getSnapshot()->request->param($varPage);
|
||||
|
||||
if (filter_var($page, FILTER_VALIDATE_INT) !== false && (int) $page >= 1) {
|
||||
return (int) $page;
|
||||
}
|
||||
|
||||
return 1;
|
||||
});
|
||||
}
|
||||
}
|
||||
33
src/resetters/ResetService.php
Normal file
33
src/resetters/ResetService.php
Normal file
@@ -0,0 +1,33 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\resetters;
|
||||
|
||||
use think\App;
|
||||
use think\worker\concerns\ModifyProperty;
|
||||
use think\worker\contract\ResetterInterface;
|
||||
use think\worker\Sandbox;
|
||||
|
||||
class ResetService implements ResetterInterface
|
||||
{
|
||||
use ModifyProperty;
|
||||
|
||||
/**
|
||||
* "handle" function for resetting app.
|
||||
*
|
||||
* @param App $app
|
||||
* @param Sandbox $sandbox
|
||||
*/
|
||||
public function handle(App $app, Sandbox $sandbox)
|
||||
{
|
||||
foreach ($sandbox->getServices() as $service) {
|
||||
$this->modifyProperty($service, $app);
|
||||
if (method_exists($service, 'register')) {
|
||||
$service->register();
|
||||
}
|
||||
if (method_exists($service, 'boot')) {
|
||||
$app->invoke([$service, 'boot']);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
112
src/response/File.php
Normal file
112
src/response/File.php
Normal file
@@ -0,0 +1,112 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\response;
|
||||
|
||||
use DateTime;
|
||||
use RuntimeException;
|
||||
use SplFileInfo;
|
||||
use think\Response;
|
||||
|
||||
class File extends Response
|
||||
{
|
||||
public const DISPOSITION_ATTACHMENT = 'attachment';
|
||||
public const DISPOSITION_INLINE = 'inline';
|
||||
|
||||
protected $header = [
|
||||
'Content-Type' => 'application/octet-stream',
|
||||
'Accept-Ranges' => 'bytes',
|
||||
];
|
||||
|
||||
/**
|
||||
* @var SplFileInfo
|
||||
*/
|
||||
protected $file;
|
||||
|
||||
public function __construct($file, ?string $contentDisposition = null, bool $autoEtag = true, bool $autoLastModified = true, bool $autoContentType = true)
|
||||
{
|
||||
$this->setFile($file, $contentDisposition, $autoEtag, $autoLastModified, $autoContentType);
|
||||
}
|
||||
|
||||
public function getFile()
|
||||
{
|
||||
return $this->file;
|
||||
}
|
||||
|
||||
public function setFile($file, ?string $contentDisposition = null, bool $autoEtag = true, bool $autoLastModified = true, bool $autoContentType = true)
|
||||
{
|
||||
if (!$file instanceof SplFileInfo) {
|
||||
$file = new SplFileInfo((string) $file);
|
||||
}
|
||||
|
||||
if (!$file->isReadable()) {
|
||||
throw new RuntimeException('File must be readable.');
|
||||
}
|
||||
|
||||
$this->header['Content-Length'] = $file->getSize();
|
||||
|
||||
$this->file = $file;
|
||||
|
||||
if ($autoEtag) {
|
||||
$this->setAutoEtag();
|
||||
}
|
||||
|
||||
if ($autoLastModified) {
|
||||
$this->setAutoLastModified();
|
||||
}
|
||||
|
||||
if ($contentDisposition) {
|
||||
$this->setContentDisposition($contentDisposition);
|
||||
}
|
||||
|
||||
if ($autoContentType) {
|
||||
$this->setAutoContentType();
|
||||
}
|
||||
|
||||
return $this;
|
||||
}
|
||||
|
||||
public function setAutoContentType()
|
||||
{
|
||||
if (extension_loaded('fileinfo')) {
|
||||
$finfo = finfo_open(FILEINFO_MIME_TYPE);
|
||||
|
||||
$mimeType = finfo_file($finfo, $this->file->getPathname());
|
||||
if ($mimeType) {
|
||||
$this->header['Content-Type'] = $mimeType;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public function setContentDisposition(string $disposition, string $filename = '')
|
||||
{
|
||||
if ('' === $filename) {
|
||||
$filename = $this->file->getFilename();
|
||||
}
|
||||
|
||||
$this->header['Content-Disposition'] = "{$disposition}; filename=\"{$filename}\"";
|
||||
|
||||
return $this;
|
||||
}
|
||||
|
||||
public function setAutoLastModified()
|
||||
{
|
||||
$mTime = $this->file->getMTime();
|
||||
if ($mTime) {
|
||||
$date = DateTime::createFromFormat('U', (string) $mTime);
|
||||
$this->lastModified($date->format('D, d M Y H:i:s') . ' GMT');
|
||||
}
|
||||
return $this;
|
||||
}
|
||||
|
||||
public function setAutoEtag()
|
||||
{
|
||||
$eTag = "W/\"" . sha1_file($this->file->getPathname()) . "\"";
|
||||
|
||||
return $this->eTag($eTag);
|
||||
}
|
||||
|
||||
protected function sendData(string $data): void
|
||||
{
|
||||
readfile($this->file->getPathname());
|
||||
}
|
||||
}
|
||||
22
src/response/Iterator.php
Normal file
22
src/response/Iterator.php
Normal file
@@ -0,0 +1,22 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\response;
|
||||
|
||||
use IteratorAggregate;
|
||||
use think\Response;
|
||||
use Traversable;
|
||||
|
||||
class Iterator extends Response implements IteratorAggregate
|
||||
{
|
||||
protected $iterator;
|
||||
|
||||
public function __construct(Traversable $iterator)
|
||||
{
|
||||
$this->iterator = $iterator;
|
||||
}
|
||||
|
||||
public function getIterator(): Traversable
|
||||
{
|
||||
return $this->iterator;
|
||||
}
|
||||
}
|
||||
66
src/response/Websocket.php
Normal file
66
src/response/Websocket.php
Normal file
@@ -0,0 +1,66 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\response;
|
||||
|
||||
use think\Event;
|
||||
use think\Response;
|
||||
|
||||
class Websocket extends Response
|
||||
{
|
||||
protected $listeners = [];
|
||||
|
||||
public function onOpen($listener)
|
||||
{
|
||||
$this->listeners['Open'] = $listener;
|
||||
return $this;
|
||||
}
|
||||
|
||||
public function onMessage($listener)
|
||||
{
|
||||
$this->listeners['Message'] = $listener;
|
||||
return $this;
|
||||
}
|
||||
|
||||
public function onEvent($listener)
|
||||
{
|
||||
$this->listeners['Event'] = $listener;
|
||||
return $this;
|
||||
}
|
||||
|
||||
public function onClose($listener)
|
||||
{
|
||||
$this->listeners['Close'] = $listener;
|
||||
return $this;
|
||||
}
|
||||
|
||||
public function onConnect($listener)
|
||||
{
|
||||
$this->listeners['Connect'] = $listener;
|
||||
return $this;
|
||||
}
|
||||
|
||||
public function onDisconnect($listener)
|
||||
{
|
||||
$this->listeners['Disconnect'] = $listener;
|
||||
return $this;
|
||||
}
|
||||
|
||||
public function onPing($listener)
|
||||
{
|
||||
$this->listeners['Ping'] = $listener;
|
||||
return $this;
|
||||
}
|
||||
|
||||
public function onPong($listener)
|
||||
{
|
||||
$this->listeners['Pong'] = $listener;
|
||||
return $this;
|
||||
}
|
||||
|
||||
public function subscribe(Event $event)
|
||||
{
|
||||
foreach ($this->listeners as $eventName => $listener) {
|
||||
$event->listen('worker.websocket.' . $eventName, $listener);
|
||||
}
|
||||
}
|
||||
}
|
||||
8
src/watcher/Driver.php
Normal file
8
src/watcher/Driver.php
Normal file
@@ -0,0 +1,8 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\watcher;
|
||||
|
||||
interface Driver
|
||||
{
|
||||
public function watch(callable $callback);
|
||||
}
|
||||
72
src/watcher/Find.php
Normal file
72
src/watcher/Find.php
Normal file
@@ -0,0 +1,72 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\watcher;
|
||||
|
||||
use Symfony\Component\Process\Process;
|
||||
use Workerman\Timer;
|
||||
|
||||
class Find implements Driver
|
||||
{
|
||||
protected $name;
|
||||
protected $directory;
|
||||
protected $exclude;
|
||||
|
||||
public function __construct($directory, $exclude, $name)
|
||||
{
|
||||
$this->directory = $directory;
|
||||
$this->exclude = $exclude;
|
||||
$this->name = $name;
|
||||
}
|
||||
|
||||
public function watch(callable $callback)
|
||||
{
|
||||
$ms = 2000;
|
||||
$seconds = ceil(($ms + 1000) / 1000);
|
||||
$minutes = sprintf('-%.2f', $seconds / 60);
|
||||
|
||||
$dest = implode(' ', $this->directory);
|
||||
|
||||
$name = empty($this->name) ? '' : ' \( ' . join(' -o ', array_map(fn($v) => "-name \"{$v}\"", $this->name)) . ' \)';
|
||||
$notName = '';
|
||||
$notPath = '';
|
||||
if (!empty($this->exclude)) {
|
||||
$excludeDirs = $excludeFiles = [];
|
||||
foreach ($this->exclude as $directory) {
|
||||
$directory = rtrim($directory, '/');
|
||||
if (is_dir($directory)) {
|
||||
$excludeDirs[] = $directory;
|
||||
} else {
|
||||
$excludeFiles[] = $directory;
|
||||
}
|
||||
}
|
||||
|
||||
if (!empty($excludeFiles)) {
|
||||
$notPath = ' -not \( ' . join(' -and ', array_map(fn($v) => "-name \"{$v}\"", $excludeFiles)) . ' \)';
|
||||
}
|
||||
|
||||
if (!empty($excludeDirs)) {
|
||||
$notPath = ' -not \( ' . join(' -and ', array_map(fn($v) => "-path \"{$v}/*\"", $excludeDirs)) . ' \)';
|
||||
}
|
||||
}
|
||||
|
||||
$command = "find {$dest}{$name}{$notName}{$notPath} -mmin {$minutes} -type f -print";
|
||||
|
||||
Timer::add($ms / 1000, function () use ($callback, $command) {
|
||||
$stdout = $this->exec($command);
|
||||
if (!empty($stdout)) {
|
||||
call_user_func($callback);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
public function exec($command)
|
||||
{
|
||||
$process = Process::fromShellCommandline($command);
|
||||
$process->run();
|
||||
if ($process->isSuccessful()) {
|
||||
return $process->getOutput();
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
}
|
||||
52
src/watcher/Scan.php
Normal file
52
src/watcher/Scan.php
Normal file
@@ -0,0 +1,52 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\watcher;
|
||||
|
||||
use Symfony\Component\Finder\Finder;
|
||||
use Symfony\Component\Finder\SplFileInfo;
|
||||
use Workerman\Timer;
|
||||
|
||||
class Scan implements Driver
|
||||
{
|
||||
protected $finder;
|
||||
|
||||
protected $files = [];
|
||||
|
||||
public function __construct($directory, $exclude, $name)
|
||||
{
|
||||
$this->finder = new Finder();
|
||||
$this->finder
|
||||
->files()
|
||||
->name($name)
|
||||
->in($directory)
|
||||
->exclude($exclude);
|
||||
}
|
||||
|
||||
protected function findFiles()
|
||||
{
|
||||
$files = [];
|
||||
/** @var SplFileInfo $f */
|
||||
foreach ($this->finder as $f) {
|
||||
$files[$f->getRealpath()] = $f->getMTime();
|
||||
}
|
||||
return $files;
|
||||
}
|
||||
|
||||
public function watch(callable $callback)
|
||||
{
|
||||
$this->files = $this->findFiles();
|
||||
|
||||
Timer::add(2, function () use ($callback) {
|
||||
$files = $this->findFiles();
|
||||
|
||||
foreach ($files as $path => $time) {
|
||||
if (empty($this->files[$path]) || $this->files[$path] != $time) {
|
||||
call_user_func($callback);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
$this->files = $files;
|
||||
});
|
||||
}
|
||||
}
|
||||
15
src/websocket/Event.php
Normal file
15
src/websocket/Event.php
Normal file
@@ -0,0 +1,15 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\websocket;
|
||||
|
||||
class Event
|
||||
{
|
||||
public $type;
|
||||
public $data;
|
||||
|
||||
public function __construct($type, $data = null)
|
||||
{
|
||||
$this->type = $type;
|
||||
$this->data = $data;
|
||||
}
|
||||
}
|
||||
10
src/websocket/Frame.php
Normal file
10
src/websocket/Frame.php
Normal file
@@ -0,0 +1,10 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\websocket;
|
||||
|
||||
class Frame
|
||||
{
|
||||
public function __construct(public int $fd, public string $data)
|
||||
{
|
||||
}
|
||||
}
|
||||
71
src/websocket/Handler.php
Normal file
71
src/websocket/Handler.php
Normal file
@@ -0,0 +1,71 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\websocket;
|
||||
|
||||
use think\Event;
|
||||
use think\Request;
|
||||
use think\worker\contract\websocket\HandlerInterface;
|
||||
use think\worker\websocket\Event as WsEvent;
|
||||
|
||||
class Handler implements HandlerInterface
|
||||
{
|
||||
protected $event;
|
||||
|
||||
public function __construct(Event $event)
|
||||
{
|
||||
$this->event = $event;
|
||||
}
|
||||
|
||||
/**
|
||||
* "onOpen" listener.
|
||||
*
|
||||
* @param Request $request
|
||||
*/
|
||||
public function onOpen(Request $request)
|
||||
{
|
||||
$this->event->trigger('worker.websocket.Open', $request);
|
||||
}
|
||||
|
||||
/**
|
||||
* "onMessage" listener.
|
||||
*
|
||||
* @param Frame $frame
|
||||
*/
|
||||
public function onMessage(Frame $frame)
|
||||
{
|
||||
$this->event->trigger('worker.websocket.Message', $frame);
|
||||
|
||||
$event = $this->decode($frame->data);
|
||||
if ($event) {
|
||||
$this->event->trigger('worker.websocket.Event', $event);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* "onClose" listener.
|
||||
*/
|
||||
public function onClose()
|
||||
{
|
||||
$this->event->trigger('worker.websocket.Close');
|
||||
}
|
||||
|
||||
protected function decode($payload)
|
||||
{
|
||||
$data = json_decode($payload, true);
|
||||
if (!empty($data['type'])) {
|
||||
return new WsEvent($data['type'], $data['data'] ?? null);
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
public function encodeMessage($message)
|
||||
{
|
||||
if ($message instanceof WsEvent) {
|
||||
return json_encode([
|
||||
'type' => $message->type,
|
||||
'data' => $message->data,
|
||||
]);
|
||||
}
|
||||
return $message;
|
||||
}
|
||||
}
|
||||
67
src/websocket/Pusher.php
Normal file
67
src/websocket/Pusher.php
Normal file
@@ -0,0 +1,67 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\websocket;
|
||||
|
||||
use think\worker\Manager;
|
||||
use think\worker\message\PushMessage;
|
||||
|
||||
/**
|
||||
* Class Pusher
|
||||
*/
|
||||
class Pusher
|
||||
{
|
||||
|
||||
/** @var Room */
|
||||
protected $room;
|
||||
|
||||
/** @var Manager */
|
||||
protected $manager;
|
||||
|
||||
protected $to = [];
|
||||
|
||||
public function __construct(Manager $manager, Room $room)
|
||||
{
|
||||
$this->manager = $manager;
|
||||
$this->room = $room;
|
||||
}
|
||||
|
||||
public function to(...$values)
|
||||
{
|
||||
foreach ($values as $value) {
|
||||
if (is_array($value)) {
|
||||
$this->to(...$value);
|
||||
} elseif (!in_array($value, $this->to)) {
|
||||
$this->to[] = $value;
|
||||
}
|
||||
}
|
||||
|
||||
return $this;
|
||||
}
|
||||
|
||||
/**
|
||||
* Push message to related descriptors
|
||||
* @param $data
|
||||
* @return void
|
||||
*/
|
||||
public function push($data): void
|
||||
{
|
||||
$fds = [];
|
||||
|
||||
foreach ($this->to as $room) {
|
||||
$clients = $this->room->getClients($room);
|
||||
if (!empty($clients)) {
|
||||
$fds = array_merge($fds, $clients);
|
||||
}
|
||||
}
|
||||
|
||||
foreach (array_unique($fds) as $fd) {
|
||||
[$workerId, $fd] = explode('.', $fd);
|
||||
$this->manager->sendMessage((int) $workerId, new PushMessage((int) $fd, $data));
|
||||
}
|
||||
}
|
||||
|
||||
public function emit(string $event, ...$data): void
|
||||
{
|
||||
$this->push(new Event($event, $data));
|
||||
}
|
||||
}
|
||||
54
src/websocket/Room.php
Normal file
54
src/websocket/Room.php
Normal file
@@ -0,0 +1,54 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\websocket;
|
||||
|
||||
use think\worker\Conduit;
|
||||
|
||||
class Room
|
||||
{
|
||||
|
||||
public function __construct(protected Conduit $conduit)
|
||||
{
|
||||
|
||||
}
|
||||
|
||||
public function add($fd, ...$rooms)
|
||||
{
|
||||
$this->conduit->sAdd($this->getClientKey($fd), ...$rooms);
|
||||
|
||||
foreach ($rooms as $room) {
|
||||
$this->conduit->sAdd($this->getRoomKey($room), $fd);
|
||||
}
|
||||
}
|
||||
|
||||
public function delete($fd, ...$rooms)
|
||||
{
|
||||
$rooms = count($rooms) ? $rooms : $this->getRooms($fd);
|
||||
|
||||
$this->conduit->sRem($this->getClientKey($fd), ...$rooms);
|
||||
|
||||
foreach ($rooms as $room) {
|
||||
$this->conduit->sRem($this->getRoomKey($room), $fd);
|
||||
}
|
||||
}
|
||||
|
||||
public function getClients(string $room)
|
||||
{
|
||||
return $this->conduit->sMembers($this->getRoomKey($room)) ?: [];
|
||||
}
|
||||
|
||||
public function getRooms(string $fd)
|
||||
{
|
||||
return $this->conduit->sMembers($this->getClientKey($fd)) ?: [];
|
||||
}
|
||||
|
||||
protected function getClientKey(string $key)
|
||||
{
|
||||
return "ws:client:{$key}";
|
||||
}
|
||||
|
||||
protected function getRoomKey($room)
|
||||
{
|
||||
return "ws:room:{$room}";
|
||||
}
|
||||
}
|
||||
81
src/websocket/socketio/EnginePacket.php
Normal file
81
src/websocket/socketio/EnginePacket.php
Normal file
@@ -0,0 +1,81 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\websocket\socketio;
|
||||
|
||||
class EnginePacket
|
||||
{
|
||||
/**
|
||||
* Engine.io packet type `open`.
|
||||
*/
|
||||
const OPEN = 0;
|
||||
|
||||
/**
|
||||
* Engine.io packet type `close`.
|
||||
*/
|
||||
const CLOSE = 1;
|
||||
|
||||
/**
|
||||
* Engine.io packet type `ping`.
|
||||
*/
|
||||
const PING = 2;
|
||||
|
||||
/**
|
||||
* Engine.io packet type `pong`.
|
||||
*/
|
||||
const PONG = 3;
|
||||
|
||||
/**
|
||||
* Engine.io packet type `message`.
|
||||
*/
|
||||
const MESSAGE = 4;
|
||||
|
||||
/**
|
||||
* Engine.io packet type 'upgrade'
|
||||
*/
|
||||
const UPGRADE = 5;
|
||||
|
||||
/**
|
||||
* Engine.io packet type `noop`.
|
||||
*/
|
||||
const NOOP = 6;
|
||||
|
||||
public $type;
|
||||
|
||||
public $data = '';
|
||||
|
||||
public function __construct($type, $data = '')
|
||||
{
|
||||
$this->type = $type;
|
||||
$this->data = $data;
|
||||
}
|
||||
|
||||
public static function open($payload)
|
||||
{
|
||||
return new self(self::OPEN, $payload);
|
||||
}
|
||||
|
||||
public static function pong($payload = '')
|
||||
{
|
||||
return new self(self::PONG, $payload);
|
||||
}
|
||||
|
||||
public static function ping()
|
||||
{
|
||||
return new self(self::PING);
|
||||
}
|
||||
|
||||
public static function message($payload)
|
||||
{
|
||||
return new self(self::MESSAGE, $payload);
|
||||
}
|
||||
|
||||
public static function fromString(string $packet)
|
||||
{
|
||||
return new self(substr($packet, 0, 1), substr($packet, 1) ?: '');
|
||||
}
|
||||
|
||||
public function toString()
|
||||
{
|
||||
return $this->type . $this->data;
|
||||
}
|
||||
}
|
||||
198
src/websocket/socketio/Handler.php
Normal file
198
src/websocket/socketio/Handler.php
Normal file
@@ -0,0 +1,198 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\websocket\socketio;
|
||||
|
||||
use Exception;
|
||||
|
||||
use think\Config;
|
||||
use think\Event;
|
||||
use think\Request;
|
||||
use think\worker\contract\websocket\HandlerInterface;
|
||||
use think\worker\Websocket;
|
||||
use think\worker\websocket\Event as WsEvent;
|
||||
use think\worker\websocket\Frame;
|
||||
use Workerman\Timer;
|
||||
|
||||
class Handler implements HandlerInterface
|
||||
{
|
||||
/** @var Config */
|
||||
protected $config;
|
||||
|
||||
protected $event;
|
||||
|
||||
protected $websocket;
|
||||
|
||||
protected $eio;
|
||||
|
||||
protected $pingTimeoutTimer = 0;
|
||||
protected $pingIntervalTimer = 0;
|
||||
|
||||
protected $pingInterval;
|
||||
protected $pingTimeout;
|
||||
|
||||
public function __construct(Event $event, Config $config, Websocket $websocket)
|
||||
{
|
||||
$this->event = $event;
|
||||
$this->config = $config;
|
||||
$this->websocket = $websocket;
|
||||
$this->pingInterval = $this->config->get('worker.websocket.ping_interval', 25000);
|
||||
$this->pingTimeout = $this->config->get('worker.websocket.ping_timeout', 60000);
|
||||
}
|
||||
|
||||
/**
|
||||
* "onOpen" listener.
|
||||
*
|
||||
* @param Request $request
|
||||
*/
|
||||
public function onOpen(Request $request)
|
||||
{
|
||||
$this->eio = $request->param('EIO');
|
||||
|
||||
$payload = json_encode(
|
||||
[
|
||||
'sid' => base64_encode(uniqid()),
|
||||
'upgrades' => [],
|
||||
'pingInterval' => $this->pingInterval,
|
||||
'pingTimeout' => $this->pingTimeout,
|
||||
]
|
||||
);
|
||||
|
||||
$this->push(EnginePacket::open($payload));
|
||||
|
||||
$this->event->trigger('worker.websocket.Open', $request);
|
||||
|
||||
if ($this->eio < 4) {
|
||||
$this->resetPingTimeout($this->pingInterval + $this->pingTimeout);
|
||||
$this->onConnect();
|
||||
} else {
|
||||
$this->schedulePing();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* "onMessage" listener.
|
||||
*
|
||||
* @param Frame $frame
|
||||
*/
|
||||
public function onMessage(Frame $frame)
|
||||
{
|
||||
$enginePacket = EnginePacket::fromString($frame->data);
|
||||
|
||||
$this->event->trigger('worker.websocket.Message', $enginePacket);
|
||||
|
||||
$this->resetPingTimeout($this->pingInterval + $this->pingTimeout);
|
||||
|
||||
switch ($enginePacket->type) {
|
||||
case EnginePacket::MESSAGE:
|
||||
$packet = Packet::fromString($enginePacket->data);
|
||||
switch ($packet->type) {
|
||||
case Packet::CONNECT:
|
||||
$this->onConnect($packet->data);
|
||||
break;
|
||||
case Packet::EVENT:
|
||||
$type = array_shift($packet->data);
|
||||
$data = $packet->data;
|
||||
$result = $this->event->trigger('worker.websocket.Event', new WsEvent($type, $data));
|
||||
|
||||
if ($packet->id !== null) {
|
||||
$responsePacket = Packet::create(Packet::ACK, [
|
||||
'id' => $packet->id,
|
||||
'nsp' => $packet->nsp,
|
||||
'data' => $result,
|
||||
]);
|
||||
|
||||
$this->push($responsePacket);
|
||||
}
|
||||
break;
|
||||
case Packet::DISCONNECT:
|
||||
$this->event->trigger('worker.websocket.Disconnect');
|
||||
$this->websocket->close();
|
||||
break;
|
||||
default:
|
||||
$this->websocket->close();
|
||||
break;
|
||||
}
|
||||
break;
|
||||
case EnginePacket::PING:
|
||||
$this->event->trigger('worker.websocket.Ping');
|
||||
$this->push(EnginePacket::pong($enginePacket->data));
|
||||
break;
|
||||
case EnginePacket::PONG:
|
||||
$this->event->trigger('worker.websocket.Pong');
|
||||
$this->schedulePing();
|
||||
break;
|
||||
default:
|
||||
$this->websocket->close();
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* "onClose" listener.
|
||||
*/
|
||||
public function onClose()
|
||||
{
|
||||
Timer::del($this->pingTimeoutTimer);
|
||||
Timer::del($this->pingIntervalTimer);
|
||||
$this->event->trigger('worker.websocket.Close');
|
||||
}
|
||||
|
||||
protected function onConnect($data = null)
|
||||
{
|
||||
try {
|
||||
$this->event->trigger('worker.websocket.Connect', $data);
|
||||
$packet = Packet::create(Packet::CONNECT);
|
||||
if ($this->eio >= 4) {
|
||||
$packet->data = ['sid' => base64_encode(uniqid())];
|
||||
}
|
||||
} catch (Exception $exception) {
|
||||
$packet = Packet::create(Packet::CONNECT_ERROR, [
|
||||
'data' => ['message' => $exception->getMessage()],
|
||||
]);
|
||||
}
|
||||
|
||||
$this->push($packet);
|
||||
}
|
||||
|
||||
protected function resetPingTimeout($timeout)
|
||||
{
|
||||
Timer::del($this->pingTimeoutTimer);
|
||||
$this->pingTimeoutTimer = Timer::delay($timeout, function () {
|
||||
$this->websocket->close();
|
||||
});
|
||||
}
|
||||
|
||||
protected function schedulePing()
|
||||
{
|
||||
Timer::del($this->pingIntervalTimer);
|
||||
$this->pingIntervalTimer = Timer::delay($this->pingInterval, function () {
|
||||
$this->push(EnginePacket::ping());
|
||||
$this->resetPingTimeout($this->pingTimeout);
|
||||
});
|
||||
}
|
||||
|
||||
public function encodeMessage($message)
|
||||
{
|
||||
if ($message instanceof WsEvent) {
|
||||
$message = Packet::create(Packet::EVENT, [
|
||||
'data' => array_merge([$message->type], $message->data),
|
||||
]);
|
||||
}
|
||||
|
||||
if ($message instanceof Packet) {
|
||||
$message = EnginePacket::message($message->toString());
|
||||
}
|
||||
|
||||
if ($message instanceof EnginePacket) {
|
||||
$message = $message->toString();
|
||||
}
|
||||
|
||||
return $message;
|
||||
}
|
||||
|
||||
protected function push($data)
|
||||
{
|
||||
$this->websocket->push($data);
|
||||
}
|
||||
|
||||
}
|
||||
134
src/websocket/socketio/Packet.php
Normal file
134
src/websocket/socketio/Packet.php
Normal file
@@ -0,0 +1,134 @@
|
||||
<?php
|
||||
|
||||
namespace think\worker\websocket\socketio;
|
||||
|
||||
/**
|
||||
* Class Packet
|
||||
*/
|
||||
class Packet
|
||||
{
|
||||
/**
|
||||
* Socket.io packet type `connect`.
|
||||
*/
|
||||
const CONNECT = 0;
|
||||
|
||||
/**
|
||||
* Socket.io packet type `disconnect`.
|
||||
*/
|
||||
const DISCONNECT = 1;
|
||||
|
||||
/**
|
||||
* Socket.io packet type `event`.
|
||||
*/
|
||||
const EVENT = 2;
|
||||
|
||||
/**
|
||||
* Socket.io packet type `ack`.
|
||||
*/
|
||||
const ACK = 3;
|
||||
|
||||
/**
|
||||
* Socket.io packet type `connect_error`.
|
||||
*/
|
||||
const CONNECT_ERROR = 4;
|
||||
|
||||
/**
|
||||
* Socket.io packet type 'binary event'
|
||||
*/
|
||||
const BINARY_EVENT = 5;
|
||||
|
||||
/**
|
||||
* Socket.io packet type `binary ack`. For acks with binary arguments.
|
||||
*/
|
||||
const BINARY_ACK = 6;
|
||||
|
||||
public $type;
|
||||
public $nsp = '/';
|
||||
public $data = null;
|
||||
public $id = null;
|
||||
|
||||
public function __construct(int $type)
|
||||
{
|
||||
$this->type = $type;
|
||||
}
|
||||
|
||||
public static function create($type, array $decoded = [])
|
||||
{
|
||||
$new = new self($type);
|
||||
$new->id = $decoded['id'] ?? null;
|
||||
if (isset($decoded['nsp'])) {
|
||||
$new->nsp = $decoded['nsp'] ?: '/';
|
||||
} else {
|
||||
$new->nsp = '/';
|
||||
}
|
||||
$new->data = $decoded['data'] ?? null;
|
||||
return $new;
|
||||
}
|
||||
|
||||
public function toString()
|
||||
{
|
||||
$str = '' . $this->type;
|
||||
if ($this->nsp && '/' !== $this->nsp) {
|
||||
$str .= $this->nsp . ',';
|
||||
}
|
||||
|
||||
if ($this->id !== null) {
|
||||
$str .= $this->id;
|
||||
}
|
||||
|
||||
if (null !== $this->data) {
|
||||
$str .= json_encode($this->data, JSON_UNESCAPED_UNICODE | JSON_UNESCAPED_SLASHES);
|
||||
}
|
||||
return $str;
|
||||
}
|
||||
|
||||
public static function fromString(string $str)
|
||||
{
|
||||
$i = 0;
|
||||
|
||||
$packet = new Packet((int) substr($str, 0, 1));
|
||||
|
||||
// look up namespace (if any)
|
||||
if ('/' === substr($str, $i + 1, 1)) {
|
||||
$nsp = '';
|
||||
while (++$i) {
|
||||
$c = substr($str, $i, 1);
|
||||
if (',' === $c) {
|
||||
break;
|
||||
}
|
||||
$nsp .= $c;
|
||||
if ($i === strlen($str)) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
$packet->nsp = $nsp;
|
||||
} else {
|
||||
$packet->nsp = '/';
|
||||
}
|
||||
|
||||
// look up id
|
||||
$next = substr($str, $i + 1, 1);
|
||||
if ('' !== $next && is_numeric($next)) {
|
||||
$id = '';
|
||||
while (++$i) {
|
||||
$c = substr($str, $i, 1);
|
||||
if (null == $c || !is_numeric($c)) {
|
||||
--$i;
|
||||
break;
|
||||
}
|
||||
$id .= substr($str, $i, 1);
|
||||
if ($i === strlen($str)) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
$packet->id = intval($id);
|
||||
}
|
||||
|
||||
// look up json data
|
||||
if (substr($str, ++$i, 1)) {
|
||||
$packet->data = json_decode(substr($str, $i), true);
|
||||
}
|
||||
|
||||
return $packet;
|
||||
}
|
||||
}
|
||||
2
tests/Pest.php
Normal file
2
tests/Pest.php
Normal file
@@ -0,0 +1,2 @@
|
||||
<?php
|
||||
define('STUB_DIR', realpath(__DIR__ . '/stub'));
|
||||
153
tests/feature/HttpTest.php
Normal file
153
tests/feature/HttpTest.php
Normal file
@@ -0,0 +1,153 @@
|
||||
<?php
|
||||
|
||||
use GuzzleHttp\Client;
|
||||
use GuzzleHttp\Cookie\CookieJar;
|
||||
use Symfony\Component\Process\Process;
|
||||
|
||||
$process = null;
|
||||
beforeAll(function () use (&$process) {
|
||||
$process = new Process(['php', 'think', 'worker'], STUB_DIR, [
|
||||
'PHP_WEBSOCKET_ENABLE' => 'false',
|
||||
'PHP_QUEUE_ENABLE' => 'false',
|
||||
]);
|
||||
$process->start();
|
||||
$wait = 0;
|
||||
|
||||
while (!$process->getOutput()) {
|
||||
$wait++;
|
||||
if ($wait > 30) {
|
||||
throw new Exception('server start failed');
|
||||
}
|
||||
sleep(1);
|
||||
}
|
||||
});
|
||||
|
||||
afterAll(function () use (&$process) {
|
||||
echo $process->getOutput();
|
||||
$process->stop();
|
||||
});
|
||||
|
||||
beforeEach(function () {
|
||||
$this->httpClient = new Client([
|
||||
'base_uri' => 'http://127.0.0.1:8080',
|
||||
'cookies' => true,
|
||||
'http_errors' => false,
|
||||
'timeout' => 1,
|
||||
]);
|
||||
});
|
||||
|
||||
it('callback route', function () {
|
||||
$response = $this->httpClient->get('/');
|
||||
|
||||
expect($response->getStatusCode())
|
||||
->toBe(200)
|
||||
->and($response->getBody()->getContents())
|
||||
->toBe('hello world');
|
||||
});
|
||||
|
||||
it('controller route', function () {
|
||||
$jar = new CookieJar();
|
||||
|
||||
$response = $this->httpClient->get('/test', ['cookies' => $jar]);
|
||||
|
||||
expect($response->getStatusCode())
|
||||
->toBe(200)
|
||||
->and($response->getBody()->getContents())
|
||||
->toBe('test')
|
||||
->and($jar->getCookieByName('name')->getValue())
|
||||
->toBe('think');
|
||||
});
|
||||
|
||||
it('json post', function () {
|
||||
|
||||
$data = [
|
||||
'name' => 'think',
|
||||
];
|
||||
$response = $this->httpClient->post('/json', [
|
||||
'json' => $data,
|
||||
]);
|
||||
|
||||
expect($response->getStatusCode())
|
||||
->toBe(200)
|
||||
->and($response->getBody()->getContents())
|
||||
->toBe(json_encode($data));
|
||||
});
|
||||
|
||||
it('put and delete request', function () {
|
||||
$response = $this->httpClient->put('/');
|
||||
|
||||
expect($response->getStatusCode())
|
||||
->toBe(200)
|
||||
->and($response->getBody()->getContents())
|
||||
->toBe('put');
|
||||
|
||||
$response = $this->httpClient->delete('/');
|
||||
|
||||
expect($response->getStatusCode())
|
||||
->toBe(200)
|
||||
->and($response->getBody()->getContents())
|
||||
->toBe('delete');
|
||||
});
|
||||
|
||||
it('file response', function () {
|
||||
$response = $this->httpClient->get('/static/asset.txt');
|
||||
|
||||
expect($response->getStatusCode())
|
||||
->toBe(200)
|
||||
->and($response->getBody()->getContents())
|
||||
->toBe(file_get_contents(STUB_DIR . '/public/asset.txt'));
|
||||
});
|
||||
|
||||
it('sse', function () {
|
||||
$response = $this->httpClient->get('/sse', [
|
||||
'stream' => true,
|
||||
'timeout' => 3,
|
||||
]);
|
||||
|
||||
$body = $response->getBody();
|
||||
|
||||
$buffer = '';
|
||||
while (!$body->eof()) {
|
||||
$text = $body->read(1);
|
||||
if ($text == "\r") {
|
||||
continue;
|
||||
}
|
||||
$buffer .= $text;
|
||||
if ($text == "\n") {
|
||||
if ($buffer != "\n") {
|
||||
expect($buffer)->toStartWith('data: ');
|
||||
}
|
||||
$buffer = '';
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
it('hot update', function () {
|
||||
$response = $this->httpClient->get('/hot');
|
||||
|
||||
expect($response->getStatusCode())
|
||||
->toBe(404);
|
||||
|
||||
$route = <<<PHP
|
||||
<?php
|
||||
|
||||
use think\\facade\\Route;
|
||||
|
||||
Route::get('/hot', function () {
|
||||
return 'hot';
|
||||
});
|
||||
PHP;
|
||||
|
||||
file_put_contents(STUB_DIR . '/route/hot.php', $route);
|
||||
|
||||
sleep(2);
|
||||
|
||||
$response = $this->httpClient->get('/hot');
|
||||
|
||||
expect($response->getStatusCode())
|
||||
->toBe(200)
|
||||
->and($response->getBody()->getContents())
|
||||
->toBe('hot');
|
||||
})->after(function () {
|
||||
@unlink(STUB_DIR . '/route/hot.php');
|
||||
})->skipOnWindows();
|
||||
5
tests/feature/QueueTest.php
Normal file
5
tests/feature/QueueTest.php
Normal file
@@ -0,0 +1,5 @@
|
||||
<?php
|
||||
|
||||
it('queue', function () {
|
||||
|
||||
});
|
||||
77
tests/feature/WebsocketTest.php
Normal file
77
tests/feature/WebsocketTest.php
Normal file
@@ -0,0 +1,77 @@
|
||||
<?php
|
||||
|
||||
use GuzzleHttp\Client;
|
||||
use React\EventLoop\Loop;
|
||||
use Symfony\Component\Process\Process;
|
||||
use function Ratchet\Client\connect;
|
||||
|
||||
$process = null;
|
||||
beforeAll(function () use (&$process) {
|
||||
$process = new Process(['php', 'think', 'worker'], STUB_DIR, [
|
||||
'PHP_WEBSOCKET_ENABLE' => 'true',
|
||||
'PHP_QUEUE_ENABLE' => 'false',
|
||||
'PHP_HOT_ENABLE' => 'false',
|
||||
]);
|
||||
$process->start();
|
||||
$wait = 0;
|
||||
|
||||
while (!$process->getOutput()) {
|
||||
$wait++;
|
||||
if ($wait > 30) {
|
||||
throw new Exception('server start failed');
|
||||
}
|
||||
sleep(1);
|
||||
}
|
||||
});
|
||||
|
||||
afterAll(function () use (&$process) {
|
||||
echo $process->getOutput();
|
||||
$process->stop();
|
||||
});
|
||||
|
||||
beforeEach(function () {
|
||||
$this->httpClient = new Client([
|
||||
'base_uri' => 'http://127.0.0.1:8080',
|
||||
'cookies' => true,
|
||||
'http_errors' => false,
|
||||
'timeout' => 1,
|
||||
]);
|
||||
});
|
||||
|
||||
it('http', function () {
|
||||
$response = $this->httpClient->get('/');
|
||||
|
||||
expect($response->getStatusCode())
|
||||
->toBe(200)
|
||||
->and($response->getBody()->getContents())
|
||||
->toBe('hello world');
|
||||
});
|
||||
|
||||
it('websocket', function () {
|
||||
$connected = 0;
|
||||
$messages = [];
|
||||
connect('ws://127.0.0.1:8080/websocket')
|
||||
->then(function (\Ratchet\Client\WebSocket $conn) use (&$connected, &$messages) {
|
||||
$connected++;
|
||||
$conn->on('message', function ($msg) use ($conn, &$messages) {
|
||||
$messages[] = (string) $msg;
|
||||
$conn->close();
|
||||
});
|
||||
});
|
||||
|
||||
connect('ws://127.0.0.1:8080/websocket')
|
||||
->then(function (\Ratchet\Client\WebSocket $conn) use (&$connected, &$messages) {
|
||||
$connected++;
|
||||
$conn->on('message', function ($msg) use ($conn, &$messages) {
|
||||
$messages[] = (string) $msg;
|
||||
$conn->close();
|
||||
});
|
||||
|
||||
$conn->send('hello');
|
||||
});
|
||||
|
||||
Loop::get()->run();
|
||||
|
||||
expect($connected)->toBe(2);
|
||||
expect($messages)->toBe(['hello', 'hello']);
|
||||
});
|
||||
1
tests/stub/.env
Normal file
1
tests/stub/.env
Normal file
@@ -0,0 +1 @@
|
||||
APP_DEBUG=true
|
||||
20
tests/stub/app/controller/Index.php
Normal file
20
tests/stub/app/controller/Index.php
Normal file
@@ -0,0 +1,20 @@
|
||||
<?php
|
||||
|
||||
namespace app\controller;
|
||||
|
||||
use think\facade\Cookie;
|
||||
use think\Request;
|
||||
|
||||
class Index
|
||||
{
|
||||
public function test()
|
||||
{
|
||||
Cookie::set('name', 'think');
|
||||
return 'test';
|
||||
}
|
||||
|
||||
public function json(Request $request)
|
||||
{
|
||||
return json($request->post());
|
||||
}
|
||||
}
|
||||
4
tests/stub/config/app.php
Normal file
4
tests/stub/config/app.php
Normal file
@@ -0,0 +1,4 @@
|
||||
<?php
|
||||
return [
|
||||
'error_message' => 'error',
|
||||
];
|
||||
9
tests/stub/config/cache.php
Normal file
9
tests/stub/config/cache.php
Normal file
@@ -0,0 +1,9 @@
|
||||
<?php
|
||||
return [
|
||||
'default' => 'file',
|
||||
'stores' => [
|
||||
'file' => [
|
||||
'type' => 'File',
|
||||
],
|
||||
]
|
||||
];
|
||||
40
tests/stub/config/queue.php
Normal file
40
tests/stub/config/queue.php
Normal file
@@ -0,0 +1,40 @@
|
||||
<?php
|
||||
// +----------------------------------------------------------------------
|
||||
// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2006-2016 http://thinkphp.cn All rights reserved.
|
||||
// +----------------------------------------------------------------------
|
||||
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
||||
// +----------------------------------------------------------------------
|
||||
// | Author: yunwuxin <448901948@qq.com>
|
||||
// +----------------------------------------------------------------------
|
||||
|
||||
return [
|
||||
'default' => 'redis',
|
||||
'connections' => [
|
||||
'sync' => [
|
||||
'type' => 'sync',
|
||||
],
|
||||
'database' => [
|
||||
'type' => 'database',
|
||||
'queue' => 'default',
|
||||
'table' => 'jobs',
|
||||
'connection' => null,
|
||||
],
|
||||
'redis' => [
|
||||
'type' => 'redis',
|
||||
'queue' => 'default',
|
||||
'host' => env('REDIS_HOST', 'redis'),
|
||||
'port' => env('REDIS_PORT', 6379),
|
||||
'password' => '',
|
||||
'select' => 0,
|
||||
'timeout' => 0,
|
||||
'persistent' => true,
|
||||
'retry_after' => 600,
|
||||
],
|
||||
],
|
||||
'failed' => [
|
||||
'type' => 'none',
|
||||
'table' => 'failed_jobs',
|
||||
],
|
||||
];
|
||||
45
tests/stub/config/worker.php
Normal file
45
tests/stub/config/worker.php
Normal file
@@ -0,0 +1,45 @@
|
||||
<?php
|
||||
// +----------------------------------------------------------------------
|
||||
// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2006-2018 http://thinkphp.cn All rights reserved.
|
||||
// +----------------------------------------------------------------------
|
||||
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
||||
// +----------------------------------------------------------------------
|
||||
// | Author: liu21st <liu21st@gmail.com>
|
||||
// +----------------------------------------------------------------------
|
||||
|
||||
use think\worker\websocket\Handler;
|
||||
|
||||
return [
|
||||
'http' => [
|
||||
'enable' => env('HTTP_ENABLE', true),
|
||||
'host' => '0.0.0.0',
|
||||
'port' => 8080,
|
||||
'worker_num' => 2,
|
||||
'options' => [],
|
||||
],
|
||||
'websocket' => [
|
||||
'enable' => env('WEBSOCKET_ENABLE', true),
|
||||
'handler' => Handler::class,
|
||||
'ping_interval' => 25000,
|
||||
'ping_timeout' => 60000,
|
||||
],
|
||||
//队列
|
||||
'queue' => [
|
||||
'enable' => env('QUEUE_ENABLE', true),
|
||||
'workers' => [
|
||||
'default' => [],
|
||||
],
|
||||
],
|
||||
//共享数据
|
||||
'conduit' => [
|
||||
'type' => 'socket',
|
||||
],
|
||||
'hot_update' => [
|
||||
'enable' => env('HOT_ENABLE', true),
|
||||
'name' => ['*.php'],
|
||||
'include' => [app_path(), config_path(), root_path('route')],
|
||||
'exclude' => [],
|
||||
],
|
||||
];
|
||||
1
tests/stub/public/asset.txt
Normal file
1
tests/stub/public/asset.txt
Normal file
@@ -0,0 +1 @@
|
||||
Asset
|
||||
51
tests/stub/route/app.php
Normal file
51
tests/stub/route/app.php
Normal file
@@ -0,0 +1,51 @@
|
||||
<?php
|
||||
|
||||
use think\facade\Route;
|
||||
|
||||
Route::get('/', function () {
|
||||
return 'hello world';
|
||||
});
|
||||
|
||||
Route::put('/', function () {
|
||||
return 'put';
|
||||
});
|
||||
|
||||
Route::delete('/', function () {
|
||||
return 'delete';
|
||||
});
|
||||
|
||||
Route::get('/sse', function () {
|
||||
|
||||
$generator = function () {
|
||||
foreach (range(0, 9) as $event) {
|
||||
yield 'data: ' . json_encode($event) . "\n\n";
|
||||
}
|
||||
|
||||
yield "data: [DONE]\n\n";
|
||||
};
|
||||
|
||||
$response = new \think\worker\response\Iterator($generator());
|
||||
|
||||
return $response->header([
|
||||
'Content-Type' => 'text/event-stream',
|
||||
'Cache-Control' => 'no-cache, must-revalidate',
|
||||
]);
|
||||
});
|
||||
|
||||
Route::get('/websocket', function () {
|
||||
return (new \think\worker\response\Websocket())
|
||||
->onOpen(function (\think\worker\Websocket $websocket) {
|
||||
$websocket->join('foo');
|
||||
})
|
||||
->onMessage(function (\think\worker\Websocket $websocket, \think\worker\websocket\Frame $frame) {
|
||||
$websocket->to('foo')->push($frame->data);
|
||||
});
|
||||
});
|
||||
|
||||
Route::get('test', 'index/test');
|
||||
Route::post('json', 'index/json');
|
||||
|
||||
Route::get('static/:path', function (string $path) {
|
||||
$filename = public_path() . $path;
|
||||
return new \think\worker\response\File($filename);
|
||||
})->pattern(['path' => '.*\.\w+$']);
|
||||
2
tests/stub/runtime/.gitignore
vendored
Normal file
2
tests/stub/runtime/.gitignore
vendored
Normal file
@@ -0,0 +1,2 @@
|
||||
*
|
||||
!.gitignore
|
||||
12
tests/stub/think
Normal file
12
tests/stub/think
Normal file
@@ -0,0 +1,12 @@
|
||||
#!/usr/bin/env php
|
||||
<?php
|
||||
|
||||
use think\App;
|
||||
|
||||
require __DIR__ . '/../../vendor/autoload.php';
|
||||
|
||||
$app = new App(__DIR__);
|
||||
|
||||
$app->console->addCommands([\think\worker\command\Server::class]);
|
||||
|
||||
$app->console->run();
|
||||
Reference in New Issue
Block a user