[pinot] branch master updated: [multistage]use all servers if no server instance found for intermediate stage (#11452)

[xiangfu]

This is an automated email from the ASF dual-hosted git repository.

xiangfu pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/pinot.git


The following commit(s) were added to refs/heads/master by this push:
     new 9e0752c267 [multistage]use all servers if no server instance found for 
intermediate stage (#11452)
9e0752c267 is described below

commit 9e0752c267bbac1239ca776267c3ded113ee5a3c
Author: Haitao Zhang <hai...@startree.ai>
AuthorDate: Tue Aug 29 16:38:36 2023 -0700

    [multistage]use all servers if no server instance found for intermediate 
stage (#11452)
    
    * use all servers if no server instance found for intermediate stage
    
    * fix linting issue
    
    * address comments
---
 .../org/apache/pinot/query/routing/WorkerManager.java     | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git 
a/pinot-query-planner/src/main/java/org/apache/pinot/query/routing/WorkerManager.java
 
b/pinot-query-planner/src/main/java/org/apache/pinot/query/routing/WorkerManager.java
index 2befb33306..163ccb6bd6 100644
--- 
a/pinot-query-planner/src/main/java/org/apache/pinot/query/routing/WorkerManager.java
+++ 
b/pinot-query-planner/src/main/java/org/apache/pinot/query/routing/WorkerManager.java
@@ -313,11 +313,16 @@ public class WorkerManager {
           }
         }
       }
-      serverInstances = new ArrayList<>(servers.size());
-      for (String server : servers) {
-        ServerInstance serverInstance = enabledServerInstanceMap.get(server);
-        if (serverInstance != null) {
-          serverInstances.add(serverInstance);
+      if (servers.isEmpty()) {
+        // fall back to use all enabled servers if no server is found for the 
tables
+        serverInstances = new ArrayList<>(enabledServerInstanceMap.values());
+      } else {
+        serverInstances = new ArrayList<>(servers.size());
+        for (String server : servers) {
+          ServerInstance serverInstance = enabledServerInstanceMap.get(server);
+          if (serverInstance != null) {
+            serverInstances.add(serverInstance);
+          }
         }
       }
     }


---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscr...@pinot.apache.org
For additional commands, e-mail: commits-h...@pinot.apache.org