|
|
|
@ -25,6 +25,9 @@ import java.util.Date;
|
|
|
|
import java.util.List;
|
|
|
|
import java.util.List;
|
|
|
|
import java.util.Map;
|
|
|
|
import java.util.Map;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
import static com.docus.server.common.MsgConstants.COLLECTOR_RESTART;
|
|
|
|
|
|
|
|
import static com.docus.server.common.MsgConstants.HAS_COLLECTOR_COUNT;
|
|
|
|
|
|
|
|
|
|
|
|
@Slf4j
|
|
|
|
@Slf4j
|
|
|
|
@AllArgsConstructor
|
|
|
|
@AllArgsConstructor
|
|
|
|
@NoArgsConstructor
|
|
|
|
@NoArgsConstructor
|
|
|
|
@ -41,102 +44,161 @@ public class ClientHandler extends SimpleChannelInboundHandler<CommMsg<Serializa
|
|
|
|
|
|
|
|
|
|
|
|
log.info("======== 【终端客户端-收到调度器消息】, 消息时间={}, 消息类型={}, 消息内容={}", messageTime, messageType, messageContent + " ======== ");
|
|
|
|
log.info("======== 【终端客户端-收到调度器消息】, 消息时间={}, 消息类型={}, 消息内容={}", messageTime, messageType, messageContent + " ======== ");
|
|
|
|
|
|
|
|
|
|
|
|
if (messageType.equals(MsgConstants.TERMINATOR_RESTART)) {
|
|
|
|
switch (messageType) {
|
|
|
|
log.info("【终端客户端-收到调度器消息】接受到终端重启命令，内容={}", messageContent);
|
|
|
|
case MsgConstants.TERMINATOR_RESTART:
|
|
|
|
|
|
|
|
terminatorRestart(messageContent);
|
|
|
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
case MsgConstants.COLLECTOR_RESTART:
|
|
|
|
|
|
|
|
collectorRestart(messageContent);
|
|
|
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
case MsgConstants.VIRTUAL_RESTART:
|
|
|
|
|
|
|
|
virtualRestart(messageContent);
|
|
|
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
case MsgConstants.UPDATE_COLLECTOR_FILE:
|
|
|
|
|
|
|
|
updateCollectorFile(messageContent);
|
|
|
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
case MsgConstants.UPDATE_COLLECTOR_CONFIG:
|
|
|
|
|
|
|
|
updateCollectorConfig(messageContent);
|
|
|
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
case MsgConstants.SCH_DISTRIBUTE_TASKS:
|
|
|
|
|
|
|
|
schDistributeTasks(commMsg, messageContent);
|
|
|
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
case MsgConstants.HAS_VALID_COLLECTOR:
|
|
|
|
|
|
|
|
hasValidCollector(messageContent);
|
|
|
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
case MsgConstants.HAS_COLLECTOR_COUNT:
|
|
|
|
|
|
|
|
hasCollectorCount(context, messageContent);
|
|
|
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
default:
|
|
|
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
} else if (messageType.equals(MsgConstants.COLLECTOR_RESTART)) {
|
|
|
|
private void hasCollectorCount(ChannelHandlerContext context, String messageContent) {
|
|
|
|
log.info("【终端客户端-收到调度器消息】收到采集器重启命令，内容={}", messageContent);
|
|
|
|
log.info("【终端客户端-收到调度器消息】收到获取采集器数量命令，内容={}", messageContent);
|
|
|
|
|
|
|
|
|
|
|
|
} else if (messageType.equals(MsgConstants.VIRTUAL_RESTART)) {
|
|
|
|
CollectorChannelCacheMap collectorChannelCacheMap = nettyClient.getCollectorChannelCacheMap();
|
|
|
|
log.info("【终端客户端-收到调度器消息】收到虚拟机重启命令，内容={}", messageContent);
|
|
|
|
Map<String, Channel> collectorMap = collectorChannelCacheMap.getAll();
|
|
|
|
|
|
|
|
|
|
|
|
} else if (messageType.equals(MsgConstants.UPDATE_COLLECTOR_FILE)) {
|
|
|
|
String packageValue = nettyClient.getRedisMsg();
|
|
|
|
log.info("【终端客户端-收到调度器消息】收到更新采集器文件命令，内容={}", messageContent);
|
|
|
|
List<LoadSchCollectorVO> redisLoadSchCollectors = JSON.fromJSONWithGeneric(packageValue, new TypeReference<List<LoadSchCollectorVO>>() {
|
|
|
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
|
|
} else if (messageType.equals(MsgConstants.UPDATE_COLLECTOR_CONFIG)) {
|
|
|
|
List<LoadSchCollectorVO> objects = Lists.newArrayList();
|
|
|
|
log.info("【终端客户端-收到调度器消息】收到更新采集器配置命令，内容={}", messageContent);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
} else if (messageType.equals(MsgConstants.SCH_DISTRIBUTE_TASKS)) {
|
|
|
|
for (LoadSchCollectorVO loadSchCollectorVO : redisLoadSchCollectors) {
|
|
|
|
log.info("【终端客户端-收到调度器消息】收到采集调度器下发任务命令，内容={}", messageContent);
|
|
|
|
if (collectorMap.containsKey(String.valueOf(loadSchCollectorVO.getCollectorId()))) {
|
|
|
|
|
|
|
|
objects.add(loadSchCollectorVO);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
//下发任务： 判断这个任务采集器类型在指定目录是否存在、
|
|
|
|
CommMsg<Serializable> collectorCountMsg = CommMsg.builder()
|
|
|
|
//如果不存在：下拉该采集器类型最新的版本更新包到指定的部署路径，并且启动该exe采集器，将任务tcp给对应采集器
|
|
|
|
.messageType(HAS_COLLECTOR_COUNT)
|
|
|
|
//如果存在：
|
|
|
|
.messageTime(DateUtil.formatDateTime(new Date()))
|
|
|
|
|
|
|
|
.content(JSON.toJSON(objects))
|
|
|
|
|
|
|
|
.build();
|
|
|
|
|
|
|
|
|
|
|
|
SchCollectorTaskDTO schCollectorTaskDTO = JSON.fromJSON(messageContent, SchCollectorTaskDTO.class);
|
|
|
|
context.channel().writeAndFlush(collectorCountMsg);
|
|
|
|
String collectorId = schCollectorTaskDTO.getTaskInfo().getTasks().get(0).getCollectorId();
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
Long compareCollectorId = Long.valueOf(collectorId);
|
|
|
|
private void hasValidCollector(String messageContent) {
|
|
|
|
|
|
|
|
log.info("【终端客户端-收到调度器消息】是否有可用类型的采集器命令，内容={}", messageContent);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
String packageValue = nettyClient.getRedisMsg();
|
|
|
|
private void schDistributeTasks(CommMsg commMsg, String messageContent) throws Exception {
|
|
|
|
CollectorChannelCacheMap collectorChannelCacheMap = nettyClient.getCollectorChannelCacheMap();
|
|
|
|
log.info("【终端客户端-收到调度器消息】收到采集调度器下发任务命令，内容={}", messageContent);
|
|
|
|
String saveCollectorPackagePath = nettyClient.getSaveCollectorPackagePath();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
List<LoadSchCollectorVO> redisLoadSchCollectors = JSON.fromJSONWithGeneric(packageValue, new TypeReference<List<LoadSchCollectorVO>>() {
|
|
|
|
//下发任务： 判断这个任务采集器类型在指定目录是否存在、
|
|
|
|
});
|
|
|
|
//如果不存在：下拉该采集器类型最新的版本更新包到指定的部署路径，并且启动该exe采集器，将任务tcp给对应采集器
|
|
|
|
|
|
|
|
//如果存在：
|
|
|
|
|
|
|
|
|
|
|
|
Map<Long, LoadSchCollectorVO> collectorVoMap = ListUtils.toMap(redisLoadSchCollectors, LoadSchCollectorVO::getCollectorId);
|
|
|
|
SchCollectorTaskDTO schCollectorTaskDTO = JSON.fromJSON(messageContent, SchCollectorTaskDTO.class);
|
|
|
|
|
|
|
|
String collectorId = schCollectorTaskDTO.getTaskInfo().getTasks().get(0).getCollectorId();
|
|
|
|
|
|
|
|
|
|
|
|
if (collectorVoMap.containsKey(compareCollectorId)) {
|
|
|
|
Long compareCollectorId = Long.valueOf(collectorId);
|
|
|
|
LoadSchCollectorVO loadSchCollectorVO = collectorVoMap.get(compareCollectorId);
|
|
|
|
|
|
|
|
String processName = loadSchCollectorVO.getProcessName();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if (Func.isNotBlank(processName)) {
|
|
|
|
String packageValue = nettyClient.getRedisMsg();
|
|
|
|
boolean isExistProcess = StartUpExeUtils.checkProcessOnly(processName);
|
|
|
|
CollectorChannelCacheMap collectorChannelCacheMap = nettyClient.getCollectorChannelCacheMap();
|
|
|
|
//存在指定采集器id进程
|
|
|
|
String saveCollectorPackagePath = nettyClient.getSaveCollectorPackagePath();
|
|
|
|
if (isExistProcess) {
|
|
|
|
|
|
|
|
collectorChannelCacheMap.writeAndFlush(collectorId, commMsg);
|
|
|
|
|
|
|
|
} else {
|
|
|
|
|
|
|
|
//不存在进程，则启动指定采集器进程,等待启动完成下发任务
|
|
|
|
|
|
|
|
StartUpExeUtils.startUpExeOnly(saveCollectorPackagePath + collectorId + "\\collector\\" + processName);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
while (true) {
|
|
|
|
List<LoadSchCollectorVO> redisLoadSchCollectors = JSON.fromJSONWithGeneric(packageValue, new TypeReference<List<LoadSchCollectorVO>>() {
|
|
|
|
Channel channel = collectorChannelCacheMap.get(collectorId);
|
|
|
|
});
|
|
|
|
log.info("等待采集器={}启动中", collectorId);
|
|
|
|
|
|
|
|
if (null != channel && channel.isOpen()) {
|
|
|
|
|
|
|
|
log.info("等待采集器={}启动成功并注册到终端", collectorId);
|
|
|
|
|
|
|
|
collectorChannelCacheMap.writeAndFlush(collectorId, commMsg);
|
|
|
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Map<Long, LoadSchCollectorVO> collectorVoMap = ListUtils.toMap(redisLoadSchCollectors, LoadSchCollectorVO::getCollectorId);
|
|
|
|
|
|
|
|
|
|
|
|
} else if (messageType.equals(MsgConstants.HAS_VALID_COLLECTOR)) {
|
|
|
|
if (collectorVoMap.containsKey(compareCollectorId)) {
|
|
|
|
log.info("【终端客户端-收到调度器消息】是否有可用类型的采集器命令，内容={}", messageContent);
|
|
|
|
LoadSchCollectorVO loadSchCollectorVO = collectorVoMap.get(compareCollectorId);
|
|
|
|
|
|
|
|
String processName = loadSchCollectorVO.getProcessName();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if (Func.isNotBlank(processName)) {
|
|
|
|
|
|
|
|
boolean isExistProcess = StartUpExeUtils.checkProcessOnly(processName);
|
|
|
|
|
|
|
|
//存在指定采集器id进程
|
|
|
|
|
|
|
|
if (isExistProcess) {
|
|
|
|
|
|
|
|
collectorChannelCacheMap.writeAndFlush(collectorId, commMsg);
|
|
|
|
|
|
|
|
} else {
|
|
|
|
|
|
|
|
//不存在进程，则启动指定采集器进程,等待启动完成下发任务
|
|
|
|
|
|
|
|
StartUpExeUtils.startUpExeOnly(saveCollectorPackagePath + collectorId + "\\collector\\" + processName);
|
|
|
|
|
|
|
|
|
|
|
|
//下发任务： 判断这个任务采集器类型在指定目录是否存在、
|
|
|
|
while (true) {
|
|
|
|
//如果不存在：下拉该采集器类型最新的版本更新包到指定的部署路径，并且启动该exe采集器，将任务tcp给对应采集器
|
|
|
|
Channel channel = collectorChannelCacheMap.get(collectorId);
|
|
|
|
//如果存在：
|
|
|
|
log.info("等待采集器={}启动中", collectorId);
|
|
|
|
} else if (messageType.equals(MsgConstants.HAS_COLLECTOR_COUNT)) {
|
|
|
|
if (null != channel && channel.isOpen()) {
|
|
|
|
log.info("【终端客户端-收到调度器消息】收到获取采集器数量命令，内容={}", messageContent);
|
|
|
|
log.info("等待采集器={}启动成功并注册到终端", collectorId);
|
|
|
|
|
|
|
|
collectorChannelCacheMap.writeAndFlush(collectorId, commMsg);
|
|
|
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
Thread.sleep(5000);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
CollectorChannelCacheMap collectorChannelCacheMap = nettyClient.getCollectorChannelCacheMap();
|
|
|
|
private void updateCollectorConfig(String messageContent) {
|
|
|
|
Map<String, Channel> collectorMap = collectorChannelCacheMap.getAll();
|
|
|
|
log.info("【终端客户端-收到调度器消息】收到更新采集器配置命令，内容={}", messageContent);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
String packageValue = nettyClient.getRedisMsg();
|
|
|
|
private void updateCollectorFile(String messageContent) {
|
|
|
|
List<LoadSchCollectorVO> redisLoadSchCollectors = JSON.fromJSONWithGeneric(packageValue, new TypeReference<List<LoadSchCollectorVO>>() {
|
|
|
|
log.info("【终端客户端-收到调度器消息】收到更新采集器文件命令，内容={}", messageContent);
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
List<LoadSchCollectorVO> objects = Lists.newArrayList();
|
|
|
|
private void virtualRestart(String messageContent) {
|
|
|
|
|
|
|
|
log.info("【终端客户端-收到调度器消息】收到虚拟机重启命令，内容={}", messageContent);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
for (LoadSchCollectorVO loadSchCollectorVO : redisLoadSchCollectors) {
|
|
|
|
private void collectorRestart(String messageContent) {
|
|
|
|
if (collectorMap.containsKey(String.valueOf(loadSchCollectorVO.getCollectorId()))) {
|
|
|
|
log.info("【终端客户端-收到调度器消息】收到采集器重启命令，内容={}", messageContent);
|
|
|
|
objects.add(loadSchCollectorVO);
|
|
|
|
String packageValue = nettyClient.getRedisMsg();
|
|
|
|
|
|
|
|
String saveCollectorPackagePath = nettyClient.getSaveCollectorPackagePath();
|
|
|
|
|
|
|
|
List<LoadSchCollectorVO> redisLoadSchCollectors = JSON.fromJSONWithGeneric(packageValue, new TypeReference<List<LoadSchCollectorVO>>() {
|
|
|
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Map<Long, LoadSchCollectorVO> collectorVoMap = ListUtils.toMap(redisLoadSchCollectors, LoadSchCollectorVO::getCollectorId);
|
|
|
|
|
|
|
|
Long collectorId = Long.valueOf(messageContent);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if (collectorVoMap.containsKey(collectorId)) {
|
|
|
|
|
|
|
|
LoadSchCollectorVO loadSchCollectorVO = collectorVoMap.get(collectorId);
|
|
|
|
|
|
|
|
String processName = loadSchCollectorVO.getProcessName();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if (Func.isNotBlank(processName)) {
|
|
|
|
|
|
|
|
boolean isExistProcess = StartUpExeUtils.checkProcessOnly(processName);
|
|
|
|
|
|
|
|
//存在指定采集器id进程
|
|
|
|
|
|
|
|
if (isExistProcess) {
|
|
|
|
|
|
|
|
boolean result = StartUpExeUtils.startUpExe(saveCollectorPackagePath + collectorId + "\\collector\\" + processName, processName);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
CommMsg<Serializable> collectorRestartMsg = CommMsg.builder()
|
|
|
|
|
|
|
|
.content(result)
|
|
|
|
|
|
|
|
.messageTime(DateUtil.formatDateTime(new Date()))
|
|
|
|
|
|
|
|
.messageType(COLLECTOR_RESTART)
|
|
|
|
|
|
|
|
.build();
|
|
|
|
|
|
|
|
nettyClient.getCollectorChannelCacheMap().writeAndFlush(messageContent, collectorRestartMsg);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
CommMsg<Serializable> collectorCountMsg = CommMsg.builder()
|
|
|
|
|
|
|
|
.messageType(MsgConstants.HAS_COLLECTOR_COUNT)
|
|
|
|
|
|
|
|
.messageTime(DateUtil.formatDateTime(new Date()))
|
|
|
|
|
|
|
|
.content(JSON.toJSON(objects))
|
|
|
|
|
|
|
|
.build();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
context.channel().writeAndFlush(collectorCountMsg);
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private void terminatorRestart(String messageContent) {
|
|
|
|
|
|
|
|
log.info("【终端客户端-收到调度器消息】接受到终端重启命令，内容={}", messageContent);
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
linrf
2 years ago