parent
cdf4778483
commit
531bdbfdbc
|
@ -65,35 +65,35 @@ public class TaskManager extends AbstractComponent implements ClusterStateListen
|
|||
*/
|
||||
public Task register(String type, String action, TransportRequest request) {
|
||||
Task task = request.createTask(taskIdGenerator.incrementAndGet(), type, action, request.getParentTask());
|
||||
if (task == null) {
|
||||
return null;
|
||||
}
|
||||
assert task.getParentTaskId().equals(request.getParentTask()) : "Request [ " + request + "] didn't preserve it parentTaskId";
|
||||
if (task != null) {
|
||||
if (logger.isTraceEnabled()) {
|
||||
logger.trace("register {} [{}] [{}] [{}]", task.getId(), type, action, task.getDescription());
|
||||
}
|
||||
if (logger.isTraceEnabled()) {
|
||||
logger.trace("register {} [{}] [{}] [{}]", task.getId(), type, action, task.getDescription());
|
||||
}
|
||||
|
||||
if (task instanceof CancellableTask) {
|
||||
CancellableTask cancellableTask = (CancellableTask) task;
|
||||
CancellableTaskHolder holder = new CancellableTaskHolder(cancellableTask);
|
||||
CancellableTaskHolder oldHolder = cancellableTasks.put(task.getId(), holder);
|
||||
assert oldHolder == null;
|
||||
// Check if this task was banned before we start it
|
||||
if (task.getParentTaskId().isSet() && banedParents.isEmpty() == false) {
|
||||
String reason = banedParents.get(task.getParentTaskId());
|
||||
if (reason != null) {
|
||||
try {
|
||||
holder.cancel(reason);
|
||||
throw new IllegalStateException("Task cancelled before it started: " + reason);
|
||||
} finally {
|
||||
// let's clean up the registration
|
||||
unregister(task);
|
||||
}
|
||||
if (task instanceof CancellableTask) {
|
||||
CancellableTask cancellableTask = (CancellableTask) task;
|
||||
CancellableTaskHolder holder = new CancellableTaskHolder(cancellableTask);
|
||||
CancellableTaskHolder oldHolder = cancellableTasks.put(task.getId(), holder);
|
||||
assert oldHolder == null;
|
||||
// Check if this task was banned before we start it
|
||||
if (task.getParentTaskId().isSet() && banedParents.isEmpty() == false) {
|
||||
String reason = banedParents.get(task.getParentTaskId());
|
||||
if (reason != null) {
|
||||
try {
|
||||
holder.cancel(reason);
|
||||
throw new IllegalStateException("Task cancelled before it started: " + reason);
|
||||
} finally {
|
||||
// let's clean up the registration
|
||||
unregister(task);
|
||||
}
|
||||
}
|
||||
} else {
|
||||
Task previousTask = tasks.put(task.getId(), task);
|
||||
assert previousTask == null;
|
||||
}
|
||||
|
||||
} else {
|
||||
Task previousTask = tasks.put(task.getId(), task);
|
||||
assert previousTask == null;
|
||||
}
|
||||
return task;
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue