diff --git a/gluten-flink/runtime/src/main/java/org/apache/gluten/table/runtime/config/VeloxConnectorConfig.java b/gluten-flink/runtime/src/main/java/org/apache/gluten/table/runtime/config/VeloxConnectorConfig.java index 13b195b0bd3..c3a6a4f8a11 100644 --- a/gluten-flink/runtime/src/main/java/org/apache/gluten/table/runtime/config/VeloxConnectorConfig.java +++ b/gluten-flink/runtime/src/main/java/org/apache/gluten/table/runtime/config/VeloxConnectorConfig.java @@ -38,12 +38,14 @@ public class VeloxConnectorConfig { "connector-from-elements", "connector-print"); private static final String keyTaskIndex = "task_index"; + private static final String keyTaskParallelism = "task_parallelism"; private static final String keyQueryUUId = "query_uuid"; public static ConnectorConfig getConfig(RuntimeContext context) { Map configMap = new HashMap<>(); TaskInfo taskInfo = context.getTaskInfo(); configMap.put(keyTaskIndex, String.valueOf(taskInfo.getIndexOfThisSubtask())); + configMap.put(keyTaskParallelism, String.valueOf(taskInfo.getNumberOfParallelSubtasks())); configMap.put( keyQueryUUId, UUID.nameUUIDFromBytes(context.getJobInfo().getJobId().toHexString().getBytes())