somandal commented on code in PR #15360:
URL: https://github.com/apache/pinot/pull/15360#discussion_r2012274941


##########
pinot-controller/src/main/java/org/apache/pinot/controller/helix/core/rebalance/DefaultRebalancePreChecker.java:
##########
@@ -109,12 +111,18 @@ private RebalancePreCheckerResult 
checkReloadNeededOnServers(String rebalanceJob
     try (PoolingHttpClientConnectionManager connectionManager = new 
PoolingHttpClientConnectionManager()) {
       TableMetadataReader metadataReader = new 
TableMetadataReader(_executorService, connectionManager,
           _pinotHelixResourceManager);
+      // Only send needReload request to servers that are part of the current 
assignment. The tagged server list may
+      // include new servers which are part of target assignment but not 
current assignment. needReload throws an
+      // exception for servers that don't contain segments for the given table
+      Set<String> serverInstanceSet = 
getCurrentlyAssignedServerSet(currentAssignment);
       TableMetadataReader.TableReloadJsonResponse needsReloadMetadataPair =
-          
metadataReader.getServerCheckSegmentsReloadMetadata(tableNameWithType, 30_000);
+          
metadataReader.getServerCheckSegmentsReloadMetadataForServerSet(tableNameWithType,
 30_000,

Review Comment:
   done



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscr...@pinot.apache.org

For queries about this service, please contact Infrastructure at:
us...@infra.apache.org


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

Reply via email to