Merge pull request #2 from hsldymq/feat-worker-shutdown-event

feat: worker 增加 shutdown 事件
hsldymq · Dec 29, 2023 · c7b3f88 · c7b3f88
2 parents 50a6fb7 + 30fee26
commit c7b3f88
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/examples/01.basic.php b/examples/01.basic.php
@@ -31,6 +31,9 @@
     ->registerEvent('error', function (string $reason, \Throwable $e, Worker $worker) {
         echo "Worker Error. Reason: {$reason}, Message: {$e->getMessage()}\n";
         $worker->shutdown();
+    })
+    ->registerEvent("shutdown", function (Worker $worker) {
+        echo "worker shutdown event triggered: {$worker->getWorkerID()}\n";
     });
 
 $params = require __DIR__.'/amqp_params.php';

diff --git a/src/Worker.php b/src/Worker.php
@@ -22,6 +22,9 @@
  * @event disconnected      与dispatcher的连接中断,即将退出
  *                          参数: \Archman\BugsBunny\Worker $worker
  *
+ * @event shutdown          worker退出
+ *                          参数: \Archman\BugsBunny\Worker $worker
+ *
  * @event error             发生错误
  *                          参数: string $reason, \Throwable $ex, \Archman\BugsBunny\Worker $worker
  *                          $reason enum:
@@ -113,6 +116,10 @@ public function run()
 
             $this->errorlessEmit('patrolling');
         }
+
+        if (!$this->passiveShutdown) {
+            $this->errorlessEmit('shutdown');
+        }
     }
 
     /**
@@ -179,6 +186,7 @@ public function handleMessage(Message $msg)
 
         if ($this->noMore) {
             if ($this->passiveShutdown) {
+                $this->errorlessEmit('shutdown');
                 $this->sendMessage(new Message(MessageTypeEnum::KILL_ME, ''));
             } else {
                 $this->sendMessage(new Message(MessageTypeEnum::I_QUIT, ''));