<?php

namespace Kiri\Server\Task;


use Kiri;
use Kiri\Server\Constant;
use Kiri\Server\ServerInterface;
use Psr\Container\ContainerExceptionInterface;
use Psr\Container\NotFoundExceptionInterface;
use ReflectionException;
use Swoole\Server;
use Kiri\Di\Inject\Container;

/**
 *
 */
class Task implements TaskInterface
{

    /**
     * @var ServerInterface
     */
    #[Container(ServerInterface::class)]
    public ServerInterface $server;


    /**
     * @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 ContainerExceptionInterface
     * @throws NotFoundExceptionInterface
     * @throws ReflectionException
     */
    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 mixed
     * @throws ReflectionException
     */
    public function onTask(Server $server, int $task_id, int $src_worker_id, mixed $data): mixed
    {
        $data = json_decode($data, true);
        if (is_null($data)) {
            return null;
        }
        $data[0] = Kiri::getDi()->get($data[0]);
        return call_user_func($data, $task_id, $src_worker_id);
    }


    /**
     * @param mixed $data
     * @param float $timeout
     * @param int $dstWorkerId
     * @return mixed
     */
    public function taskWait(mixed $data, float $timeout = 0.5, int $dstWorkerId = -1): mixed
    {
        return $this->server->taskwait($data, $timeout, $dstWorkerId);
    }


    /**
     * @param array $tasks
     * @param float $timeout
     * @return false|array
     */
    public function taskCo(array $tasks, float $timeout = 0.5): false|array
    {
        return $this->server->taskCo($tasks, $timeout);
    }


    /**
     * @param array $tasks
     * @param float $timeout
     * @return false|array
     */
    public function taskWaitMulti(array $tasks, float $timeout = 0.5): false|array
    {
        return $this->server->taskWaitMulti($tasks, $timeout);
    }


}