package org.yzh.client.netty;
|
|
import io.github.yezhihao.netmc.core.model.Message;
|
import io.netty.bootstrap.Bootstrap;
|
import io.netty.buffer.ByteBuf;
|
import io.netty.buffer.ByteBufUtil;
|
import io.netty.buffer.Unpooled;
|
import io.netty.channel.*;
|
import io.netty.channel.nio.NioEventLoopGroup;
|
import io.netty.channel.socket.nio.NioChannelOption;
|
import io.netty.channel.socket.nio.NioSocketChannel;
|
import io.netty.handler.codec.ByteToMessageDecoder;
|
import io.netty.handler.codec.DelimiterBasedFrameDecoder;
|
import io.netty.handler.codec.MessageToByteEncoder;
|
import lombok.extern.slf4j.Slf4j;
|
|
import java.util.List;
|
|
/**
|
* @author yezhihao
|
* https://gitee.com/yezhihao/jt808-server
|
*/
|
@Slf4j
|
public class TCPClient {
|
|
private EventLoopGroup workerGroup;
|
|
private Channel channel;
|
|
private ClientConfig config;
|
private String id;
|
|
public TCPClient(String id, ClientConfig config) {
|
this.id = id;
|
this.config = config;
|
}
|
|
private void startInternal() {
|
try {
|
this.workerGroup = new NioEventLoopGroup();
|
Bootstrap bootstrap = new Bootstrap();
|
bootstrap.channel(NioSocketChannel.class);
|
bootstrap.group(workerGroup);
|
bootstrap.option(NioChannelOption.SO_REUSEADDR, true)
|
.option(NioChannelOption.TCP_NODELAY, true)
|
.option(NioChannelOption.SO_KEEPALIVE, true)
|
.handler(new ChannelInitializer<NioSocketChannel>() {
|
@Override
|
public void initChannel(NioSocketChannel ch) {
|
ch.pipeline()
|
.addLast("frameDecoder", new DelimiterBasedFrameDecoder(config.maxFrameLength,
|
Unpooled.wrappedBuffer(config.delimiter),
|
Unpooled.wrappedBuffer(config.delimiter, config.delimiter)))
|
.addLast("decoder", new ByteToMessageDecoder() {
|
@Override
|
protected void decode(ChannelHandlerContext ctx, ByteBuf buf, List<Object> out) {
|
log.info("<<<<<{}", ByteBufUtil.hexDump(buf));
|
Object msg = config.decoder.decode(buf, null);
|
log.info("<<<<<<<<<<{}", msg);
|
if (msg != null)
|
out.add(msg);
|
buf.skipBytes(buf.readableBytes());
|
}
|
})
|
.addLast("encoder", new MessageToByteEncoder<Message>() {
|
@Override
|
protected void encode(ChannelHandlerContext ctx, Message msg, ByteBuf out) {
|
log.info(">>>>>>>>>>{}", msg);
|
ByteBuf buf = config.encoder.encode(msg, null);
|
log.info(">>>>>{}", ByteBufUtil.hexDump(buf));
|
out.writeBytes(config.delimiter).writeBytes(buf).writeBytes(config.delimiter);
|
buf.release();
|
}
|
})
|
.addLast("adapter", config.adapter);
|
}
|
});
|
|
ChannelFuture channelFuture = bootstrap.connect(config.ip, config.port).sync();
|
this.channel = channelFuture.channel();
|
this.channel.closeFuture();
|
} catch (Exception e) {
|
log.error("===TCP Client异常关闭", e);
|
}
|
}
|
|
public void writeObject(Object message) {
|
channel.writeAndFlush(message);
|
}
|
|
public synchronized TCPClient start() {
|
startInternal();
|
log.warn("===TCP Client启动成功, id={}===", id);
|
return this;
|
}
|
|
public synchronized void stop() {
|
workerGroup.shutdownGracefully();
|
log.warn("===TCP Client已经停止, id={}===", id);
|
}
|
}
|