package com.ard.utils.netty.tcp;
|
|
import com.ard.alarm.radar.domain.ArdEquipRadar;
|
import com.ard.alarm.radar.service.IArdEquipRadarService;
|
import com.ard.utils.netty.tcp.handler.*;
|
import io.netty.bootstrap.Bootstrap;
|
import io.netty.channel.*;
|
import io.netty.channel.nio.NioEventLoopGroup;
|
import io.netty.channel.socket.SocketChannel;
|
import io.netty.channel.socket.nio.NioSocketChannel;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Qualifier;
|
import org.springframework.scheduling.annotation.Scheduled;
|
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
|
import org.springframework.stereotype.Component;
|
|
import javax.annotation.Resource;
|
import java.util.*;
|
import java.util.concurrent.ConcurrentHashMap;
|
import java.util.concurrent.TimeUnit;
|
|
/**
|
* 客户端初始化
|
*
|
* @author lijiamin
|
*/
|
@Component
|
@Slf4j(topic = "netty")
|
public class RadarNettyTcpClient {
|
|
@Resource
|
IArdEquipRadarService ardEquipRadarService;
|
|
@Autowired
|
@Qualifier("exec")
|
ThreadPoolTaskExecutor executorService;
|
|
EventLoopGroup eventLoopGroup = new NioEventLoopGroup();
|
|
public static ConcurrentHashMap<String, MessageHandler> SucMessageHandlerMap = new ConcurrentHashMap();//成功连接的ip端口对应的报文解析器
|
public static ConcurrentHashMap<String, Integer> aroundScanfMap = new ConcurrentHashMap();//是否启动周视
|
public static final Map<String, Channel> SERVER_MAP = new ConcurrentHashMap();
|
public static final Map<String, ArdEquipRadar> RADAR_MAP = new ConcurrentHashMap();
|
|
/**
|
* 初始化方法
|
*/
|
public void initAllChannel(List<Map<String, Object>> portHosts) {
|
for (Map<String, Object> map : portHosts) {
|
String host = (String) map.get("IP");
|
Integer port = (Integer) map.get("PORT");
|
try {
|
connectServer(host, port);
|
} catch (Exception ex) {
|
log.error(ex.getMessage());
|
}
|
}
|
}
|
|
/**
|
* 初始化Bootstrap
|
* 刘苏义
|
* 2024/3/25 9:53:57
|
*/
|
public Bootstrap getBootstrap(EventLoopGroup group) {
|
if (null == group) {
|
group = eventLoopGroup;
|
}
|
Bootstrap bootstrap = new Bootstrap();
|
bootstrap.group(group)
|
.channel(NioSocketChannel.class)
|
.option(ChannelOption.TCP_NODELAY, true)
|
.option(ChannelOption.SO_KEEPALIVE, true)
|
.handler(new ChannelInitializer<SocketChannel>() {
|
@Override
|
protected void initChannel(SocketChannel socketChannel) throws Exception {
|
ChannelPipeline pipeline = socketChannel.pipeline();
|
pipeline.addLast("handler", new ClientHandler());
|
}
|
});
|
return bootstrap;
|
}
|
|
/**
|
* 服务连接
|
* 刘苏义
|
* 2024/3/25 9:52:28
|
*/
|
public void connectServer(String host, Integer port) {
|
try {
|
//异步连接tcp服务端
|
Bootstrap bootstrap = getBootstrap(null);
|
bootstrap.remoteAddress(host, port);
|
ChannelFuture channelFuture = bootstrap.connect().sync();
|
if (channelFuture.isSuccess()) {
|
log.info("雷达【" + host + ":" + port + "】连接成功");
|
String serverKey = String.format("%s:%d", host, port);
|
SERVER_MAP.put(serverKey, channelFuture.channel());
|
SucMessageHandlerMap.put(serverKey, new MessageHandler());
|
}
|
channelFuture.channel().closeFuture().addListener(new ChannelFutureListener() {
|
public void operationComplete(ChannelFuture future) throws Exception {
|
String serverKey = String.format("%s:%d", host, port);
|
SERVER_MAP.remove(serverKey);
|
//服务端离线，触发重连操作
|
reconnect(host, port);
|
}
|
});
|
} catch (Exception e) {
|
reconnect(host, port);
|
throw new RuntimeException("雷达【" + host + ":" + port + "】连接异常:" + e.getMessage());
|
}
|
}
|
|
/**
|
* 重连服务端
|
* 刘苏义
|
* 2024/3/25 9:52:15
|
*/
|
public void reconnect(final String host, final int port) {
|
log.debug("雷达【" + host + ":" + port + "】连接失败，进行重连...");
|
executorService.submit(new Runnable() {
|
public void run() {
|
for (; ; ) {
|
try {
|
TimeUnit.SECONDS.sleep(3);
|
connectServer(host, port);
|
if (SERVER_MAP.containsKey(String.format("%s:%d", host, port))) {
|
break;
|
}
|
} catch (Exception e) {
|
log.error(e.getMessage());
|
break;
|
}
|
}
|
}
|
});
|
}
|
|
/**
|
* 监测雷达连接状态
|
*/
|
@Scheduled(cron = "0/3 * * * * ?")
|
public void monitorConnectStatus() {
|
try {
|
log.debug("实时监测雷达连接状态...");
|
List<ArdEquipRadar> ardEquipRadars = ardEquipRadarService.selectArdEquipRadarList(new ArdEquipRadar());
|
ardEquipRadars.stream().forEach(ardEquipRadar -> {
|
//判断与雷达客户端连接状态
|
boolean online = SERVER_MAP.containsKey(ardEquipRadar.getIp() + ":" + ardEquipRadar.getPort());
|
if (online) {
|
//连接雷达客户端成功 判断周扫状态
|
if (aroundScanfMap.containsKey(ardEquipRadar.getIp() + ":" + ardEquipRadar.getPort())) {
|
Integer state = aroundScanfMap.get(ardEquipRadar.getIp() + ":" + ardEquipRadar.getPort());
|
if (state == 1) {//1-周扫打开-连接成功
|
if (ardEquipRadar.getState() == null || !ardEquipRadar.getState().equals("1")) {
|
ardEquipRadar.setState("1");
|
ardEquipRadarService.updateArdEquipRadar(ardEquipRadar);
|
}
|
} else {
|
if (ardEquipRadar.getState() == null || !ardEquipRadar.getState().equals("2")) {
|
//2-周扫未开
|
ardEquipRadar.setState("2");
|
ardEquipRadarService.updateArdEquipRadar(ardEquipRadar);
|
}
|
|
}
|
} else {
|
if (ardEquipRadar.getState() == null || !ardEquipRadar.getState().equals("2")) {
|
//2-周扫未开
|
ardEquipRadar.setState("2");
|
ardEquipRadarService.updateArdEquipRadar(ardEquipRadar);
|
}
|
}
|
} else {
|
//连接雷达客户端失败
|
if (ardEquipRadar.getState() == null || !ardEquipRadar.getState().equals("0")) {
|
//0-客户端不通
|
ardEquipRadar.setState("0");
|
ardEquipRadarService.updateArdEquipRadar(ardEquipRadar);
|
}
|
}
|
});
|
} catch (Exception e) {
|
log.error("定时监测雷达连接状态异常:" + e.getMessage());
|
}
|
}
|
}
|
