From 3b5906f5187d92e6464295f7ab051d1d58b74dca Mon Sep 17 00:00:00 2001
From: guandeng <guandeng@Hotmail.com>
Date: Tue, 12 Dec 2023 15:37:26 +0800
Subject: [PATCH] Support `ping` method for `grpc client`. (#6373)

---
 src/GrpcClient.php | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/src/GrpcClient.php b/src/GrpcClient.php
index 07f3796..2c56d4b 100644
--- a/src/GrpcClient.php
+++ b/src/GrpcClient.php
@@ -19,6 +19,7 @@
 use InvalidArgumentException;
 use RuntimeException;
 use Swoole\Coroutine\Http2\Client as SwooleHttp2Client;
+use Swoole\Http2\Response;
 
 class GrpcClient
 {
@@ -136,7 +137,7 @@ public function closeRecv()
                 // If this channel has pending pop, we should push 'false' to negate the pop.
                 // Otherwise we should release it directly.
                 while ($channel->stats()['consumer_num'] !== 0) {
-                    $channel->push(false);
+                    $channel->push(-1);
                 }
                 $this->channelPool->release($channel);
             }
@@ -234,12 +235,12 @@ public function recv(int $streamId, float $timeout = null)
         $channel = $this->recvChannelMap[$streamId] ?? null;
         if ($channel instanceof Channel) {
             $response = $channel->pop($timeout === null ? $this->timeout : $timeout);
-            // Pop timeout
-            if ($response === false && $channel->errCode === SWOOLE_CHANNEL_TIMEOUT) {
+            if ($response === -1) {
                 unset($this->recvChannelMap[$streamId]);
+                return false;
             }
             // Unset recvChannelMap arfter recv
-            if (! $response->pipeline) {
+            if (($response === false && $channel->errCode === SWOOLE_CHANNEL_TIMEOUT) || ($response instanceof Response && ! $response->pipeline)) {
                 unset($this->recvChannelMap[$streamId]);
                 if (! $channel->isEmpty()) {
                     $channel->pop();
@@ -258,6 +259,11 @@ public function getErrCode(): int
         return $this->httpClient ? $this->httpClient->errCode : 0;
     }
 
+    public function ping(): bool
+    {
+        return $this->getHttpClient()->ping();
+    }
+
     /**
      * @param bool|float $yield
      */
@@ -316,6 +322,9 @@ private function runReceiveCoroutine()
                         break;
                     }
                 } else {
+                    if ($this->ping()) {
+                        continue;
+                    }
                     // If no response, then close all the connection.
                     if ($this->closeRecv()) {
                         break;