事件onStart onWorkerStart onFinish等事件支持自定义监听

Author: toxmc

config/swoole_http.php

@@ -1,6 +1,7 @@
 <?php
 
 use Swoole\Table;
+use FastLaravel\Http\Server\Event;
 
 return [
     /*
@@ -178,6 +179,23 @@
         Illuminate\Pagination\PaginationServiceProvider::class,
     ],
 
+    /*
+    |--------------------------------------------------------------------------
+    | The event listener mappings for the application.
+    | event => listener
+    |
+    | @see: class FastLaravel\Http\Server\Event
+    |--------------------------------------------------------------------------
+    */
+    'listens' => [
+        // Event::START => [
+        //     'App\Listeners\StartListener',
+        // ],
+        // Event::WORKER_START => [
+        //     'App\Listeners\WorkerStartListener',
+        // ]
+    ],
+
     /*
     |--------------------------------------------------------------------------
     | Define your swoole tables here.

src/Commands/HttpController.php

@@ -6,6 +6,7 @@
 use Inhere\Console\Controller;
 use Illuminate\Config\Repository;
 use Illuminate\Events\Dispatcher;
+use Illuminate\Support\Facades\Event;
 use Illuminate\Contracts\Queue\Factory as QueueFactoryContract;
 use FastLaravel\Http\Server\Manager;
 
@@ -122,11 +123,21 @@ public function startCommand()
             'app'         => $this->config('app'),
             'swoole_http' => $this->config('server')
         ]));
-        $this->laravelApp->singleton('events', function ($app) {
-            return (new Dispatcher($app))->setQueueResolver(function () use ($app) {
+
+        // 注册swoole回调监听事件
+        $listens = $this->config('server')['listens'] ?? [];
+        $this->laravelApp->singleton('events', function ($app) use($listens) {
+            $event = (new Dispatcher($app))->setQueueResolver(function () use ($app) {
                 return $app->make(QueueFactoryContract::class);
             });
+            foreach ($listens as $eventName => $listeners) {
+                foreach ($listeners as $listener) {
+                    $event->listen($eventName, $listener);
+                }
+            }
+            return $event;
         });
+
         $this->output->table($this->getInfos(false), 'fast laravel http info.');
         (new Manager($this->laravelApp, 'laravel'))->start();
     }

src/Commands/HttpServerCommand.php

@@ -118,10 +118,15 @@ public function publish()
         $basePath = base_path();
         $configPath = $basePath . '/config/swoole_http.php';
         $taskMiddlewarePath = $basePath . '/app/Http/Middleware/TaskWorker.php';
+        $startListener = $basePath . '/app/Listeners/StartListener.php';
+        $workerStartListener = $basePath . '/app/Listeners/WorkerStartListener.php';
+
         $todoList = [
             ['from' => realpath(__DIR__ . '/../../config/swoole_http.php'), 'to' => $configPath, 'mode' => 0644],
             ['from' => realpath(__DIR__ . '/../../fast'), 'to' => $basePath . '/fast', 'mode' => 0755, 'link' => true],
             ['from' => realpath(__DIR__ . '/../Middleware/TaskWorker.php'), 'to' => $taskMiddlewarePath, 'mode' => 0755, 'link' => true],
+            ['from' => realpath(__DIR__ . '/../Listeners/StartListener.php'), 'to' => $startListener, 'mode' => 0755, 'link' => true],
+            ['from' => realpath(__DIR__ . '/../Listeners/WorkerStartListener.php'), 'to' => $workerStartListener, 'mode' => 0755, 'link' => true],
         ];
 
         foreach ($todoList as $id => $todo) {

src/Listeners/StartListener.php

@@ -0,0 +1,25 @@
+<?php
+
+namespace App\Listeners;
+
+class StartListener
+{
+    /**
+     * Create the event listener.
+     */
+    public function __construct()
+    {
+        //
+    }
+
+    /**
+     * Handle the event.
+     * @param \Swoole\Http\Server $server
+     *
+     * @return void
+     */
+    public function handle($server)
+    {
+        echo "In StartListener\n";
+    }
+}

src/Listeners/WorkerStartListener.php

