add client keep alive params

Author: scott-cotton

go-routesapi/cmd/sdroutes/main.go

@@ -10,6 +10,7 @@ import (
 	"github.com/signadot/routesapi/go-routesapi"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/credentials/insecure"
+	"google.golang.org/grpc/keepalive"
 	"google.golang.org/protobuf/encoding/protojson"
 )
 
@@ -40,7 +41,12 @@ func main() {
 	flag.StringVar(&sandboxName, "sandbox-name", "", "routes routing to sandbox with given name")
 	flag.BoolVar(&watch, "watch", false, "whether to watch")
 	flag.Parse()
-	conn, err := grpc.Dial(addr, grpc.WithTransportCredentials(insecure.NewCredentials()))
+	conn, err := grpc.Dial(addr,
+		grpc.WithTransportCredentials(insecure.NewCredentials()),
+		grpc.WithKeepaliveParams(keepalive.ClientParameters{
+			Time:    5 * time.Second,
+			Timeout: 15 * time.Second,
+		}))
 	if err != nil {
 		log.Fatal(err)
 	}

go-routesapi/watched/watched.go

@@ -3,12 +3,14 @@ package watched
 import (
 	"context"
 	"sync"
+	"time"
 
 	"github.com/signadot/routesapi/go-routesapi"
 	"github.com/signadot/routesapi/go-routesapi/internal/indices"
 	"github.com/signadot/routesapi/go-routesapi/internal/queue"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/credentials/insecure"
+	"google.golang.org/grpc/keepalive"
 	"google.golang.org/protobuf/proto"
 )
 
@@ -43,7 +45,11 @@ type watched struct {
 // NewWatched creates a Watched.  The set of the workload rules returned from
 // the returned Watched corresponds to those specified in q.
 func NewWatched(ctx context.Context, cfg *Config, q *routesapi.WorkloadRoutingRulesRequest) (Watched, error) {
-	conn, err := grpc.Dial(cfg.Addr, grpc.WithTransportCredentials(insecure.NewCredentials()))
+	conn, err := grpc.Dial(cfg.Addr, grpc.WithTransportCredentials(insecure.NewCredentials()),
+		grpc.WithKeepaliveParams(keepalive.ClientParameters{
+			Time:    5 * time.Second,
+			Timeout: 15 * time.Second,
+		}))
 	if err != nil {
 		return nil, err
 	}