|
|
@ -159,6 +159,7 @@ public class TaskPriorityQueueConsumer extends Thread{ |
|
|
|
SQLTaskExecutionContext sqlTaskExecutionContext = new SQLTaskExecutionContext(); |
|
|
|
SQLTaskExecutionContext sqlTaskExecutionContext = new SQLTaskExecutionContext(); |
|
|
|
DataxTaskExecutionContext dataxTaskExecutionContext = new DataxTaskExecutionContext(); |
|
|
|
DataxTaskExecutionContext dataxTaskExecutionContext = new DataxTaskExecutionContext(); |
|
|
|
ProcedureTaskExecutionContext procedureTaskExecutionContext = new ProcedureTaskExecutionContext(); |
|
|
|
ProcedureTaskExecutionContext procedureTaskExecutionContext = new ProcedureTaskExecutionContext(); |
|
|
|
|
|
|
|
SqoopTaskExecutionContext sqoopTaskExecutionContext = new SqoopTaskExecutionContext(); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// SQL task
|
|
|
|
// SQL task
|
|
|
@ -178,6 +179,10 @@ public class TaskPriorityQueueConsumer extends Thread{ |
|
|
|
setProcedureTaskRelation(procedureTaskExecutionContext, taskNode); |
|
|
|
setProcedureTaskRelation(procedureTaskExecutionContext, taskNode); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if (taskType == TaskType.SQOOP){ |
|
|
|
|
|
|
|
setSqoopTaskRelation(sqoopTaskExecutionContext,taskNode); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
return TaskExecutionContextBuilder.get() |
|
|
|
return TaskExecutionContextBuilder.get() |
|
|
|
.buildTaskInstanceRelatedInfo(taskInstance) |
|
|
|
.buildTaskInstanceRelatedInfo(taskInstance) |
|
|
@ -222,6 +227,28 @@ public class TaskPriorityQueueConsumer extends Thread{ |
|
|
|
dataxTaskExecutionContext.setTargetConnectionParams(dataTarget.getConnectionParams()); |
|
|
|
dataxTaskExecutionContext.setTargetConnectionParams(dataTarget.getConnectionParams()); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/** |
|
|
|
|
|
|
|
* set datax task relation |
|
|
|
|
|
|
|
* @param sqoopTaskExecutionContext sqoopTaskExecutionContext |
|
|
|
|
|
|
|
* @param taskNode taskNode |
|
|
|
|
|
|
|
*/ |
|
|
|
|
|
|
|
private void setSqoopTaskRelation(SqoopTaskExecutionContext sqoopTaskExecutionContext, TaskNode taskNode) { |
|
|
|
|
|
|
|
DataxParameters dataxParameters = JSONObject.parseObject(taskNode.getParams(), DataxParameters.class); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
DataSource dataSource = processService.findDataSourceById(dataxParameters.getDataSource()); |
|
|
|
|
|
|
|
DataSource dataTarget = processService.findDataSourceById(dataxParameters.getDataTarget()); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
sqoopTaskExecutionContext.setDataSourceId(dataxParameters.getDataSource()); |
|
|
|
|
|
|
|
sqoopTaskExecutionContext.setSourcetype(dataSource.getType().getCode()); |
|
|
|
|
|
|
|
sqoopTaskExecutionContext.setSourceConnectionParams(dataSource.getConnectionParams()); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
sqoopTaskExecutionContext.setDataTargetId(dataxParameters.getDataTarget()); |
|
|
|
|
|
|
|
sqoopTaskExecutionContext.setTargetType(dataTarget.getType().getCode()); |
|
|
|
|
|
|
|
sqoopTaskExecutionContext.setTargetConnectionParams(dataTarget.getConnectionParams()); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
/** |
|
|
|
/** |
|
|
|
* set SQL task relation |
|
|
|
* set SQL task relation |
|
|
|
* @param sqlTaskExecutionContext sqlTaskExecutionContext |
|
|
|
* @param sqlTaskExecutionContext sqlTaskExecutionContext |
|
|
|