Fixed a bug where channels could not specify workerId publication

workbunny · May 17, 2024 · b1ca99f · b1ca99f
1 parent df45fcc
commit b1ca99f
Showing 1 changed file with 21 additions and 4 deletions.
diff --git a/src/Traits/ChannelMethods.php b/src/Traits/ChannelMethods.php
@@ -80,7 +80,7 @@ protected static function _ChPublish(string $key, mixed $message, bool $store =
         $func = __FUNCTION__;
         $params = func_get_args();
         self::_Atomic($key, function () use (
-            $key, $message, $func, $params, $store
+            $key, $message, $func, $params, $store, $workerId
         ) {
             /**
              * [
@@ -94,16 +94,33 @@ protected static function _ChPublish(string $key, mixed $message, bool $store =
             // 如果还没有监听器，将数据投入默认
             if (!$channel) {
                 if ($store) {
-                    $channel['--default--']['value'][] = $message;
+                    // 非指定workerId
+                    if ($workerId !== null) {
+                        $channel['--default--']['value'][] = $message;
+                    }
+                    // 指定workerId
+                    else {
+                        $channel[$workerId]['value'][] = $message;
+                    }
+
                 }
             }
             // 否则将消息投入到每个worker的监听器数据中
             else {
-                foreach ($channel as $workerId => $item) {
-                    if ($store or isset($item['futureId'])) {
+                // 非指定workerId
+                if ($workerId !== null) {
+                    if ($store or isset($channel[$workerId]['futureId'])) {
                         $channel[$workerId]['value'][] = $message;
                     }
                 }
+                // 指定workerId
+                else {
+                    foreach ($channel as $workerId => $item) {
+                        if ($store or isset($item['futureId'])) {
+                            $channel[$workerId]['value'][] = $message;
+                        }
+                    }
+                }
             }
 
             self::_Set($channelName, $channel);