This commit is contained in:
2021-04-26 17:34:33 +08:00
parent 76f9688ea6
commit 52357f8283
7 changed files with 577 additions and 487 deletions
+66 -66
View File
@@ -20,82 +20,82 @@ use Swoole\Server;
class OnPipeMessage extends Callback
{
/**
* @param Server $server
* @param int $src_worker_id
* @param $swollen_universalize
* @throws Exception
*/
public function onHandler(Server $server, int $src_worker_id, $swollen_universalize)
{
match ($swollen_universalize['action'] ?? null) {
'kafka' => $this->onKafkaWorker($swollen_universalize),
'crontab' => $this->onCrontabWorker($swollen_universalize),
default => $this->onMessageWorker($server, $src_worker_id, $swollen_universalize)
};
}
/**
* @param Server $server
* @param int $src_worker_id
* @param $swollen_universalize
* @throws Exception
*/
public function onHandler(Server $server, int $src_worker_id, $swollen_universalize)
{
match ($swollen_universalize['action'] ?? null) {
'kafka' => $this->onKafkaWorker($swollen_universalize),
'crontab' => $this->onCrontabWorker($swollen_universalize),
default => $this->onMessageWorker($server, $src_worker_id, $swollen_universalize)
};
}
/**
* @param array $message
* @return string
* @throws Exception
*/
private function onCrontabWorker(array $message): string
{
if (empty($message['handler'] ?? null)) {
throw new Exception('unknown handler');
}
/** @var Crontab $handler */
$handler = $message['handler'];
defer(function () use ($handler) {
$return = $handler->isRecover();
if ($return === 999) {
$name = $handler->getName();
/**
* @param array $message
* @return string
* @throws Exception
*/
private function onCrontabWorker(array $message): string
{
if (empty($message['handler'] ?? null)) {
throw new Exception('unknown handler');
}
/** @var Crontab $handler */
$handler = swoole_unserialize($message['handler']);
defer(function () use ($handler) {
if ($handler->isRecover() !== 999) {
return;
}
$redis = Snowflake::app()->getRedis();
$redis = Snowflake::app()->getRedis();
if ($redis->exists('stop:crontab:' . $name)) {
$redis->del('crontab:' . $name);
$redis->del('stop:crontab:' . $name);
} else {
$redis->set('crontab:' . $name, swoole_serialize($handler));
$tickTime = time() + $handler->getTickTime();
$redis->zAdd(Producer::CRONTAB_KEY, $tickTime, $name);
}
}
});
$handler->increment()->execute();
return 'success';
}
$name = $handler->getName();
if (!$redis->exists('stop:crontab:' . $name)) {
$redis->set('crontab:' . $name, swoole_serialize($handler));
$tickTime = time() + $handler->getTickTime();
$redis->zAdd(Producer::CRONTAB_KEY, $tickTime, $name);
} else {
$redis->del('crontab:' . $name);
$redis->del('stop:crontab:' . $name);
}
});
$handler->increment()->execute();
return 'success';
}
/**
* @param $server
* @param $src_worker_id
* @param $message
* @return string
* @throws Exception
*/
private function onMessageWorker($server, $src_worker_id, $message): string
{
fire(Event::PIPE_MESSAGE, [$server, $src_worker_id, $message]);
/**
* @param $server
* @param $src_worker_id
* @param $message
* @return string
* @throws Exception
*/
private function onMessageWorker($server, $src_worker_id, $message): string
{
fire(Event::PIPE_MESSAGE, [$server, $src_worker_id, $message]);
return 'success';
}
return 'success';
}
/**
* @param array $message
* @return string
*/
private function onKafkaWorker(array $message): string
{
[$topic, $rdMessage] = $message['body'];
/**
* @param array $message
* @return string
*/
private function onKafkaWorker(array $message): string
{
[$topic, $rdMessage] = $message['body'];
call_user_func($message['handler'], new Struct($topic, $rdMessage));
call_user_func($message['handler'], new Struct($topic, $rdMessage));
return 'success';
}
return 'success';
}
}