Files
kiri-core/HttpServer/Events/OnPipeMessage.php
T

78 lines
1.8 KiB
PHP
Raw Normal View History

2020-08-31 01:27:08 +08:00
<?php
2020-10-29 18:17:25 +08:00
declare(strict_types=1);
2020-08-31 01:27:08 +08:00
2020-09-02 11:38:47 +08:00
namespace HttpServer\Events;
2020-08-31 01:27:08 +08:00
2021-04-07 02:31:43 +08:00
use Annotation\Loader;
2020-09-02 17:02:13 +08:00
use Exception;
2020-09-04 01:05:33 +08:00
use HttpServer\Abstracts\Callback;
2021-04-11 17:18:54 +08:00
use Kafka\ConsumerInterface;
use Kafka\Struct;
use Kafka\TaskContainer;
2020-09-02 17:02:13 +08:00
use Snowflake\Event;
use Snowflake\Exception\ComponentException;
use Snowflake\Snowflake;
use Swoole\Server;
2020-08-31 01:27:08 +08:00
2020-09-02 17:02:13 +08:00
/**
* Class OnPipeMessage
* @package HttpServer\Events
*/
2020-08-31 01:27:08 +08:00
class OnPipeMessage extends Callback
{
2021-04-07 02:31:43 +08:00
/**
* @param Server $server
* @param int $src_worker_id
* @param $message
* @throws ComponentException
* @throws Exception
*/
public function onHandler(Server $server, int $src_worker_id, $message)
{
2021-04-11 17:18:54 +08:00
try {
$swoole_unserialize = swoole_unserialize($message);
match ($swoole_unserialize['action'] ?? null) {
'kafka' => $this->onKafkaWorker($swoole_unserialize),
default => $this->onMessageWorker($server, $src_worker_id, $message)
};
} catch (\Throwable $exception) {
$this->addError($exception);
} finally {
fire(Event::SYSTEM_RESOURCE_RELEASES);
}
2021-04-07 02:31:43 +08:00
}
2020-08-31 01:27:08 +08:00
2021-04-11 17:18:54 +08:00
/**
* @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)
{
2021-04-11 19:21:14 +08:00
[$topic, $rdMessage] = $message['body'];
2021-04-11 19:02:42 +08:00
2021-04-11 19:21:14 +08:00
call_user_func($message['handler'], new Struct($topic, $rdMessage[1]));
2021-04-11 19:02:42 +08:00
2021-04-11 17:18:54 +08:00
return 'success';
}
2020-08-31 01:27:08 +08:00
}