\Imi\TaskTaskManager

<?php
namespace Imi\Task;

use Imi\ServerManage;
use Imi\Task\Parser\TaskParser;
use Imi\Task\Interfaces\ITaskParam;
use Imi\Task\Interfaces\ITaskHandler;

abstract class TaskManager
{
    /**
     * 投递异步任务
     * 调用成功返回任务ID，失败返回false
     * @param TaskInfo $taskInfo
     * @param int $workerID
     * @return int|bool
     */
    public static function post(TaskInfo $taskInfo, $workerID = -1)
    {
        return ServerManage::getServer('main')->getSwooleServer()->task($taskInfo, $workerID, [$taskInfo->getTaskHandler(), 'finish']);
    }

    /**
     * 使用任务名称投递异步任务
     * 调用成功返回任务ID，失败返回false
     * 
     * @param string $name
     * @param mixed $data
     * @param integer $workerID
     * @return int|bool
     */
    public static function nPost(string $name, $data, $workerID = -1)
    {
        $task = TaskParser::getInstance()->getTask($name);
        $paramClass = $task['Task']->paramClass;
        return static::post(new TaskInfo(new $task['className'], new $paramClass($data)), $workerID);
    }

    /**
     * 投递任务，协程挂起等待，单位：秒
     * 返回值为任务直接结果
     * @param TaskInfo $taskInfo
     * @param float $timeout
     * @param int $workerID
     * @return string|bool
     */
    public static function postWait(TaskInfo $taskInfo, $timeout, $workerID = -1)
    {
        $server = ServerManage::getServer('main')->getSwooleServer();
        $result = $server->taskwait($taskInfo, $timeout, $workerID);
        $taskInfo->getTaskHandler()->finish($server, -1, $result);
        return $result;
    }

    /**
     * 使用任务名称投递任务，协程挂起等待，单位：秒
     * 返回值为任务直接结果
     * 
     * @param string $name
     * @param mixed $data
     * @param float $timeout
     * @param int $workerID
     * @return string|bool
     */
    public static function nPostWait(string $name, $data, $timeout, $workerID = -1)
    {
        $task = TaskParser::getInstance()->getTask($name);
        $paramClass = $task['Task']->paramClass;
        return static::postWait(new TaskInfo(new $task['className'], new $paramClass($data)), $timeout, $workerID);
    }

    /**
     * 投递任务，协程方式等待全部执行完毕或超时，单位：秒
     * $tasks必须为数组，有两种情况
     * 1. TaskInfo数组
     * 2. ['task名称', 参数] 参数可以被省略
     * 
     * 返回值为任务直接结果
     * @param TaskInfo[]|array $tasks
     * @param float $timeout
     * @return array
     */
    public static function postCo(array $tasks, $timeout)
    {
        $server = ServerManage::getServer('main')->getSwooleServer();
        $taskParser = TaskParser::getInstance();
        foreach($tasks as $i => $item)
        {
            if(!$item instanceof TaskInfo)
            {
                $task = $taskParser->getTask($item[0]);
                $paramClass = $task['Task']->paramClass;
                $tasks[$i] = new TaskInfo(new $task['className'], new $paramClass($item[1] ?? []));
            }
        }
        $result = $server->taskCo($tasks, $timeout);
        foreach($result as $i => $item)
        {
            $tasks[$i]->getTaskHandler()->finish($server, -1, $item);
        }
        return $result;
    }

    /**
     * 获取 TaskInfo
     *
     * @param string $name
     * @return TaskInfo
     */
    public static function getTaskInfo(string $name, $data): TaskInfo
    {
        $task = TaskParser::getInstance()->getTask($name);
        $paramClass = $task['Task']->paramClass;
        return new TaskInfo(new $task['className'], new $paramClass($data));
    }

}