add keepalive to EndpointGroupGRPCOpts (#6810)

Signed-off-by: Walther Lee <[email protected]>
thanos-io · Oct 18, 2023 · 31cba70 · 31cba70
1 parent 2b982d5
commit 31cba70
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pkg/extgrpc/client.go b/pkg/extgrpc/client.go
@@ -5,6 +5,7 @@ package extgrpc
 
 import (
 	"math"
+	"time"
 
 	"github.com/go-kit/log"
 	"github.com/go-kit/log/level"
@@ -15,6 +16,7 @@ import (
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/credentials"
 	"google.golang.org/grpc/credentials/insecure"
+	"google.golang.org/grpc/keepalive"
 
 	"github.com/thanos-io/thanos/pkg/tls"
 	"github.com/thanos-io/thanos/pkg/tracing"
@@ -38,6 +40,7 @@ func EndpointGroupGRPCOpts() []grpc.DialOption {
 
 	return []grpc.DialOption{
 		grpc.WithDefaultServiceConfig(serviceConfig),
+		grpc.WithKeepaliveParams(keepalive.ClientParameters{Time: 10 * time.Second, Timeout: 5 * time.Second}),
 	}
 }