@@ -0,0 +1,27 @@
+<?php
+
+namespace App\Listeners;
+
+class WorkerStartListener
+{
+    /**
+     * Create the event listener.
+     */
+    public function __construct()
+    {
+        //
+    }
+
+    /**
+     * Handle the event.
+     *
+     * @param  \Swoole\Http\Server  $server
+     * @params int $workerId
+     *
+     * @return void
+     */
+    public function handle($server, $workerId)
+    {
+        echo "In workerStartListener:worker id:{$workerId}\n";
+    }
+}

src/Server/Event.php

@@ -0,0 +1,41 @@
+<?php
+
+namespace FastLaravel\Http\Server;
+
+/**
+ * Class Event
+ * fast laravel支持监听的事件
+ * @package FastLaravel\Http\Server
+ */
+class Event
+{
+    /**
+     * 对应 onStart 回调
+     */
+    const START = 'start';
+
+    /**
+     * 对应 onManagerStart 回调
+     */
+    const MANAGER_START = 'managerStart';
+
+    /**
+     * 对应 onWorkerStart 回调
+     */
+    const WORKER_START = 'workerStart';
+
+    /**
+     * 对应 onFinish 回调
+     */
+    const FINISH = 'finish';
+
+    /**
+     * 对应 onWorkerStop 回调
+     */
+    const WORKER_STOP = 'workerStop';
+
+    /**
+     * 对应 onShutdown 回调
+     */
+    const SHUTDOWN = 'shutdown';
+}

src/Server/Manager.php

@@ -13,6 +13,7 @@
 use FastLaravel\Http\Process\HotReload;
 use FastLaravel\Http\Database\ConnectionResolver;
 use FastLaravel\Http\Facade\Show;
+use FastLaravel\Http\Server\Event;
 use Illuminate\Contracts\Container\Container;
 use Illuminate\Contracts\Debug\ExceptionHandler;
 
@@ -264,7 +265,7 @@ protected function createTracker()
      */
     public function onStart($server)
     {
-        $this->container->make('events')->dispatch('start', func_get_args());
+        $this->container->make('events')->dispatch(Event::START, func_get_args());
         $this->setProcessName('master process');
         $this->createPidFile();
         if (isTesting()) {
@@ -282,19 +283,20 @@ public function onStart($server)
      */
     public function onManagerStart($server)
     {
-        $this->container->make('events')->dispatch('managerStart', func_get_args());
+        $this->container->make('events')->dispatch(Event::MANAGER_START, func_get_args());
         $this->setProcessName('manager process');
     }
 
     /**
      * "onWorkerStart" callback.
      *
      * @param HttpServer $server
+     * @param int $workerId
      * @throws
      */
-    public function onWorkerStart($server)
+    public function onWorkerStart($server, $workerId)
     {
-        $this->container->make('events')->dispatch('workerStart', func_get_args());
+        $this->container->make('events')->dispatch(Event::WORKER_START, func_get_args());
         $this->clearCache();
 
         // init laravel app in task workers
@@ -417,23 +419,36 @@ public function onTaskCo(HttpServer $server, $task)
 
     /**
      * Set onFinish callback.
+     * @param HttpServer $server
+     * @param int $taskId
+     * @param string $data
+     * @throws
      */
-    public function onFinish(HttpServer $server, $taskId, $data)
+    public function onFinish($server, $taskId, $data)
     {
+        $this->container->make('events')->dispatch(Event::FINISH, func_get_args());
     }
 
     /**
      * Set onWorkerStop callback.
+     *
+     * @param HttpServer $server
+     * @param int $workerId
+     * @throws
      */
-    public function onWorkerStop()
+    public function onWorkerStop($server, $workerId)
     {
+        $this->container->make('events')->dispatch(Event::WORKER_STOP, func_get_args());
     }
 
     /**
      * Set onShutdown callback.
+     * @param HttpServer $server
+     * @throws
      */
-    public function onShutdown()
+    public function onShutdown($server)
     {
+        $this->container->make('events')->dispatch(Event::SHUTDOWN, func_get_args());
         $this->removePidFile();
     }