Make keepalive configurable

Author: Mitch528

ios/Grpc.m

@@ -17,6 +17,12 @@ @interface RCT_EXTERN_MODULE(Grpc, RCTEventEmitter)
         compressorName: (NSString *) compressorName
         limit: (NSString *) limit)
 
+RCT_EXTERN_METHOD(setKeepalive:
+(nonnull NSNumber *) enabled
+        time: (nonnull NSNumber *) time
+        timeout: (nonnull NSNumber *) timeout
+)
+
 RCT_EXTERN_METHOD(getIsInsecure:
     (RCTPromiseResolveBlock) resolve reject:
     (RCTPromiseRejectBlock) reject)

ios/RNGrpc.swift

@@ -22,6 +22,9 @@ class RNGrpc: RCTEventEmitter {
     var grpcCompression: Bool?
     var grpcCompressorName: String?
     var grpcCompressionLimit: Int?
+    var grpcKeepaliveEnabled: Bool?
+    var grpcKeepaliveTime: Int64?
+    var grpcKeepaliveTimeout: Int64?
     var calls = [Int: GrpcCall]()
 
     deinit {
@@ -44,6 +47,12 @@ class RNGrpc: RCTEventEmitter {
         self.grpcCompression = enabled.boolValue
         self.grpcCompressorName = compressorName
         self.grpcCompressionLimit = Int(limit ?? "")
+
+    @objc
+    public func setKeepalive(_ enabled: NSNumber, time: NSNumber, timeout: NSNumber) {
+        self.grpcKeepaliveEnabled = enabled.boolValue
+        self.grpcKeepaliveTime = time.int64Value
+        self.grpcKeepaliveTimeout = timeout.int64Value
     }
 
     @objc
@@ -361,6 +370,20 @@ class RNGrpc: RCTEventEmitter {
             config.maximumReceiveMessageLength = maxReceiveSize
         }
 
+        if let enabled = grpcKeepaliveEnabled, enabled {
+            let interval = self.grpcKeepaliveTime != nil ? TimeAmount.seconds(self.grpcKeepaliveTime!) : TimeAmount.nanoseconds(.max)
+
+            let timeout = TimeAmount.seconds(grpcKeepaliveTimeout ?? 20)
+
+            let keepalive = ClientConnectionKeepalive(
+                    interval: interval,
+                    timeout: timeout,
+                    permitWithoutCalls: true
+            )
+
+            config.keepalive = keepalive
+        }
+
         return try? GRPCChannelPool.with(configuration: config)
     }
 

src/client.ts

@@ -19,6 +19,7 @@ type GrpcType = {
   setHost(host: string): void;
   setInsecure(insecure: boolean): void;
   setCompression(enable: boolean, compressorName: string, limit?: string): void;
+  setKeepalive(enabled: boolean, time: number, timeout: number): void;
   setResponseSizeLimit(limitInBytes: number): void;
   unaryCall(
     id: number,
@@ -181,6 +182,9 @@ export class GrpcClient {
   ): void {
     Grpc.setCompression(enable, compressorName, limit?.toString());
   }
+  setKeepalive(enabled: boolean, time: number, timeout: number): void {
+    Grpc.setKeepalive(enabled, time, timeout);
+  }
   setResponseSizeLimit(limitInBytes: number): void {
     Grpc.setResponseSizeLimit(limitInBytes);
   }