chore: add support for grpc kepalive

momentohq · Nov 22, 2024 · 721ba40 · 721ba40
1 parent 4cd8ebd
commit 721ba40
Show file tree

Hide file tree

Showing 3 changed files with 48 additions and 2 deletions.
diff --git a/src/Cache/Internal/DataGrpcManager.php b/src/Cache/Internal/DataGrpcManager.php
@@ -32,6 +32,18 @@ public function __construct(ICredentialProvider $authProvider, IConfiguration $c
         if ($forceNewChannel) {
             $channelArgs["force_new"] = $forceNewChannel;
         }
+        if ($configuration->getTransportStrategy()->getGrpcConfig()->getKeepAlivePermitWithoutCalls()) {
+            $channelArgs["grpc.keepalive_permit_without_calls"] =
+                $configuration->getTransportStrategy()->getGrpcConfig()->getKeepAlivePermitWithoutCalls();
+        }
+        if ($configuration->getTransportStrategy()->getGrpcConfig()->getKeepAliveTimeoutMS()) {
+            $channelArgs["grpc.keepalive_timeout_ms"] =
+                $configuration->getTransportStrategy()->getGrpcConfig()->getKeepAliveTimeoutMS();
+        }
+        if ($configuration->getTransportStrategy()->getGrpcConfig()->getKeepAliveTimeMS()) {
+            $channelArgs["grpc.keepalive_time_ms"] =
+                $configuration->getTransportStrategy()->getGrpcConfig()->getKeepAliveTimeMS();
+        }
         $this->channel = new Channel($endpoint, $channelArgs);
         $interceptors = [
             new AuthorizationInterceptor($authProvider->getAuthToken()),

diff --git a/src/Config/Transport/IGrpcConfiguration.php b/src/Config/Transport/IGrpcConfiguration.php
@@ -16,4 +16,11 @@ public function withForceNewChannel(bool $forceNewChannel) : IGrpcConfiguration;
     public function getNumGrpcChannels(): int;
 
     public function withNumGrpcChannels(int $numGrpcChannels): IGrpcConfiguration;
+
+    public function getKeepAlivePermitWithoutCalls(): ?int;
+
+    public function getKeepAliveTimeoutMS(): ?int;
+
+    public function getKeepAliveTimeMS(): ?int;
+
 }
diff --git a/src/Config/Transport/StaticGrpcConfiguration.php b/src/Config/Transport/StaticGrpcConfiguration.php
@@ -9,12 +9,39 @@ class StaticGrpcConfiguration implements IGrpcConfiguration
     private ?int $deadlineMilliseconds;
     private bool $forceNewChannel;
     private int $numGrpcChannels;
+    private ?int $keepAlivePermitWithoutCalls;
+    private ?int $keepAliveTimeoutMS;
+    private ?int $keepAliveTimeMS;
 
-    public function __construct(?int $deadlineMilliseconds = null, bool $forceNewChannel = false, int $numGrpcChannels = 1)
-    {
+    public function __construct(
+        ?int $deadlineMilliseconds = null,
+        bool $forceNewChannel = false,
+        int $numGrpcChannels = 1,
+        ?int $keepAlivePermitWithoutCalls = 0,
+        ?int $keepAliveTimeoutMS = null,
+        ?int $keepAliveTimeMS = null
+    ) {
         $this->deadlineMilliseconds = $deadlineMilliseconds;
         $this->forceNewChannel = $forceNewChannel;
         $this->numGrpcChannels = $numGrpcChannels;
+        $this->keepAlivePermitWithoutCalls = $keepAlivePermitWithoutCalls;
+        $this->keepAliveTimeoutMS = $keepAliveTimeoutMS;
+        $this->keepAliveTimeMS = $keepAliveTimeMS;
+    }
+
+    public function getKeepAlivePermitWithoutCalls(): ?int
+    {
+        return $this->keepAlivePermitWithoutCalls;
+    }
+
+    public function getKeepAliveTimeoutMS(): ?int
+    {
+        return $this->keepAliveTimeoutMS;
+    }
+
+    public function getKeepAliveTimeMS(): ?int
+    {
+        return $this->keepAliveTimeMS;
     }
 
     public function getDeadlineMilliseconds(): ?int