modify

2021-04-07 02:31:43 +08:00
parent 964ce69eca
commit 1ff368b199
3 changed files with 38 additions and 25 deletions
@@ -4,6 +4,7 @@ declare(strict_types=1);
 namespace HttpServer\Events;


+use Annotation\Loader;
 use Exception;
 use HttpServer\Abstracts\Callback;
 use Snowflake\Event;
@@ -18,18 +19,26 @@ use Swoole\Server;
 class OnPipeMessage extends Callback
 {

-	/**
-	 * @param Server $server
-	 * @param int $src_worker_id
-	 * @param $message
-	 * @throws ComponentException
-	 * @throws Exception
-	 */
-	public function onHandler(Server $server, int $src_worker_id, $message)
-	{
-		// TODO: Implement onHandler() method.
-		$events = Snowflake::app()->getEvent();
-		$events->trigger(Event::PIPE_MESSAGE, [$server, $src_worker_id, $message]);
-	}
+    /**
+     * @param Server $server
+     * @param int $src_worker_id
+     * @param $message
+     * @throws ComponentException
+     * @throws Exception
+     */
+    public function onHandler(Server $server, int $src_worker_id, $message)
+    {
+        // TODO: Implement onHandler() method.
+        if (is_array($message) && ($message[1] ?? null) instanceof Loader) {
+            if ($src_worker_id > $server->setting['worker_num']) {
+                $message[1]->loadByDirectory(MODEL_PATH);
+            } else {
+                $message[1]->loadByDirectory(APP_PATH);
+            }
+        } else {
+            $events = Snowflake::app()->getEvent();
+            $events->trigger(Event::PIPE_MESSAGE, [$server, $src_worker_id, $message]);
+        }
+    }

 }