|
|
@ -28,10 +28,14 @@ import org.apache.dolphinscheduler.server.master.runner.task.ITaskProcessor; |
|
|
|
import org.apache.dolphinscheduler.server.master.runner.task.TaskAction; |
|
|
|
import org.apache.dolphinscheduler.server.master.runner.task.TaskAction; |
|
|
|
|
|
|
|
|
|
|
|
import java.util.Map; |
|
|
|
import java.util.Map; |
|
|
|
|
|
|
|
import org.slf4j.Logger; |
|
|
|
|
|
|
|
import org.slf4j.LoggerFactory; |
|
|
|
|
|
|
|
|
|
|
|
@AutoService(StateEventHandler.class) |
|
|
|
@AutoService(StateEventHandler.class) |
|
|
|
public class TaskTimeoutStateEventHandler implements StateEventHandler { |
|
|
|
public class TaskTimeoutStateEventHandler implements StateEventHandler { |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private static final Logger logger = LoggerFactory.getLogger(TaskTimeoutStateEventHandler.class); |
|
|
|
|
|
|
|
|
|
|
|
@Override |
|
|
|
@Override |
|
|
|
public boolean handleStateEvent(WorkflowExecuteRunnable workflowExecuteRunnable, |
|
|
|
public boolean handleStateEvent(WorkflowExecuteRunnable workflowExecuteRunnable, |
|
|
|
StateEvent stateEvent) throws StateEventHandleError { |
|
|
|
StateEvent stateEvent) throws StateEventHandleError { |
|
|
@ -49,14 +53,23 @@ public class TaskTimeoutStateEventHandler implements StateEventHandler { |
|
|
|
if (TimeoutFlag.CLOSE == taskInstance.getTaskDefine().getTimeoutFlag()) { |
|
|
|
if (TimeoutFlag.CLOSE == taskInstance.getTaskDefine().getTimeoutFlag()) { |
|
|
|
return true; |
|
|
|
return true; |
|
|
|
} |
|
|
|
} |
|
|
|
TaskTimeoutStrategy taskTimeoutStrategy = taskInstance.getTaskDefine().getTimeoutNotifyStrategy(); |
|
|
|
TaskTimeoutStrategy taskTimeoutStrategy = taskInstance.getTaskDefine() |
|
|
|
Map<Long, ITaskProcessor> activeTaskProcessMap = workflowExecuteRunnable.getActiveTaskProcessMap(); |
|
|
|
.getTimeoutNotifyStrategy(); |
|
|
|
if (TaskTimeoutStrategy.FAILED == taskTimeoutStrategy |
|
|
|
Map<Long, ITaskProcessor> activeTaskProcessMap = workflowExecuteRunnable |
|
|
|
|| TaskTimeoutStrategy.WARNFAILED == taskTimeoutStrategy) { |
|
|
|
.getActiveTaskProcessMap(); |
|
|
|
|
|
|
|
if ((TaskTimeoutStrategy.FAILED == taskTimeoutStrategy |
|
|
|
|
|
|
|
|| TaskTimeoutStrategy.WARNFAILED == taskTimeoutStrategy)) { |
|
|
|
|
|
|
|
if (activeTaskProcessMap.containsKey(taskInstance.getTaskCode())) { |
|
|
|
ITaskProcessor taskProcessor = activeTaskProcessMap.get(taskInstance.getTaskCode()); |
|
|
|
ITaskProcessor taskProcessor = activeTaskProcessMap.get(taskInstance.getTaskCode()); |
|
|
|
taskProcessor.action(TaskAction.TIMEOUT); |
|
|
|
taskProcessor.action(TaskAction.TIMEOUT); |
|
|
|
|
|
|
|
} else { |
|
|
|
|
|
|
|
logger.warn( |
|
|
|
|
|
|
|
"cannot find the task processor for task {}, so skip task processor action.", |
|
|
|
|
|
|
|
taskInstance.getTaskCode()); |
|
|
|
} |
|
|
|
} |
|
|
|
if (TaskTimeoutStrategy.WARN == taskTimeoutStrategy || TaskTimeoutStrategy.WARNFAILED == taskTimeoutStrategy) { |
|
|
|
} |
|
|
|
|
|
|
|
if (TaskTimeoutStrategy.WARN == taskTimeoutStrategy |
|
|
|
|
|
|
|
|| TaskTimeoutStrategy.WARNFAILED == taskTimeoutStrategy) { |
|
|
|
workflowExecuteRunnable.processTimeout(); |
|
|
|
workflowExecuteRunnable.processTimeout(); |
|
|
|
workflowExecuteRunnable.taskTimeout(taskInstance); |
|
|
|
workflowExecuteRunnable.taskTimeout(taskInstance); |
|
|
|
} |
|
|
|
} |
|
|
|