This commit is contained in:
2021-02-26 19:53:28 +08:00
parent cd3ed6937c
commit 6e3ecce453
4 changed files with 46 additions and 7 deletions
+3
View File
@@ -6,6 +6,7 @@ namespace HttpServer\Events;
use Exception; use Exception;
use HttpServer\Abstracts\Callback; use HttpServer\Abstracts\Callback;
use Snowflake\Snowflake;
use Swoole\Server; use Swoole\Server;
/** /**
@@ -25,6 +26,8 @@ class OnFinish extends Callback
$data = json_decode($data, true); $data = json_decode($data, true);
$data['work_id'] = $task_id; $data['work_id'] = $task_id;
$this->write(var_export($data, true), 'Task'); $this->write(var_export($data, true), 'Task');
Snowflake::app()->decrement();
} }
} }
+2 -1
View File
@@ -98,8 +98,9 @@ class OnTask extends Callback
*/ */
private function runTaskHandler($data): ?array private function runTaskHandler($data): ?array
{ {
$serialize = $this->before($data); Snowflake::app()->increment();
try { try {
$serialize = $this->before($data);
$params = $serialize->getParams(); $params = $serialize->getParams();
if (is_object($params)) { if (is_object($params)) {
$params = get_object_vars($params); $params = get_object_vars($params);
+4 -6
View File
@@ -22,6 +22,9 @@ class OnWorkerStart extends Callback
{ {
private int $_taskTable = 0;
/** /**
* @param Server $server * @param Server $server
* @param int $worker_id * @param int $worker_id
@@ -54,7 +57,6 @@ class OnWorkerStart extends Callback
* @param Server $server * @param Server $server
* @param int $workerId * @param int $workerId
* 异步任务管制 * 异步任务管制
* @throws ComponentException
*/ */
public function onTaskSignal(Server $server, int $workerId) public function onTaskSignal(Server $server, int $workerId)
{ {
@@ -63,11 +65,7 @@ class OnWorkerStart extends Callback
if ($sigkill === false) { if ($sigkill === false) {
return $server->stop($workerId); return $server->stop($workerId);
} }
while (Snowflake::app()->isRun()) {
$this->error(env('workerId') . '::' . $server->getWorkerStatus($workerId));
$this->error(env('workerId') . '::' . $server->stats()['coroutine_num']);
while ($server->getWorkerStatus($workerId) === SWOOLE_WORKER_BUSY) {
Coroutine::sleep(0.01); Coroutine::sleep(0.01);
} }
return $server->stop($workerId); return $server->stop($workerId);
+37
View File
@@ -50,6 +50,12 @@ abstract class BaseApplication extends Service
use TraitApplication; use TraitApplication;
private string $state = '';
private int $taskNumber = 0;
/** /**
* @var string * @var string
*/ */
@@ -78,6 +84,37 @@ abstract class BaseApplication extends Service
} }
/**
* @return bool
*/
public function isRun(): bool
{
return $this->state == 'SWOOLE_WORKER_BUSY';
}
public function decrement()
{
$this->taskNumber -= 1;
if ($this->taskNumber <= 0) {
$this->taskNumber = 0;
$this->state = 'SWOOLE_WORKER_IDLE';
}
}
public function increment()
{
$this->taskNumber += 1;
if ($this->taskNumber <= 0) {
$this->taskNumber = 0;
$this->state = 'SWOOLE_WORKER_IDLE';
} else {
$this->state = 'SWOOLE_WORKER_BUSY';
}
}
/** /**
* @return array * @return array
*/ */