Merge pull request #15861 from 0SkillAllLuck/main

gRPC LoadBalancingPolicy Support
quarkusio · Mar 19, 2021 · 82e052b · 82e052b
2 parents f54f17f + 66e8fc9
commit 82e052b
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)