package com.ruoyi.dataInterchange.server;
|
|
import com.alibaba.fastjson.JSON;
|
import com.ruoyi.dataInterchange.model.DOWNDisconnectInform;
|
import com.ruoyi.dataInterchange.model.UPConnectReq;
|
import com.ruoyi.dataInterchange.model.UPConnectRsp;
|
import com.ruoyi.dataInterchange.model.enu.DataType;
|
import com.ruoyi.dataInterchange.netty.client.ChannelMap;
|
import com.ruoyi.dataInterchange.netty.client.NettyClient;
|
import com.ruoyi.dataInterchange.util.jtt809.packet.common.OuterPacket;
|
import com.ruoyi.system.api.feignClient.EnterpriseClient;
|
import com.ruoyi.system.api.model.Enterprise;
|
import io.netty.buffer.ByteBuf;
|
import io.netty.buffer.Unpooled;
|
import io.netty.channel.ChannelHandlerContext;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.data.redis.core.RedisTemplate;
|
import org.springframework.stereotype.Component;
|
|
import javax.annotation.Resource;
|
import java.util.Random;
|
import java.util.concurrent.ExecutorService;
|
import java.util.concurrent.LinkedBlockingQueue;
|
import java.util.concurrent.ThreadPoolExecutor;
|
import java.util.concurrent.TimeUnit;
|
|
/**
|
* @author zhibing.pu
|
* @Date 2025/3/4 18:40
|
*/
|
@Slf4j
|
@Component
|
public class ConnectReqService {
|
|
@Resource
|
private EnterpriseClient enterpriseClient;
|
|
@Resource
|
private RedisTemplate redisTemplate;
|
|
|
/**
|
* 主链路登录请求业务逻辑
|
*
|
* @param ctx
|
*/
|
public void connect(ChannelHandlerContext ctx, OuterPacket outerPacket) {
|
ByteBuf byteBuf = Unpooled.wrappedBuffer(outerPacket.getBody());
|
//解析封装原元数据
|
UPConnectReq upConnectReq = new UPConnectReq().decode(byteBuf);
|
log.info("主链路登录请求体：" + JSON.toJSONString(upConnectReq));
|
Enterprise data = enterpriseClient.getEnterprise(upConnectReq.getUserId() + "").getData();
|
UPConnectRsp upConnectRsp = new UPConnectRsp();
|
if (null == data) {
|
//用户没有注册
|
upConnectRsp.setResult(0x03);
|
} else {
|
//IP地址不对
|
if (!upConnectReq.getDownLinkIp().equals(data.getIpWhitelist())) {
|
upConnectRsp.setResult(0x01);
|
} else if (!data.getPassword().equals(upConnectReq.getPassword())) {
|
upConnectRsp.setResult(0x04);
|
} else if (outerPacket.getGnsscenterId() != Integer.valueOf(data.getCode())) {
|
upConnectRsp.setResult(0x02);
|
} else {
|
upConnectRsp.setResult(0x00);
|
}
|
}
|
// 随机一个校验码
|
int verifyCode = new Random().nextInt(9999);
|
upConnectRsp.setVerifyCode(verifyCode);
|
//主链路登录应答
|
log.info("主链路登录应答({})：{}", DataType.UP_CONNECT_RSP.getCode(), JSON.toJSONString(upConnectRsp));
|
byte[] body = upConnectRsp.encode();
|
OuterPacket out = new OuterPacket(DataType.UP_CONNECT_RSP.getCode(),outerPacket.getGnsscenterId() , body);
|
ctx.writeAndFlush(out);
|
ctx.flush();
|
if (upConnectRsp.getResult() == 0x00) {
|
redisTemplate.opsForValue().set("login:" + outerPacket.getGnsscenterId(), System.currentTimeMillis(), 1, TimeUnit.MINUTES);
|
//保存链路
|
ChannelMap.addServerChannel(outerPacket.getGnsscenterId(), ctx.channel());
|
//从链路连接
|
downConnect(ctx, outerPacket.getGnsscenterId(), upConnectReq.getDownLinkIp(), upConnectReq.getDownLinkPort(), verifyCode);
|
} else {
|
ctx.close();
|
}
|
}
|
|
|
/**
|
* 从链路连接
|
*
|
* @param inferiorPlatformId 企业唯一标识
|
* @param host 连接IP
|
* @param port 连接端口
|
*/
|
public void downConnect(ChannelHandlerContext ctx, int inferiorPlatformId, String host, int port, int verifyCode) {
|
try {
|
boolean b = downConnect(inferiorPlatformId, host, port, verifyCode);
|
} catch (Exception e) {
|
downDisconnectInform(ctx, inferiorPlatformId, 0x00);
|
throw new RuntimeException(e);
|
}
|
}
|
|
|
/**
|
* 从链路连接
|
*
|
* @param inferiorPlatformId 企业唯一标识
|
* @param host 连接IP
|
* @param port 连接端口
|
*/
|
public boolean downConnect(int inferiorPlatformId, String host, int port, int verifyCode) {
|
try {
|
ExecutorService executorService = new ThreadPoolExecutor(1, 1,
|
0L, TimeUnit.MILLISECONDS,
|
new LinkedBlockingQueue<Runnable>());
|
executorService.execute(new Runnable() {
|
@Override
|
public void run() {
|
try {
|
new NettyClient(host, port).start(inferiorPlatformId, verifyCode);
|
} catch (Exception e) {
|
throw new RuntimeException(e);
|
}
|
}
|
});
|
return true;
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
return false;
|
}
|
|
|
/**
|
* 从链路断开通知
|
*
|
* @param ctx
|
* @param errorCode
|
*/
|
public void downDisconnectInform(ChannelHandlerContext ctx, int inferiorPlatformId, int errorCode) {
|
DOWNDisconnectInform downDisconnectInform = new DOWNDisconnectInform();
|
downDisconnectInform.setErrorCode(errorCode);
|
log.info("从链路断开通知({})：{}", DataType.DOWN_DISCONNECT_INFORM.getCode(), JSON.toJSONString(downDisconnectInform));
|
byte[] body = downDisconnectInform.encode();
|
OuterPacket out = new OuterPacket(DataType.DOWN_CONNECT_REQ.getCode(), inferiorPlatformId, body);
|
ctx.writeAndFlush(out);
|
ctx.flush();
|
}
|
}
|
