(pinot) branch master updated: Expose Get Segments/Table Config Change capture (#13915)

[jackie]

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

jackie 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 f50f321f05 Expose Get Segments/Table Config Change capture  (#13915)
f50f321f05 is described below

commit f50f321f054685a453d852441dea6dbafe098bad
Author: Chaitanya Deepthi <45308220+deepthi...@users.noreply.github.com>
AuthorDate: Tue Sep 3 17:25:10 2024 -0700

    Expose Get Segments/Table Config Change capture  (#13915)
---
 .../pinot/controller/util/TableMetadataReader.java | 22 +++++++++++++---------
 1 file changed, 13 insertions(+), 9 deletions(-)

diff --git 
a/pinot-controller/src/main/java/org/apache/pinot/controller/util/TableMetadataReader.java
 
b/pinot-controller/src/main/java/org/apache/pinot/controller/util/TableMetadataReader.java
index a7a53d421d..c9e87b396b 100644
--- 
a/pinot-controller/src/main/java/org/apache/pinot/controller/util/TableMetadataReader.java
+++ 
b/pinot-controller/src/main/java/org/apache/pinot/controller/util/TableMetadataReader.java
@@ -60,15 +60,7 @@ public class TableMetadataReader {
 
   public Map<String, JsonNode> getServerCheckSegmentsReloadMetadata(String 
tableNameWithType, int timeoutMs)
       throws InvalidConfigException, IOException {
-    TableType tableType = 
TableNameBuilder.getTableTypeFromTableName(tableNameWithType);
-    List<String> serverInstances = 
_pinotHelixResourceManager.getServerInstancesForTable(tableNameWithType, 
tableType);
-    Set<String> serverInstanceSet = new HashSet<>(serverInstances);
-    BiMap<String, String> endpoints = 
_pinotHelixResourceManager.getDataInstanceAdminEndpoints(serverInstanceSet);
-    ServerSegmentMetadataReader serverSegmentMetadataReader =
-        new ServerSegmentMetadataReader(_executor, _connectionManager);
-    List<String> segmentsMetadata =
-        
serverSegmentMetadataReader.getCheckReloadSegmentsFromServer(tableNameWithType, 
serverInstanceSet, endpoints,
-            timeoutMs);
+    List<String> segmentsMetadata = getReloadCheckResponses(tableNameWithType, 
timeoutMs);
     Map<String, JsonNode> response = new HashMap<>();
     for (String segmentMetadata : segmentsMetadata) {
       JsonNode responseJson = JsonUtils.stringToJsonNode(segmentMetadata);
@@ -77,6 +69,18 @@ public class TableMetadataReader {
     return response;
   }
 
+  public List<String> getReloadCheckResponses(String tableNameWithType, int 
timeoutMs)
+      throws InvalidConfigException {
+    TableType tableType = 
TableNameBuilder.getTableTypeFromTableName(tableNameWithType);
+    List<String> serverInstances = 
_pinotHelixResourceManager.getServerInstancesForTable(tableNameWithType, 
tableType);
+    Set<String> serverInstanceSet = new HashSet<>(serverInstances);
+    BiMap<String, String> endpoints = 
_pinotHelixResourceManager.getDataInstanceAdminEndpoints(serverInstanceSet);
+    ServerSegmentMetadataReader serverSegmentMetadataReader =
+        new ServerSegmentMetadataReader(_executor, _connectionManager);
+    return 
serverSegmentMetadataReader.getCheckReloadSegmentsFromServer(tableNameWithType, 
serverInstanceSet, endpoints,
+        timeoutMs);
+  }
+
   /**
    * This api takes in list of segments for which we need the metadata.
    */


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