kiri-http-server/Task/Task.php

<?php

namespace Kiri\Server\Task;


use Kiri;
use Kiri\Router\Base\ExceptionHandlerDispatcher;
use Kiri\Router\Interface\ExceptionHandlerInterface;
use Kiri\Server\Constant;
use Swoole\Server;

/**
 *
 */
class Task
{


    public ExceptionHandlerInterface $exception;


    /**
     *
     */
    public function __construct()
    {
        $exception = \config('servers.task.exception');
        if (!in_array(ExceptionHandlerInterface::class, class_implements($exception))) {
            $exception = ExceptionHandlerDispatcher::class;
        }
        $this->exception = Kiri::getDi()->get($exception);
    }


    /**
     * @param Server $server
     * @return void
     */
    public function initTaskWorker(Server $server): void
    {
        if (!isset($server->setting[Constant::OPTION_TASK_WORKER_NUM])) {
            return;
        }
        if ($server->setting[Constant::OPTION_TASK_WORKER_NUM] < 1) {
            return;
        }
        $server->on('finish', [$this, 'onFinish']);
        $server->on('task', [$this, 'onTask']);
    }


    /**
     * @param Server $server
     * @param int $task_id
     * @param mixed $data
     * @return void
     * @throws
     */
    public function onFinish(Server $server, int $task_id, mixed $data): void
    {
        event(new OnTaskFinish($task_id, $data));
    }


    /**
     * @param Server $server
     * @param int $task_id
     * @param int $src_worker_id
     * @param mixed $data
     * @return void
     * @throws
     */
    public function onTask(Server $server, int $task_id, int $src_worker_id, mixed $data): void
    {
        try {
            [$handler, $params] = [$data[0], $data[1]];

            $handler = Kiri::getDi()->make($handler, $params);
            if (!($handler instanceof OnTaskInterface)) {
                throw new \Exception('Task process must implements ' . OnTaskInterface::class);
            }

            $response = call_user_func([$handler, 'process'], $task_id, $src_worker_id);
        } catch (\Throwable $throwable) {
			\Kiri::getLogger()->json_log($throwable, ['task_id' => $task_id, 'src_worker_id' => $src_worker_id, 'data' => $data]);

			$response = throwable($throwable);
        } finally {
            $server->finish($response);
        }
    }


}