Revert "改名"

This reverts commit fdf58326

kiri-task/Annotation/AsynchronousTask.php

@@ -0,0 +1,33 @@
+<?php
+
+namespace Kiri\Task\Annotation;
+
+
+use Kiri\Annotation\Attribute;
+use Kiri\Task\TaskManager;
+
+#[\Attribute(\Attribute::TARGET_CLASS)] class AsynchronousTask extends Attribute
+{
+
+
+	/**
+	 * @param string $name
+	 */
+	public function __construct(public string $name)
+	{
+	}
+
+
+	/**
+	 * @param mixed $class
+	 * @param mixed $method
+	 * @return mixed
+	 */
+	public function execute(mixed $class, mixed $method = ''): mixed
+	{
+		$AsyncTaskExecute = \Kiri::getDi()->get(TaskManager::class);
+		$AsyncTaskExecute->add($this->name, $class::class);
+		return parent::execute($class, $method); // TODO: Change the autogenerated stub
+	}
+
+}

kiri-task/AsyncTaskExecute.php

@@ -0,0 +1,41 @@
+<?php
+
+namespace Kiri\Task;
+
+use Exception;
+use Kiri;
+use Kiri\Abstracts\Component;
+use Kiri\Server\SwooleServerInterface;
+
+
+/**
+ *
+ */
+class AsyncTaskExecute extends Component
+{
+
+
+	use TaskResolve;
+
+
+
+	/**
+	 * @param OnTaskInterface|string $handler
+	 * @param array $params
+	 * @param int $workerId
+	 * @throws Exception
+	 */
+	public function execute(OnTaskInterface|string $handler, array $params = [], int $workerId = -1)
+	{
+		$server = Kiri::getDi()->get(SwooleServerInterface::class);
+		if ($workerId < 0 || $workerId > $server->setting['task_worker_num']) {
+			$workerId = random_int(0, $server->setting['task_worker_num'] - 1);
+		}
+		if (is_string($handler)) {
+			$handler = $this->handle($handler, $params);
+		}
+		$server->task(serialize($handler), $workerId);
+	}
+
+
+}

kiri-task/CoroutineTaskExecute.php

@@ -0,0 +1,85 @@
+<?php
+
+namespace Kiri\Task;
+
+use Kiri\Abstracts\Component;
+use Kiri\Core\HashMap;
+use Kiri\Exception\ConfigException;
+use ReflectionException;
+use Swoole\Coroutine;
+use Swoole\Server\Task;
+
+class CoroutineTaskExecute extends Component
+{
+
+	use TaskResolve;
+
+
+	private HashMap $hashMap;
+
+
+	private Coroutine\Channel $channel;
+
+
+	private OnServerTask $taskServer;
+
+
+	private int $total = 50;
+
+
+	/**
+	 *
+	 */
+	public function init()
+	{
+		$this->hashMap = new HashMap();
+
+		$this->channel = new Coroutine\Channel($this->total);
+
+		$this->taskServer = \Kiri::getDi()->get(OnServerTask::class);
+
+		Coroutine::create(function () {
+			$barrier = Coroutine\Barrier::make();
+			for ($i = 0; $i < 50; $i++) {
+				Coroutine::create(function () {
+					$this->handler();
+				});
+			}
+			Coroutine\Barrier::wait($barrier);
+		});
+	}
+
+
+	/**
+	 * @return void
+	 * @throws ConfigException
+	 */
+	protected function handler()
+	{
+		$data = $this->channel->pop(-1);
+
+		$task = new Task();
+		$task->data = $data;
+
+		$this->taskServer->onCoroutineTask(null, $task);
+
+		$this->handler();
+	}
+
+
+	/**
+	 * @param OnTaskInterface|string $handler
+	 * @param array $params
+	 * @param int $workerId
+	 * @return void
+	 * @throws ReflectionException
+	 */
+	public function execute(OnTaskInterface|string $handler, array $params = [], int $workerId = -1)
+	{
+		if (is_string($handler)) {
+			$handler = $this->handle($handler, $params);
+		}
+		$this->channel->push(serialize($handler));
+	}
+
+}

kiri-task/OnServerTask.php

