|
|
|
@ -17,22 +17,19 @@
|
|
|
|
|
|
|
|
|
|
package org.apache.dolphinscheduler.server.worker.processor; |
|
|
|
|
|
|
|
|
|
import com.google.common.base.Preconditions; |
|
|
|
|
import io.netty.channel.Channel; |
|
|
|
|
import org.apache.dolphinscheduler.common.utils.JSONUtils; |
|
|
|
|
import org.apache.dolphinscheduler.remote.command.Command; |
|
|
|
|
import org.apache.dolphinscheduler.remote.command.CommandType; |
|
|
|
|
import org.apache.dolphinscheduler.remote.command.HostUpdateCommand; |
|
|
|
|
import org.apache.dolphinscheduler.remote.processor.NettyRemoteChannel; |
|
|
|
|
import org.apache.dolphinscheduler.remote.processor.NettyRequestProcessor; |
|
|
|
|
import org.apache.dolphinscheduler.service.bean.SpringApplicationContext; |
|
|
|
|
|
|
|
|
|
import org.slf4j.Logger; |
|
|
|
|
import org.slf4j.LoggerFactory; |
|
|
|
|
import org.springframework.beans.factory.annotation.Autowired; |
|
|
|
|
import org.springframework.stereotype.Component; |
|
|
|
|
|
|
|
|
|
import com.google.common.base.Preconditions; |
|
|
|
|
|
|
|
|
|
import io.netty.channel.Channel; |
|
|
|
|
|
|
|
|
|
/** |
|
|
|
|
* update process host |
|
|
|
|
* this used when master failover |
|
|
|
@ -45,6 +42,7 @@ public class HostUpdateProcessor implements NettyRequestProcessor {
|
|
|
|
|
/** |
|
|
|
|
* task callback service |
|
|
|
|
*/ |
|
|
|
|
@Autowired |
|
|
|
|
private TaskCallbackService taskCallbackService; |
|
|
|
|
|
|
|
|
|
@Override |
|
|
|
@ -52,10 +50,6 @@ public class HostUpdateProcessor implements NettyRequestProcessor {
|
|
|
|
|
Preconditions.checkArgument(CommandType.PROCESS_HOST_UPDATE_REQUEST == command.getType(), String.format("invalid command type : %s", command.getType())); |
|
|
|
|
HostUpdateCommand updateCommand = JSONUtils.parseObject(command.getBody(), HostUpdateCommand.class); |
|
|
|
|
logger.info("received host update command : {}", updateCommand); |
|
|
|
|
|
|
|
|
|
if (taskCallbackService == null) { |
|
|
|
|
taskCallbackService = SpringApplicationContext.getBean(TaskCallbackService.class); |
|
|
|
|
} |
|
|
|
|
taskCallbackService.changeRemoteChannel(updateCommand.getTaskInstanceId(), new NettyRemoteChannel(channel, command.getOpaque())); |
|
|
|
|
|
|
|
|
|
} |
|
|
|
|