|
@@ -22,6 +22,7 @@ import org.apache.dolphinscheduler.common.enums.ExecutionStatus;
|
|
|
import org.apache.dolphinscheduler.common.enums.TaskType;
|
|
|
import org.apache.dolphinscheduler.common.enums.UdfType;
|
|
|
import org.apache.dolphinscheduler.common.model.TaskNode;
|
|
|
+import org.apache.dolphinscheduler.common.task.datax.DataxParameters;
|
|
|
import org.apache.dolphinscheduler.common.task.procedure.ProcedureParameters;
|
|
|
import org.apache.dolphinscheduler.common.task.sql.SqlParameters;
|
|
|
import org.apache.dolphinscheduler.common.thread.Stopper;
|
|
@@ -178,7 +179,19 @@ public class TaskUpdateQueueConsumer extends Thread{
|
|
|
|
|
|
// DATAX task
|
|
|
if (taskType == TaskType.DATAX){
|
|
|
+ DataxParameters dataxParameters = JSONObject.parseObject(taskNode.getParams(), DataxParameters.class);
|
|
|
|
|
|
+ DataSource dataSource = processService.findDataSourceById(dataxParameters.getDataSource());
|
|
|
+ DataSource dataTarget = processService.findDataSourceById(dataxParameters.getDataTarget());
|
|
|
+
|
|
|
+
|
|
|
+ dataxTaskExecutionContext.setDataSourceId(dataxParameters.getDataSource());
|
|
|
+ dataxTaskExecutionContext.setSourcetype(dataSource.getType().getCode());
|
|
|
+ dataxTaskExecutionContext.setSourceConnectionParams(dataSource.getConnectionParams());
|
|
|
+
|
|
|
+ dataxTaskExecutionContext.setDataTargetId(dataxParameters.getDataTarget());
|
|
|
+ dataxTaskExecutionContext.setTargetType(dataTarget.getType().getCode());
|
|
|
+ dataxTaskExecutionContext.setTargetConnectionParams(dataTarget.getConnectionParams());
|
|
|
}
|
|
|
|
|
|
|