as2252258@163.com
88 lines
2.0 KiB
PHP
88 lines
2.0 KiB
PHP
<?php
|
|
declare(strict_types=1);
|
|
|
|
namespace HttpServer\Events;
|
|
|
|
|
|
use Exception;
|
|
use HttpServer\Abstracts\Callback;
|
|
use Kafka\Struct;
|
|
use Snowflake\Event;
|
|
use Snowflake\Exception\ComponentException;
|
|
use Swoole\Server;
|
|
|
|
/**
|
|
* Class OnPipeMessage
|
|
* @package HttpServer\Events
|
|
*/
|
|
class OnPipeMessage extends Callback
|
|
{
|
|
|
|
/**
|
|
* @param Server $server
|
|
* @param int $src_worker_id
|
|
* @param $message
|
|
* @throws ComponentException
|
|
* @throws Exception
|
|
*/
|
|
public function onHandler(Server $server, int $src_worker_id, $message)
|
|
{
|
|
try {
|
|
$swoole_unserialize = swoole_unserialize($message);
|
|
match ($swoole_unserialize['action'] ?? null) {
|
|
'kafka' => $this->onKafkaWorker($swoole_unserialize),
|
|
'crontab' => $this->onCrontabWorker($swoole_unserialize),
|
|
default => $this->onMessageWorker($server, $src_worker_id, $message)
|
|
};
|
|
} catch (\Throwable $exception) {
|
|
$this->addError($exception);
|
|
} finally {
|
|
fire(Event::SYSTEM_RESOURCE_RELEASES);
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* @param array $message
|
|
* @return string
|
|
*/
|
|
private function onCrontabWorker(array $message)
|
|
{
|
|
/** @var \Snowflake\Crontab\Crontab $crontab */
|
|
$crontab = $message['handler'] ?? null;
|
|
$crontab->increment()->execute();
|
|
return 'success';
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $server
|
|
* @param $src_worker_id
|
|
* @param $message
|
|
* @throws \Exception
|
|
*/
|
|
private function onMessageWorker($server, $src_worker_id, $message)
|
|
{
|
|
fire(Event::PIPE_MESSAGE, [$server, $src_worker_id, $message]);
|
|
|
|
return 'success';
|
|
}
|
|
|
|
|
|
/**
|
|
* @param array $message
|
|
* @throws \ReflectionException
|
|
* @throws \Snowflake\Exception\NotFindClassException
|
|
*/
|
|
private function onKafkaWorker(array $message)
|
|
{
|
|
[$topic, $rdMessage] = $message['body'];
|
|
|
|
call_user_func($message['handler'], new Struct($topic, $rdMessage));
|
|
|
|
return 'success';
|
|
}
|
|
|
|
|
|
}
|