Skip to content

Commit 34bf5ef

Browse files
fix[backend](agentmanager): added grpc keepalive to agent-manager client to prevent idle connection resets (#2532)
1 parent 49068b4 commit 34bf5ef

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

backend/pkg/agentmanager/client.go

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ import (
66
"fmt"
77
"io"
88
"strconv"
9+
"time"
910

1011
"github.com/utmstack/utmstack/backend/pkg/authz"
1112

@@ -14,6 +15,7 @@ import (
1415
"github.com/utmstack/utmstack/backend/pkg/env"
1516
"google.golang.org/grpc"
1617
"google.golang.org/grpc/credentials"
18+
"google.golang.org/grpc/keepalive"
1719
"google.golang.org/grpc/metadata"
1820
)
1921

@@ -39,6 +41,11 @@ func NewClient() (*AgentManagerClient, error) {
3941
addr,
4042
grpc.WithTransportCredentials(tlsCreds),
4143
grpc.WithPerRPCCredentials(&internalKeyCreds{key: internalKey}),
44+
grpc.WithKeepaliveParams(keepalive.ClientParameters{
45+
Time: 30 * time.Second,
46+
Timeout: 10 * time.Second,
47+
PermitWithoutStream: true,
48+
}),
4249
)
4350
if err != nil {
4451
return nil, fmt.Errorf("agentmanager: dial %s: %w", addr, err)

0 commit comments

Comments
 (0)