gRPC LoadBalancingPolicy Support

quarkusio · Mar 18, 2021 · 66e8fc9 · 66e8fc9
1 parent 3781da9
commit 66e8fc9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/...ns/grpc/runtime/src/main/java/io/quarkus/grpc/runtime/config/GrpcClientConfiguration.java b/...ns/grpc/runtime/src/main/java/io/quarkus/grpc/runtime/config/GrpcClientConfiguration.java
@@ -137,4 +137,10 @@ public class GrpcClientConfiguration {
     @ConfigItem
     public Optional<String> userAgent;
 
+    /**
+     * Use a custom load balancing policy.
+     * Accepted values are: {@code pick_value}, {@code round_robin}, {@code grpclb}
+     */
+    @ConfigItem(defaultValue = "pick_first")
+    public String loadBalancingPolicy;
 }
diff --git a/extensions/grpc/runtime/src/main/java/io/quarkus/grpc/runtime/supports/Channels.java b/extensions/grpc/runtime/src/main/java/io/quarkus/grpc/runtime/supports/Channels.java
@@ -86,6 +86,7 @@ public static Channel createChannel(String name) throws SSLException {
         }
 
         NettyChannelBuilder builder = NettyChannelBuilder.forAddress(host, port)
+                .defaultLoadBalancingPolicy(config.loadBalancingPolicy)
                 .flowControlWindow(config.flowControlWindow.orElse(DEFAULT_FLOW_CONTROL_WINDOW))
                 .keepAliveWithoutCalls(config.keepAliveWithoutCalls)
                 .maxHedgedAttempts(config.maxHedgedAttempts)