package com.anyun.h264;
|
|
|
import com.anyun.h264.util.BytesUtils;
|
|
import io.netty.bootstrap.Bootstrap;
|
import io.netty.buffer.ByteBuf;
|
import io.netty.buffer.Unpooled;
|
import io.netty.channel.Channel;
|
import io.netty.channel.ChannelFuture;
|
import io.netty.channel.ChannelFutureListener;
|
import io.netty.channel.ChannelHandlerContext;
|
import io.netty.channel.ChannelInboundHandlerAdapter;
|
import io.netty.channel.ChannelInitializer;
|
import io.netty.channel.ChannelOption;
|
import io.netty.channel.EventLoopGroup;
|
import io.netty.channel.nio.NioEventLoopGroup;
|
import io.netty.channel.socket.SocketChannel;
|
import io.netty.channel.socket.nio.NioSocketChannel;
|
import timber.log.Timber;
|
|
import java.util.concurrent.TimeUnit;
|
|
/**
|
* JT/T 1076-2016 TCP客户端工具类
|
* 使用Netty实现TCP连接和RTP包发送
|
*/
|
public class JT1076TcpClient {
|
private static final String TAG = "JT1076TcpClient";
|
|
private String serverIp;
|
private int serverPort;
|
private EventLoopGroup workerGroup;
|
private Channel channel;
|
private boolean isConnected = false;
|
|
// 连接状态回调接口
|
public interface ConnectionListener {
|
void onConnected();
|
void onDisconnected();
|
void onError(Throwable cause);
|
}
|
|
private ConnectionListener connectionListener;
|
|
/**
|
* 设置服务器地址
|
*/
|
public void setServerAddress(String ip, int port) {
|
this.serverIp = ip;
|
this.serverPort = port;
|
}
|
|
/**
|
* 设置连接状态监听器
|
*/
|
public void setConnectionListener(ConnectionListener listener) {
|
this.connectionListener = listener;
|
}
|
|
/**
|
* 初始化TCP连接（异步）
|
*/
|
public void connect() {
|
if (serverIp == null || serverIp.isEmpty()) {
|
Timber.e( "Server IP not set");
|
if (connectionListener != null) {
|
connectionListener.onError(new IllegalArgumentException("Server IP not set"));
|
}
|
return;
|
}
|
|
if (workerGroup != null) {
|
Timber.w("TCP client already initialized, disconnecting first");
|
disconnect();
|
}
|
|
// 创建EventLoopGroup（使用单线程组即可）
|
workerGroup = new NioEventLoopGroup(1);
|
|
try {
|
Bootstrap bootstrap = new Bootstrap();
|
bootstrap.group(workerGroup)
|
.channel(NioSocketChannel.class)
|
.option(ChannelOption.TCP_NODELAY, true) // 禁用Nagle算法，降低延迟
|
.option(ChannelOption.SO_KEEPALIVE, true) // 启用TCP keepalive
|
.option(ChannelOption.CONNECT_TIMEOUT_MILLIS, 5000) // 连接超时5秒
|
.handler(new ChannelInitializer<SocketChannel>() {
|
@Override
|
protected void initChannel(SocketChannel ch) throws Exception {
|
ch.pipeline().addLast(new TcpClientHandler());
|
}
|
});
|
|
Timber.d("Connecting to TCP server: " + serverIp + ":" + serverPort);
|
|
// 异步连接
|
ChannelFuture future = bootstrap.connect(serverIp, serverPort);
|
future.addListener(new ChannelFutureListener() {
|
@Override
|
public void operationComplete(ChannelFuture future) throws Exception {
|
if (future.isSuccess()) {
|
channel = future.channel();
|
isConnected = true;
|
Timber.d("TCP connection established: " + serverIp + ":" + serverPort);
|
if (connectionListener != null) {
|
connectionListener.onConnected();
|
}
|
} else {
|
isConnected = false;
|
Throwable cause = future.cause();
|
Timber.e(cause, "TCP connection failed: " + serverIp + ":" + serverPort);
|
if (connectionListener != null) {
|
connectionListener.onError(cause);
|
}
|
// 连接失败时清理资源
|
shutdown();
|
}
|
}
|
});
|
|
} catch (Exception e) {
|
Timber.e(e, "Initialize TCP client failed");
|
isConnected = false;
|
if (connectionListener != null) {
|
connectionListener.onError(e);
|
}
|
shutdown();
|
}
|
}
|
|
/**
|
* 发送RTP包（TCP方式）
|
*/
|
public void sendPacket(byte[] packet) {
|
if (!isConnected || channel == null || !channel.isActive()) {
|
Timber.w( "TCP channel not connected, packet dropped");
|
return;
|
}
|
|
try {
|
// 将字节数组包装为ByteBuf
|
ByteBuf buffer = Unpooled.wrappedBuffer(packet);
|
String str = BytesUtils.bytesToHexString( BytesUtils.subArray(buffer.array(),0,30));
|
Timber.i( "Send TCP packet:"+ str);
|
// 异步写入
|
ChannelFuture future = channel.writeAndFlush(buffer);
|
|
// 可选：监听发送结果
|
future.addListener(new ChannelFutureListener() {
|
@Override
|
public void operationComplete(ChannelFuture future) throws Exception {
|
if (!future.isSuccess()) {
|
Timber.e(future.cause(), "Send TCP packet failed");
|
}
|
}
|
});
|
|
} catch (Exception e) {
|
Timber.e(e, "Send TCP packet error");
|
}
|
}
|
|
/**
|
* 断开TCP连接
|
*/
|
public void disconnect() {
|
isConnected = false;
|
|
if (channel != null) {
|
try {
|
ChannelFuture future = channel.close();
|
future.await(2, TimeUnit.SECONDS);
|
} catch (InterruptedException e) {
|
Timber.w(e,"Close channel interrupted");
|
Thread.currentThread().interrupt();
|
} catch (Exception e) {
|
Timber.e(e, "Close channel error");
|
}
|
channel = null;
|
}
|
|
shutdown();
|
|
if (connectionListener != null) {
|
connectionListener.onDisconnected();
|
}
|
|
Timber.d("TCP connection closed");
|
}
|
|
/**
|
* 关闭EventLoopGroup（清理资源）
|
*/
|
private void shutdown() {
|
if (workerGroup != null) {
|
try {
|
workerGroup.shutdownGracefully().await(3, TimeUnit.SECONDS);
|
} catch (InterruptedException e) {
|
Timber.w(e, "Shutdown worker group interrupted");
|
Thread.currentThread().interrupt();
|
} catch (Exception e) {
|
Timber.e(e, "Shutdown worker group error");
|
}
|
workerGroup = null;
|
}
|
}
|
|
/**
|
* 检查是否已连接
|
*/
|
public boolean isConnected() {
|
return isConnected && channel != null && channel.isActive();
|
}
|
|
/**
|
* TCP客户端通道处理器
|
*/
|
private class TcpClientHandler extends ChannelInboundHandlerAdapter {
|
|
@Override
|
public void channelActive(ChannelHandlerContext ctx) throws Exception {
|
super.channelActive(ctx);
|
Timber.d("TCP channel active: " + ctx.channel().remoteAddress());
|
isConnected = true;
|
}
|
|
@Override
|
public void channelInactive(ChannelHandlerContext ctx) throws Exception {
|
super.channelInactive(ctx);
|
Timber.d("TCP channel inactive: " + ctx.channel().remoteAddress());
|
isConnected = false;
|
channel = null;
|
if (connectionListener != null) {
|
connectionListener.onDisconnected();
|
}
|
}
|
|
@Override
|
public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
|
Timber.e(cause, "TCP channel exception");
|
isConnected = false;
|
if (connectionListener != null) {
|
connectionListener.onError(cause);
|
}
|
ctx.close();
|
}
|
|
@Override
|
public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
|
// 如果服务器有响应，可以在这里处理
|
// 对于JT/T 1076-2016 RTP发送，通常不需要处理响应
|
ByteBuf buf = (ByteBuf) msg;
|
Timber.d("Received data from server: " + buf.readableBytes() + " bytes");
|
buf.release(); // 释放ByteBuf
|
}
|
}
|
}
|
