This commit is contained in:
2021-11-30 14:32:56 +08:00
parent c260ac6df4
commit 27834d37ed
3 changed files with 91 additions and 22 deletions
+21 -11
View File
@@ -5,6 +5,7 @@ namespace Server;
use Annotation\Inject;
use Exception;
use Kiri\Abstracts\Config;
use Kiri\Di\Container;
use Psr\Container\ContainerInterface;
use Kiri\Error\Logger;
use Kiri\Exception\ConfigException;
@@ -26,7 +27,7 @@ use Server\Handler\OnPipeMessage;
use Server\Handler\OnServer;
use Server\Handler\OnServerManager;
use Server\Handler\OnServerReload;
use Server\Handler\OnServerTask;
use Server\Tasker\OnServerTask;
use Server\Handler\OnServerWorker;
use Swoole\Http\Server as HServer;
use Swoole\Process;
@@ -65,7 +66,7 @@ class ServerManager
/**
* @var ContainerInterface
* @var Container
*/
#[Inject(ContainerInterface::class)]
public ContainerInterface $container;
@@ -246,6 +247,9 @@ class ServerManager
* @param int $port
* @param int $mode
* @param array $settings
* @throws ConfigException
* @throws ContainerExceptionInterface
* @throws NotFoundExceptionInterface
* @throws Exception
*/
private function addNewListener(string $type, string $host, int $port, int $mode, array $settings = [])
@@ -279,8 +283,10 @@ class ServerManager
* @param int $port
* @param int $mode
* @param array $settings
* @throws ReflectionException
* @throws ConfigException
* @throws ContainerExceptionInterface
* @throws NotFoundExceptionInterface
* @throws ReflectionException
* @throws Exception
*/
private function createBaseServer(string $type, string $host, int $port, int $mode, array $settings = [])
@@ -320,8 +326,9 @@ class ServerManager
/**
* @param array $settings
* @throws ContainerExceptionInterface
* @throws NotFoundExceptionInterface
* @throws ReflectionException
* @throws Exception
*/
private function addDefaultListener(array $settings): void
{
@@ -336,12 +343,13 @@ class ServerManager
}
/**
* @param array $events
* @param Server|Port $server
* @throws ContainerExceptionInterface
* @throws NotFoundExceptionInterface
*/
/**
* @param array $events
* @param Server|Port $server
* @throws ContainerExceptionInterface
* @throws NotFoundExceptionInterface
* @throws ReflectionException
*/
private function addServiceEvents(array $events, Server|Port $server)
{
foreach ($events as $name => $event) {
@@ -426,10 +434,12 @@ class ServerManager
/**
* @param array|null $settings
* @throws ContainerExceptionInterface
* @throws NotFoundExceptionInterface
* @throws ReflectionException
*/
public function bindCallback(?array $settings = [])
{
// TODO: Implement bindCallback() method.
if (count($settings) < 1) {
return;
}
+65
View File
@@ -0,0 +1,65 @@
<?php
namespace Server;
use Annotation\Inject;
use Exception;
use Kiri\Di\Container;
use Psr\Container\ContainerInterface;
use Server\Contract\OnTaskInterface;
class AsyncTask
{
/**
* @var SwooleServerInterface
*/
#[Inject(SwooleServerInterface::class)]
public SwooleServerInterface $server;
/**
* @var Container
*/
#[Inject(ContainerInterface::class)]
public ContainerInterface $container;
/**
* @param OnTaskInterface|string $handler
* @param array $params
* @param int|null $workerId
* @throws Exception
*/
public function execute(OnTaskInterface|string $handler, array $params = [], int $workerId = null)
{
if ($workerId === null || $workerId <= $this->server->setting['worker_num']) {
$workerId = random_int($this->server->setting['worker_num'] + 1,
$this->server->setting['worker_num'] + 1 + $this->server->setting['task_worker_num']);
}
if (is_string($handler)) {
$handler = $this->handle($handler, $params);
}
$this->server->task(serialize($handler), $workerId);
}
/**
* @param $handler
* @param $params
* @return object
* @throws \ReflectionException
* @throws Exception
*/
private function handle($handler, $params): object
{
$implements = $this->container->getReflect($handler);
if (!in_array(OnTaskInterface::class, $implements->getInterfaceNames())) {
throw new Exception('Task must instance ' . OnTaskInterface::class);
}
return $implements->newInstanceArgs($params);
}
}
@@ -1,13 +1,12 @@
<?php
namespace Server\Handler;
namespace Server\Tasker;
use Annotation\Inject;
use Kiri\Abstracts\Logger;
use Kiri\Exception\ConfigException;
use Kiri\Kiri;
use ReflectionException;
use Server\Contract\OnTaskInterface;
use Swoole\Server;
@@ -68,19 +67,14 @@ class OnServerTask
/**
* @param $data
* @return null
* @throws ReflectionException
*/
private function resolve($data)
{
[$class, $params] = json_encode($data, true);
$reflect = Kiri::getDi()->getReflect($class);
if (!$reflect->isInstantiable()) {
return null;
$execute = unserialize($data);
if ($execute instanceof OnTaskInterface) {
return $execute->execute();
}
$class = $reflect->newInstanceArgs($params);
return $class->execute();
return null;
}