Fix apache#16767

reele · Nov 12, 2024 · 40a3d13 · 40a3d13
1 parent 8e95c9c
commit 40a3d13
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 0 deletions.
diff --git a/.../src/main/java/org/apache/dolphinscheduler/extract/master/ILogicTaskInstanceOperator.java b/.../src/main/java/org/apache/dolphinscheduler/extract/master/ILogicTaskInstanceOperator.java
@@ -19,6 +19,14 @@
 
 import org.apache.dolphinscheduler.extract.base.RpcMethod;
 import org.apache.dolphinscheduler.extract.base.RpcService;
+import org.apache.dolphinscheduler.extract.master.transportor.LogicTaskDispatchRequest;
+import org.apache.dolphinscheduler.extract.master.transportor.LogicTaskDispatchResponse;
+import org.apache.dolphinscheduler.extract.master.transportor.LogicTaskKillRequest;
+import org.apache.dolphinscheduler.extract.master.transportor.LogicTaskKillResponse;
+import org.apache.dolphinscheduler.extract.master.transportor.LogicTaskPauseRequest;
+import org.apache.dolphinscheduler.extract.master.transportor.LogicTaskPauseResponse;
+import org.apache.dolphinscheduler.extract.master.transportor.LogicTaskTakeOverRequest;
+import org.apache.dolphinscheduler.extract.master.transportor.LogicTaskTakeOverResponse;
 
 @RpcService
 public interface ILogicTaskInstanceOperator {

diff --git a/...org/apache/dolphinscheduler/server/master/engine/task/runnable/TaskExecutionRunnable.java b/...org/apache/dolphinscheduler/server/master/engine/task/runnable/TaskExecutionRunnable.java
@@ -20,6 +20,12 @@
 import static com.google.common.base.Preconditions.checkNotNull;
 import static com.google.common.base.Preconditions.checkState;
 
+import org.apache.dolphinscheduler.dao.entity.Project;
+import org.apache.dolphinscheduler.dao.entity.TaskDefinition;
+import org.apache.dolphinscheduler.dao.entity.TaskInstance;
+import org.apache.dolphinscheduler.dao.entity.WorkflowDefinition;
+import org.apache.dolphinscheduler.dao.entity.WorkflowInstance;
+import org.apache.dolphinscheduler.dao.entity.WorkflowInstanceRelation;
 import org.apache.dolphinscheduler.dao.mapper.WorkflowInstanceMapper;
 import org.apache.dolphinscheduler.dao.mapper.WorkflowInstanceRelationMapper;
 import org.apache.dolphinscheduler.extract.base.client.Clients;

diff --git a/...ain/java/org/apache/dolphinscheduler/server/master/rpc/LogicTaskInstanceOperatorImpl.java b/...ain/java/org/apache/dolphinscheduler/server/master/rpc/LogicTaskInstanceOperatorImpl.java
@@ -18,6 +18,14 @@
 package org.apache.dolphinscheduler.server.master.rpc;
 
 import org.apache.dolphinscheduler.extract.master.ILogicTaskInstanceOperator;
+import org.apache.dolphinscheduler.extract.master.transportor.LogicTaskDispatchRequest;
+import org.apache.dolphinscheduler.extract.master.transportor.LogicTaskDispatchResponse;
+import org.apache.dolphinscheduler.extract.master.transportor.LogicTaskKillRequest;
+import org.apache.dolphinscheduler.extract.master.transportor.LogicTaskKillResponse;
+import org.apache.dolphinscheduler.extract.master.transportor.LogicTaskPauseRequest;
+import org.apache.dolphinscheduler.extract.master.transportor.LogicTaskPauseResponse;
+import org.apache.dolphinscheduler.extract.master.transportor.LogicTaskTakeOverRequest;
+import org.apache.dolphinscheduler.extract.master.transportor.LogicTaskTakeOverResponse;
 
 import lombok.extern.slf4j.Slf4j;