fix: add keep alive params to grpc server (#1858)

common/config/root.go

@@ -7,6 +7,7 @@ import (
 
 	"github.com/goccy/go-yaml"
 	"github.com/odigos-io/odigos/common"
+	"github.com/odigos-io/odigos/common/consts"
 )
 
 const (
@@ -151,6 +152,13 @@ func getBasicConfig(memoryLimiterConfig GenericMap) (*Config, []string) {
 							// setting it to a large value to avoid dropping batches.
 							"max_recv_msg_size_mib": 128,
 							"endpoint":              "0.0.0.0:4317",
+							// The Node Collector opens a gRPC stream to send data. This ensures that the Node Collector establishes a new connection when the Gateway scales up to include additional instances.
+							"keepalive": GenericMap{
+								"server_parameters": GenericMap{
+									"max_connection_age":       consts.GatewayMaxConnectionAge,
+									"max_connection_age_grace": consts.GatewayMaxConnectionAgeGrace,
+								},
+							},
 						},
 						// Node collectors send in gRPC, so this is probably not needed
 						"http": GenericMap{

common/config/testdata/debugexporter.yaml

@@ -3,6 +3,10 @@ receivers:
     protocols:
       grpc:
         endpoint: 0.0.0.0:4317
+        keepalive:
+          server_parameters:
+            max_connection_age: 15s
+            max_connection_age_grace: 2s
         max_recv_msg_size_mib: 128
       http:
         endpoint: 0.0.0.0:4318

common/config/testdata/minimal.yaml

@@ -3,6 +3,10 @@ receivers:
     protocols:
       grpc:
         endpoint: 0.0.0.0:4317
+        keepalive:
+          server_parameters:
+            max_connection_age: 15s
+            max_connection_age_grace: 2s
         max_recv_msg_size_mib: 128
       http:
         endpoint: 0.0.0.0:4318

common/consts/consts.go

@@ -17,6 +17,10 @@ const (
 	InstrumentationDisabled      = "disabled"
 	OdigosReportedNameAnnotation = "odigos.io/reported-name"
 
+	// GatewayMaxConnectionAge and GatewayMaxConnectionAgeGrace are the default values for the gateway collector.
+	GatewayMaxConnectionAge      = "15s"
+	GatewayMaxConnectionAgeGrace = "2s"
+
 	// Used to store the original value of the environment variable in the pod manifest.
 	// This is used to restore the original value when an instrumentation is removed
 	// or odigos is uninstalled.