Hotfix: Allow for MessageBag as well as a single Message

Vorta · Mar 25, 2020 · 1961c6a · 1961c6a
1 parent 776735f
commit 1961c6a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/Connection/Nsqd.php b/src/Connection/Nsqd.php
@@ -181,10 +181,10 @@ public function handshake(Stream $stream): void
     }
 
     /**
-     * @param Message $message
+     * @param Message|MessageBag $message
      * @return bool
      */
-    public function publish(Message $message): bool
+    public function publish($message): bool
     {
         return
             $this->endpoint->getConnType() === 'tcp'

diff --git a/src/Queue.php b/src/Queue.php
@@ -9,6 +9,7 @@
 use NSQClient\Connection\Pool;
 use NSQClient\Logger\Logger;
 use NSQClient\Message\Message;
+use NSQClient\Message\Bag as MessageBag;
 
 /**
  * Class Queue
@@ -19,13 +20,13 @@ class Queue
     /**
      * @param Endpoint $endpoint
      * @param string $topic
-     * @param Message $message
+     * @param Message|MessageBag $message
      * @return bool
      */
     public static function publish(
         Endpoint $endpoint,
         string $topic,
-        Message $message
+        $message
     ): bool {
         $routes = Lookupd::getNodes($endpoint, $topic);
         $route = $routes[rand(0, count($routes) - 1)];