(pinot) 01/01: set tables for controller task scheduling (#15594)

[jtao]

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

jtao pushed a commit to branch hotfix-array-agg-task-mananger
in repository https://gitbox.apache.org/repos/asf/pinot.git

commit e266711f6762c8a8b8b7e955ead86c33bff780bd
Author: liranjiao <liranj...@gmail.com>
AuthorDate: Fri Apr 18 14:48:29 2025 -0700

    set tables for controller task scheduling (#15594)
---
 .../apache/pinot/controller/helix/core/minion/PinotTaskManager.java   | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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 c62d7a1035..f633a7b6c4 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 @@ public class PinotTaskManager extends 
ControllerPeriodicTask<Void> {
   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);
   }


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