@@ -0,0 +1,94 @@
+<?php
+
+
+namespace Kiri\Task;
+
+
+use Kiri\Annotation\Inject;
+use Kiri\Abstracts\Logger;
+use Kiri\Exception\ConfigException;
+use Kiri\Task\OnTaskInterface;
+use Swoole\Server;
+
+
+/**
+ * Class OnServerTask
+ * @package Server\Task
+ */
+class OnServerTask
+{
+
+
+	#[Inject(Logger::class)]
+	public Logger $logger;
+
+
+	/**
+	 * @param Server $server
+	 * @param int $task_id
+	 * @param int $src_worker_id
+	 * @param mixed $data
+	 * @throws ConfigException
+	 */
+	public function onTask(Server $server, int $task_id, int $src_worker_id, mixed $data)
+	{
+		try {
+			$data = $this->resolve($data);
+		} catch (\Throwable $exception) {
+			$data = jTraceEx($exception);
+
+			$this->logger->error('task', [error_trigger_format($exception)]);
+		} finally {
+			$server->finish($data);
+		}
+	}
+
+
+	/**
+	 * @param Server|null $server
+	 * @param Server\Task $task
+	 * @throws ConfigException
+	 */
+	public function onCoroutineTask(?Server $server, Server\Task $task)
+	{
+		try {
+			$data = $this->resolve($task->data);
+		} catch (\Throwable $exception) {
+			$data = jTraceEx($exception);
+
+			$this->logger->error('task', [error_trigger_format($exception)]);
+		} finally {
+			$task->finish($data);
+		}
+	}
+
+
+	/**
+	 * @param $data
+	 * @return null
+	 */
+	private function resolve($data)
+	{
+		$execute = unserialize($data);
+		if ($execute instanceof OnTaskInterface) {
+			return $execute->execute();
+		}
+		return null;
+	}
+
+
+	/**
+	 * @param Server $server
+	 * @param int $task_id
+	 * @param mixed $data
+	 */
+	public function onFinish(Server $server, int $task_id, mixed $data)
+	{
+		if (!($data instanceof OnTaskInterface)) {
+			return;
+		}
+		$data->finish($server, $task_id);
+	}
+
+
+}

kiri-task/OnTaskInterface.php

@@ -0,0 +1,17 @@
+<?php
+
+
+namespace Kiri\Task;
+
+
+use Swoole\Server;
+
+interface OnTaskInterface
+{
+
+	public function execute();
+
+
+	public function finish(Server $server, int $task_id);
+
+}

kiri-task/TaskManager.php

@@ -0,0 +1,77 @@
+<?php
+
+namespace Kiri\Task;
+
+use Kiri\Abstracts\Component;
+use Kiri\Core\HashMap;
+use Psr\Container\ContainerExceptionInterface;
+use Psr\Container\NotFoundExceptionInterface;
+use Swoole\Server;
+
+class TaskManager extends Component
+{
+
+
+	private HashMap $hashMap;
+
+
+	/**
+	 *
+	 */
+	public function init()
+	{
+		$this->hashMap = new HashMap();
+	}
+
+
+	/**
+	 * @param Server $swollen
+	 * @return void
+	 * @throws ContainerExceptionInterface
+	 * @throws NotFoundExceptionInterface
+	 */
+	public function taskListener(Server $swollen)
+	{
+		if (!isset($swollen->setting['task_worker_num']) || $swollen->setting['task_worker_num'] < 1) {
+			return;
+		}
+
+		$task_use_object = $swollen->setting['task_object'] ?? $swollen->setting['task_use_object'] ?? false;
+		$reflect = $this->container->get(OnServerTask::class);
+
+		$swollen->on('finish', [$reflect, 'onFinish']);
+		if ($task_use_object || $swollen->setting['task_enable_coroutine']) {
+			$swollen->on('task', [$reflect, 'onCoroutineTask']);
+		} else {
+			$swollen->on('task', [$reflect, 'onTask']);
+		}
+	}
+
+
+	/**
+	 * @param string $key
+	 * @param $handler
+	 */
+	public function add(string $key, $handler)
+	{
+		$this->hashMap->put($key, $handler);
+	}
+
+
+	/**
+	 * @param string $key
+	 * @return OnTaskInterface
+	 * @throws ContainerExceptionInterface
+	 * @throws NotFoundExceptionInterface
+	 */
+	public function get(string $key): OnTaskInterface
+	{
+		$task = $this->hashMap->get($key);
+		if (is_string($task)) {
+			$task = $this->getContainer()->get($task);
+		}
+		return $task;
+	}
+
+
+}

kiri-task/TaskResolve.php

@@ -0,0 +1,31 @@
+<?php
+
+namespace Kiri\Task;
+
+use Exception;
+use ReflectionException;
+
+trait TaskResolve
+{
+
+
+	/**
+	 * @param $handler
+	 * @param $params
+	 * @return object
+	 * @throws ReflectionException
+	 * @throws Exception
+	 */
+	private function handle($handler, $params): object
+	{
+		if (!class_exists($handler) && $this->hashMap->has($handler)) {
+			$handler = $this->hashMap->get($handler);
+		}
+		$implements = $this->container->getReflect($handler);
+		if (!in_array(OnTaskInterface::class, $implements->getInterfaceNames())) {
+			throw new Exception('Task must instance ' . OnTaskInterface::class);
+		}
+		return $implements->newInstanceArgs($params);
+	}
+
+}