enable round robin option in grpc client

Currently grpc configuration doesn't provide options to change load balancing. This PR adds default_service_config in proto so users can customize grpc connections.
buildbarn · Jan 23, 2025 · 7ebb551 · 7ebb551
1 parent 1c93796
commit 7ebb551
Show file tree

Hide file tree

Showing 3 changed files with 238 additions and 213 deletions.
diff --git a/pkg/grpc/base_client_factory.go b/pkg/grpc/base_client_factory.go
@@ -188,6 +188,13 @@ func (cf baseClientFactory) NewClientFromConfiguration(config *configuration.Cli
 			NewMetadataExtractingAndForwardingStreamClientInterceptor(extractor))
 	}
 
+	// Optional: service config.
+	if serviceConfig := config.DefaultServiceConfig; serviceConfig != "" {
+		dialOptions = append(
+			dialOptions,
+			grpc.WithDefaultServiceConfig(serviceConfig))
+	}
+
 	dialOptions = append(
 		dialOptions,
 		grpc.WithChainUnaryInterceptor(unaryInterceptors...),