Skip to content

Commit aa73a76

Browse files
authored
Add proper toString() method to UpdateTask (#21582)
Adds a proper toString() method to ClusterService.UpdateTask
1 parent d7fa2eb commit aa73a76

File tree

1 file changed

+11
-13
lines changed

1 file changed

+11
-13
lines changed

core/src/main/java/org/elasticsearch/cluster/service/ClusterService.java

+11-13
Original file line numberDiff line numberDiff line change
@@ -451,7 +451,7 @@ public <T> void submitStateUpdateTasks(final String source,
451451
// convert to an identity map to check for dups based on update tasks semantics of using identity instead of equal
452452
final IdentityHashMap<T, ClusterStateTaskListener> tasksIdentity = new IdentityHashMap<>(tasks);
453453
final List<UpdateTask<T>> updateTasks = tasksIdentity.entrySet().stream().map(
454-
entry -> new UpdateTask<>(source, entry.getKey(), config, executor, safe(entry.getValue(), logger))
454+
entry -> new UpdateTask<>(source, entry.getKey(), config.priority(), executor, safe(entry.getValue(), logger))
455455
).collect(Collectors.toList());
456456

457457
synchronized (updateTasksPerExecutor) {
@@ -590,11 +590,11 @@ <T> void runTasksForExecutor(ClusterStateTaskExecutor<T> executor) {
590590
if (pending != null) {
591591
for (UpdateTask<T> task : pending) {
592592
if (task.processed.getAndSet(true) == false) {
593-
logger.trace("will process {}", task.toString(executor));
593+
logger.trace("will process {}", task);
594594
toExecute.add(task);
595595
processTasksBySource.computeIfAbsent(task.source, s -> new ArrayList<>()).add(task.task);
596596
} else {
597-
logger.trace("skipping {}, already processed", task.toString(executor));
597+
logger.trace("skipping {}, already processed", task);
598598
}
599599
}
600600
}
@@ -652,23 +652,23 @@ <T> void runTasksForExecutor(ClusterStateTaskExecutor<T> executor) {
652652
if (assertsEnabled) {
653653
for (UpdateTask<T> updateTask : toExecute) {
654654
assert batchResult.executionResults.containsKey(updateTask.task) :
655-
"missing task result for " + updateTask.toString(executor);
655+
"missing task result for " + updateTask;
656656
}
657657
}
658658

659659
ClusterState newClusterState = batchResult.resultingState;
660660
final ArrayList<UpdateTask<T>> proccessedListeners = new ArrayList<>();
661661
// fail all tasks that have failed and extract those that are waiting for results
662662
for (UpdateTask<T> updateTask : toExecute) {
663-
assert batchResult.executionResults.containsKey(updateTask.task) : "missing " + updateTask.toString(executor);
663+
assert batchResult.executionResults.containsKey(updateTask.task) : "missing " + updateTask;
664664
final ClusterStateTaskExecutor.TaskResult executionResult =
665665
batchResult.executionResults.get(updateTask.task);
666666
executionResult.handle(
667667
() -> proccessedListeners.add(updateTask),
668668
ex -> {
669669
logger.debug(
670670
(Supplier<?>)
671-
() -> new ParameterizedMessage("cluster state update task {} failed", updateTask.toString(executor)), ex);
671+
() -> new ParameterizedMessage("cluster state update task {} failed", updateTask), ex);
672672
updateTask.listener.onFailure(updateTask.source, ex);
673673
}
674674
);
@@ -944,16 +944,13 @@ public TimeValue ackTimeout() {
944944
class UpdateTask<T> extends SourcePrioritizedRunnable {
945945

946946
public final T task;
947-
public final ClusterStateTaskConfig config;
948-
public final ClusterStateTaskExecutor<T> executor;
949947
public final ClusterStateTaskListener listener;
948+
private final ClusterStateTaskExecutor<T> executor;
950949
public final AtomicBoolean processed = new AtomicBoolean();
951950

952-
UpdateTask(String source, T task, ClusterStateTaskConfig config, ClusterStateTaskExecutor<T> executor,
953-
ClusterStateTaskListener listener) {
954-
super(config.priority(), source);
951+
UpdateTask(String source, T task, Priority priority, ClusterStateTaskExecutor<T> executor, ClusterStateTaskListener listener) {
952+
super(priority, source);
955953
this.task = task;
956-
this.config = config;
957954
this.executor = executor;
958955
this.listener = listener;
959956
}
@@ -967,7 +964,8 @@ public void run() {
967964
}
968965
}
969966

970-
public String toString(ClusterStateTaskExecutor<T> executor) {
967+
@Override
968+
public String toString() {
971969
String taskDescription = executor.describeTasks(Collections.singletonList(task));
972970
if (taskDescription.isEmpty()) {
973971
return "[" + source + "]";

0 commit comments

Comments
 (0)