Merge pull request #120 from viras777/master

Fix drop connection by RabbitMQ
jakubkulhan · May 19, 2024 · 4842701 · 4842701
2 parents 6cdf0ed + 96b55f6
commit 4842701
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/src/Bunny/Async/Client.php b/src/Bunny/Async/Client.php
@@ -297,7 +297,16 @@ public function addAwaitCallback(callable $callback)
      */
     public function onDataAvailable()
     {
-        $this->read();
+        try {
+            $this->read();
+        } catch (\Throwable $e) {
+            foreach ($this->awaitCallbacks as $k => $callback) {
+                if ($callback($e) === true) {
+                    unset($this->awaitCallbacks[$k]);
+                    break;
+                }
+            }
+        }
 
         while (($frame = $this->reader->consumeFrame($this->readBuffer)) !== null) {
             foreach ($this->awaitCallbacks as $k => $callback) {

diff --git a/src/Bunny/Client.php b/src/Bunny/Client.php
@@ -191,7 +191,7 @@ public function run($maxSeconds = null)
                     $now = microtime(true);
                     $nextStreamSelectTimeout = ($this->lastWrite ?: $now) + $this->options["heartbeat"];
                     if (!isset($nextHeartbeat)) {
-                        $nextHeartbeat = $now + $this->options["heartbeat"];
+                        $nextHeartbeat = $nextStreamSelectTimeout;;
                     }
                     if ($stopTime !== null && $stopTime < $nextStreamSelectTimeout) {
                         $nextStreamSelectTimeout = $stopTime;