update. PIPE_MESSAGE事件结束后触发COROUTINE_DEFER, COROUTINE_COMPLETE

src/server/src/Context/PipeMessageContext.php

@@ -0,0 +1,57 @@
+<?php
+/**
+ * The file is part of the swoft_marketing_engine.
+ *
+ * (c) alan <[email protected]>.
+ *
+ * 2021/1/3 1:43 下午
+ */
+
+namespace Swoft\Server\Context;
+
+use Swoft\Bean\Annotation\Mapping\Bean;
+use Swoft\Bean\Concern\PrototypeTrait;
+use Swoft\Context\AbstractContext;
+
+/**
+ * Class PipeMessageContext
+ * @package App\Context
+ *
+ * @Bean(scope=Bean::PROTOTYPE)
+ */
+class PipeMessageContext extends AbstractContext
+{
+    use PrototypeTrait;
+
+    /**
+     * @param int $srcWorkerId
+     * @param $pipeMessage
+     * @return PipeMessageContext
+     */
+    public static function new(int $srcWorkerId, $pipeMessage): self
+    {
+        $self = self::__instance();
+
+        $self->set('srcWorkerId', $srcWorkerId);
+        $self->set('pipeMessage', $pipeMessage);
+
+        return $self;
+    }
+
+    /**
+     * @return int
+     */
+    public function getSrcWorkerId(): int
+    {
+        return $this->get('srcWorkerId');
+    }
+
+    /**
+     * @return mixed
+     */
+    public function getPipeMessage()
+    {
+        return $this->get('pipeMessage');
+    }
+}
+

src/server/src/Swoole/PipeMessageListener.php

@@ -12,10 +12,15 @@
 
 use Swoft;
 use Swoft\Bean\Annotation\Mapping\Bean;
+use Swoft\Context\Context;
+use Swoft\Log\Error;
 use Swoft\Log\Helper\CLog;
 use Swoft\Server\Contract\PipeMessageInterface;
 use Swoft\Server\ServerEvent;
+use Swoft\SwoftEvent;
 use Swoole\Server;
+use Swoft\Server\Context\PipeMessageContext;
+use Throwable;
 
 /**
  * Class PipeMessageListener
@@ -34,8 +39,26 @@ class PipeMessageListener implements PipeMessageInterface
      */
     public function onPipeMessage(Server $server, int $srcWorkerId, $message): void
     {
-        CLog::debug("PipeMessage: received pipe-message fromWID=$srcWorkerId message=$message");
+        Context::set(PipeMessageContext::new($srcWorkerId, $message));
 
-        Swoft::trigger(ServerEvent::PIPE_MESSAGE, $message, $srcWorkerId, $server);
+        CLog::debug("PipeMessage: received pipe-message fromWID={$srcWorkerId}}");
+
+        try {
+
+            Swoft::trigger(ServerEvent::PIPE_MESSAGE, $message, $srcWorkerId, $server);
+
+        } catch (Throwable $e) {
+
+            Error::log("PipeMessage handle fails: {$e->getMessage()} in {$e->getFile()}:{$e->getLine()}");
+
+        } finally {
+
+            // Defer
+            Swoft::trigger(SwoftEvent::COROUTINE_DEFER);
+
+            // Destroy
+            Swoft::trigger(SwoftEvent::COROUTINE_COMPLETE);
+
+        }
     }
 }