Allow for more frequent keepalives (#387)

* Allow for more frequent keepalives * Remove server parameter
xmtp · May 2, 2024 · 9b84f9e · 9b84f9e
1 parent 827b3f2
commit 9b84f9e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/api/server.go b/pkg/api/server.go
@@ -26,6 +26,7 @@ import (
 	"github.com/xmtp/xmtp-node-go/pkg/tracing"
 	"google.golang.org/grpc/health"
 	healthgrpc "google.golang.org/grpc/health/grpc_health_v1"
+	"google.golang.org/grpc/keepalive"
 
 	"go.uber.org/zap"
 	"google.golang.org/grpc"
@@ -141,6 +142,9 @@ func (s *Server) startGRPC() error {
 		grpc.Creds(insecure.NewCredentials()),
 		grpc.UnaryInterceptor(middleware.ChainUnaryServer(unary...)),
 		grpc.StreamInterceptor(middleware.ChainStreamServer(stream...)),
+		grpc.KeepaliveEnforcementPolicy(keepalive.EnforcementPolicy{
+			MinTime: 15 * time.Second,
+		}),
 		grpc.MaxRecvMsgSize(s.Config.Options.MaxMsgSize),
 	}
 	grpcServer := grpc.NewServer(options...)