diff options
author | Henning Baldersheim <balder@yahoo-inc.com> | 2022-08-19 19:08:36 +0200 |
---|---|---|
committer | Henning Baldersheim <balder@yahoo-inc.com> | 2022-08-19 19:32:41 +0200 |
commit | 47b7b4eab90ec796bbb346e1b91960d2fa6241d0 (patch) | |
tree | ed0bd581df96db17e3ff2df236bfd67120680373 /container-search/src/main/java/com/yahoo/search/dispatch/Dispatcher.java | |
parent | 2194b0c451190e2c27425a21c432370d923e7190 (diff) |
Add best-of-random-2 dispatch policy.
Diffstat (limited to 'container-search/src/main/java/com/yahoo/search/dispatch/Dispatcher.java')
-rw-r--r-- | container-search/src/main/java/com/yahoo/search/dispatch/Dispatcher.java | 17 |
1 files changed, 15 insertions, 2 deletions
diff --git a/container-search/src/main/java/com/yahoo/search/dispatch/Dispatcher.java b/container-search/src/main/java/com/yahoo/search/dispatch/Dispatcher.java index 68a8e351b34..8bec018f8f8 100644 --- a/container-search/src/main/java/com/yahoo/search/dispatch/Dispatcher.java +++ b/container-search/src/main/java/com/yahoo/search/dispatch/Dispatcher.java @@ -96,6 +96,20 @@ public class Dispatcher extends AbstractComponent { this(new ClusterMonitor<>(searchCluster, true), searchCluster, dispatchConfig, new RpcInvokerFactory(resourcePool, searchCluster), metric); } + private static LoadBalancer.Policy toLoadBalancerPolicy(DispatchConfig.DistributionPolicy.Enum policy) { + if (policy == DispatchConfig.DistributionPolicy.ROUNDROBIN) { + return LoadBalancer.Policy.ROUNDROBIN; + } else if (policy == DispatchConfig.DistributionPolicy.BEST_OF_RANDOM_2) { + return LoadBalancer.Policy.BEST_OF_RANDOM_2; + } else if (policy == DispatchConfig.DistributionPolicy.LATENCY_AMORTIZED_OVER_REQUESTS) { + return LoadBalancer.Policy.LATENCY_AMORTIZED_OVER_REQUESTS; + } else if (policy == DispatchConfig.DistributionPolicy.LATENCY_AMORTIZED_OVER_TIME) { + return LoadBalancer.Policy.LATENCY_AMORTIZED_OVER_TIME; + } else { + return LoadBalancer.Policy.LATENCY_AMORTIZED_OVER_REQUESTS; + } + } + /* Protected for simple mocking in tests. Beware that searchCluster is shutdown on in deconstruct() */ protected Dispatcher(ClusterMonitor<Node> clusterMonitor, SearchCluster searchCluster, @@ -107,8 +121,7 @@ public class Dispatcher extends AbstractComponent { this.searchCluster = searchCluster; this.clusterMonitor = clusterMonitor; - this.loadBalancer = new LoadBalancer(searchCluster, - dispatchConfig.distributionPolicy() == DispatchConfig.DistributionPolicy.ROUNDROBIN); + this.loadBalancer = new LoadBalancer(searchCluster, toLoadBalancerPolicy(dispatchConfig.distributionPolicy())); this.invokerFactory = invokerFactory; this.metric = metric; this.metricContext = metric.createContext(null); |