[bugfix] Fixing the TableType val passed for unavailable segments in physical optimizer (#16660)

shauryachats · Shaurya Chaturvedi · web-flow · commit b00dbfec1b8f · 2025-08-22T11:26:18.000-05:00
Co-authored-by: Shaurya Chaturvedi &lt;shauryachats@uber.com&gt;
diff --git a/pinot-query-planner/src/main/java/org/apache/pinot/query/planner/physical/v2/opt/rules/LeafStageWorkerAssignmentRule.java b/pinot-query-planner/src/main/java/org/apache/pinot/query/planner/physical/v2/opt/rules/LeafStageWorkerAssignmentRule.java
@@ -174,7 +174,7 @@ private PhysicalTableScan assignTableScan(PhysicalTableScan tableScan, long requ
       }
       if (!routingTable.getUnavailableSegments().isEmpty()) {
         // Set unavailable segments in context, keyed by PRelNode ID.
-        segmentUnavailableMap.put(TableNameBuilder.forType(TableType.valueOf(tableName)).tableNameWithType(tableName),
+        segmentUnavailableMap.put(TableNameBuilder.forType(TableType.valueOf(tableType)).tableNameWithType(tableName),
             new HashSet<>(routingTable.getUnavailableSegments()));
       }
     }

Original file line number	Diff line number	Diff line change
`@@ -174,7 +174,7 @@ private PhysicalTableScan assignTableScan(PhysicalTableScan tableScan, long requ`
`174`	`174`	`}`
`175`	`175`	`if (!routingTable.getUnavailableSegments().isEmpty()) {`
`176`	`176`	`// Set unavailable segments in context, keyed by PRelNode ID.`
`177`		`- segmentUnavailableMap.put(TableNameBuilder.forType(TableType.valueOf(tableName)).tableNameWithType(tableName),`
	`177`	`+ segmentUnavailableMap.put(TableNameBuilder.forType(TableType.valueOf(tableType)).tableNameWithType(tableName),`
`178`	`178`	`new HashSet<>(routingTable.getUnavailableSegments()));`
`179`	`179`	`}`
`180`	`180`	`}`