set tables for controller task scheduling (#15594)
diff --git a/pinot-controller/src/main/java/org/apache/pinot/controller/helix/core/minion/PinotTaskManager.java b/pinot-controller/src/main/java/org/apache/pinot/controller/helix/core/minion/PinotTaskManager.java
index c62d7a1..f633a7b 100644
--- a/pinot-controller/src/main/java/org/apache/pinot/controller/helix/core/minion/PinotTaskManager.java
+++ b/pinot-controller/src/main/java/org/apache/pinot/controller/helix/core/minion/PinotTaskManager.java
@@ -19,6 +19,7 @@
package org.apache.pinot.controller.helix.core.minion;
import com.google.common.base.Preconditions;
+import com.google.common.collect.ImmutableSet;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.ArrayList;
@@ -799,7 +800,8 @@
protected void processTables(List<String> tableNamesWithType, Properties taskProperties) {
TaskSchedulingContext context = new TaskSchedulingContext()
.setLeader(true)
- .setTriggeredBy(CommonConstants.TaskTriggers.CRON_TRIGGER.name());
+ .setTriggeredBy(CommonConstants.TaskTriggers.CRON_TRIGGER.name())
+ .setTablesToSchedule(ImmutableSet.copyOf(tableNamesWithType));
// cron schedule
scheduleTasks(context);
}
