Adaptive timeout support in java dispatch

1 files changed, 3 insertions, 4 deletions

diff --git a/container-search/src/main/java/com/yahoo/search/dispatch/LoadBalancer.java b/container-search/src/main/java/com/yahoo/search/dispatch/LoadBalancer.java
index 222ae6a4ea0..df6384cf81c 100644
--- a/container-search/src/main/java/com/yahoo/search/dispatch/LoadBalancer.java
+++ b/container-search/src/main/java/com/yahoo/search/dispatch/LoadBalancer.java
@@ -1,7 +1,6 @@
 // Copyright 2018 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
 package com.yahoo.search.dispatch;
 
-import com.yahoo.search.Query;
 import com.yahoo.search.dispatch.searchcluster.Group;
 import com.yahoo.search.dispatch.searchcluster.SearchCluster;
 
@@ -45,13 +44,13 @@ public class LoadBalancer {
     }
 
     /**
-     * Select and allocate the search cluster group which is to be used for the provided query. Callers <b>must</b> call
+     * Select and allocate the search cluster group which is to be used for the next search query. Callers <b>must</b> call
      * {@link #releaseGroup} symmetrically for each taken allocation.
      *
      * @param rejectedGroups if not null, the load balancer will only return groups with IDs not in the set
      * @return The node group to target, or <i>empty</i> if the internal dispatch logic cannot be used
      */
-    public Optional<Group> takeGroupForQuery(Set<Integer> rejectedGroups) {
+    public Optional<Group> takeGroup(Set<Integer> rejectedGroups) {
         if (scoreboard == null) {
             return Optional.empty();
         }
@@ -60,7 +59,7 @@ public class LoadBalancer {
     }
 
     /**
-     * Release an allocation given by {@link #takeGroupForQuery}. The release must be done exactly once for each allocation.
+     * Release an allocation given by {@link #takeGroup}. The release must be done exactly once for each allocation.
      *
      * @param group
      *            previously allocated group