From 29977fa2d6da4f0b7a1b5f13ac3f402ef139e1f3 Mon Sep 17 00:00:00 2001
From: inhere <in.798@qq.com>
Date: Sun, 13 Sep 2020 18:16:52 +0800
Subject: [PATCH] add comments for process pool

---
 .../src/Annotation/Mapping/Process.php        |  2 +-
 src/process/src/AutoLoader.php                |  3 ++-
 src/process/src/Contract/MessageInterface.php | 27 +++++++++++++++++++
 src/process/src/ProcessDispatcher.php         |  2 +-
 src/process/src/SwooleEvent.php               |  9 ++++++-
 5 files changed, 39 insertions(+), 4 deletions(-)
 create mode 100644 src/process/src/Contract/MessageInterface.php

diff --git a/src/process/src/Annotation/Mapping/Process.php b/src/process/src/Annotation/Mapping/Process.php
index e185e7a18..780f69992 100644
--- a/src/process/src/Annotation/Mapping/Process.php
+++ b/src/process/src/Annotation/Mapping/Process.php
@@ -28,7 +28,7 @@
 class Process
 {
     /**
-     * Default
+     * Default run the first worker
      */
     public const DEFAULT = -1;
 
diff --git a/src/process/src/AutoLoader.php b/src/process/src/AutoLoader.php
index e7dacf65f..feb096cca 100644
--- a/src/process/src/AutoLoader.php
+++ b/src/process/src/AutoLoader.php
@@ -43,7 +43,8 @@ public function beans(): array
                 'on'    => [
                     SwooleEvent::WORKER_START => bean(WorkerStartListener::class),
                     SwooleEvent::WORKER_STOP  => bean(WorkerStopListener::class)
-                ]
+                ],
+                'workerNum' => 2,
             ]
         ];
     }
diff --git a/src/process/src/Contract/MessageInterface.php b/src/process/src/Contract/MessageInterface.php
new file mode 100644
index 000000000..d013764cb
--- /dev/null
+++ b/src/process/src/Contract/MessageInterface.php
@@ -0,0 +1,27 @@
+<?php declare(strict_types=1);
+/**
+ * This file is part of Swoft.
+ *
+ * @link     https://swoft.org
+ * @document https://swoft.org/docs
+ * @contact  group@swoft.org
+ * @license  https://github.com/swoft-cloud/swoft/blob/master/LICENSE
+ */
+
+namespace Swoft\Process\Contract;
+
+use Swoole\Process\Pool;
+
+/**
+ * Class MessageInterface
+ *
+ * @since 2.0
+ */
+interface MessageInterface
+{
+    /**
+     * @param Pool   $pool
+     * @param string $data
+     */
+    public function onMessage(Pool $pool, string $data): void;
+}
diff --git a/src/process/src/ProcessDispatcher.php b/src/process/src/ProcessDispatcher.php
index 0d213eca1..b6ad28518 100644
--- a/src/process/src/ProcessDispatcher.php
+++ b/src/process/src/ProcessDispatcher.php
@@ -64,8 +64,8 @@ public function dispatcher(Pool $pool, int $workerId): void
     private function getProcess(int $workerId): ProcessInterface
     {
         $processName = ProcessRegister::getProcess($workerId);
-        $process     = BeanFactory::getBean($processName);
 
+        $process = BeanFactory::getBean($processName);
         if (!$process instanceof ProcessInterface) {
             throw new ProcessException('Process must be instanceof ProcessInterface');
         }
diff --git a/src/process/src/SwooleEvent.php b/src/process/src/SwooleEvent.php
index f6398f3b5..6d260f640 100644
--- a/src/process/src/SwooleEvent.php
+++ b/src/process/src/SwooleEvent.php
@@ -12,6 +12,7 @@
 
 use Swoft\Process\Contract\WorkerStartInterface;
 use Swoft\Process\Contract\WorkerStopInterface;
+use Swoft\Process\Contract\MessageInterface;
 
 /**
  * Class SwooleEvent
@@ -30,11 +31,17 @@ class SwooleEvent
      */
     public const WORKER_STOP = 'workerStop';
 
+    /**
+     * On message
+     */
+    public const MESSAGE = 'message';
+
     /**
      * Listener mapping
      */
     public const LISTENER_MAPPING = [
         self::WORKER_START => WorkerStartInterface::class,
-        self::WORKER_STOP  => WorkerStopInterface::class
+        self::WORKER_STOP  => WorkerStopInterface::class,
+        self::MESSAGE      => MessageInterface::class
     ];
 }