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 2e6e3d6119 Increase visibility of PinotTaskManager class for better
pluggability (#14717)
2e6e3d6119 is described below
commit 2e6e3d6119c67d41675f0c967668e46dbc166be6
Author: Kartik Khare <[email protected]>
AuthorDate: Thu Dec 26 09:35:19 2024 +0530
Increase visibility of PinotTaskManager class for better pluggability
(#14717)
---
.../pinot/controller/helix/core/minion/PinotTaskManager.java | 10 +++++-----
1 file changed, 5 insertions(+), 5 deletions(-)
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 94facbc377..37208d7fd0 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
@@ -548,7 +548,7 @@ public class PinotTaskManager extends
ControllerPeriodicTask<Void> {
* Helper method to schedule tasks (all task types) for the given tables
that have the tasks enabled. Returns a map
* from the task type to the list of the tasks scheduled.
*/
- private synchronized Map<String, List<String>> scheduleTasks(List<String>
tableNamesWithType, boolean isLeader,
+ protected synchronized Map<String, List<String>> scheduleTasks(List<String>
tableNamesWithType, boolean isLeader,
@Nullable String minionInstanceTag) {
_controllerMetrics.addMeteredGlobalValue(ControllerMeter.NUMBER_TIMES_SCHEDULE_TASKS_CALLED,
1L);
@@ -586,7 +586,7 @@ public class PinotTaskManager extends
ControllerPeriodicTask<Void> {
}
@Nullable
- private synchronized List<String> scheduleTask(String taskType, List<String>
tables,
+ protected synchronized List<String> scheduleTask(String taskType,
List<String> tables,
@Nullable String minionInstanceTag) {
PinotTaskGenerator taskGenerator =
_taskGeneratorRegistry.getTaskGenerator(taskType);
Preconditions.checkState(taskGenerator != null, "Task type: %s is not
registered", taskType);
@@ -611,7 +611,7 @@ public class PinotTaskManager extends
ControllerPeriodicTask<Void> {
* Returns the list of task names, or {@code null} if no task is scheduled.
*/
@Nullable
- private List<String> scheduleTask(PinotTaskGenerator taskGenerator,
List<TableConfig> enabledTableConfigs,
+ protected List<String> scheduleTask(PinotTaskGenerator taskGenerator,
List<TableConfig> enabledTableConfigs,
boolean isLeader, @Nullable String minionInstanceTagForTask) {
String taskType = taskGenerator.getTaskType();
List<String> enabledTables =
@@ -744,7 +744,7 @@ public class PinotTaskManager extends
ControllerPeriodicTask<Void> {
}
}
- private synchronized void addTaskTypeMetricsUpdaterIfNeeded(String taskType)
{
+ protected synchronized void addTaskTypeMetricsUpdaterIfNeeded(String
taskType) {
if (!_taskTypeMetricsUpdaterMap.containsKey(taskType)) {
TaskTypeMetricsUpdater taskTypeMetricsUpdater = new
TaskTypeMetricsUpdater(taskType, this);
_pinotHelixResourceManager.getPropertyStore()
@@ -753,7 +753,7 @@ public class PinotTaskManager extends
ControllerPeriodicTask<Void> {
}
}
- private boolean isTaskSchedulable(String taskType, List<String> tables) {
+ protected boolean isTaskSchedulable(String taskType, List<String> tables) {
TaskState taskQueueState =
_helixTaskResourceManager.getTaskQueueState(taskType);
if (TaskState.STOPPED.equals(taskQueueState) ||
TaskState.STOPPING.equals(taskQueueState)) {
LOGGER.warn("Task queue is in state: {}. Tasks won't be created for
taskType: {} and tables: {}. Resume task "
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]