Java tutorial
/** * Copyright 2013-2023 Xia Jun(3979434@qq.com). * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. * *************************************************************************************** * * * Website : http://www.farsunset.com * * * *************************************************************************************** */ package com.farsunset.cim.sdk.client; import java.net.InetSocketAddress; import java.util.Random; import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; import java.util.concurrent.Semaphore; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import com.farsunset.cim.sdk.client.constant.CIMConstant; import com.farsunset.cim.sdk.client.exception.SessionDisconnectedException; import com.farsunset.cim.sdk.client.filter.ClientMessageDecoder; import com.farsunset.cim.sdk.client.filter.ClientMessageEncoder; import com.farsunset.cim.sdk.client.model.HeartbeatRequest; import com.farsunset.cim.sdk.client.model.HeartbeatResponse; import com.farsunset.cim.sdk.client.model.Intent; import com.farsunset.cim.sdk.client.model.Message; import com.farsunset.cim.sdk.client.model.ReplyBody; import com.farsunset.cim.sdk.client.model.SentBody; import io.netty.bootstrap.Bootstrap; import io.netty.channel.Channel; import io.netty.channel.ChannelFuture; import io.netty.channel.ChannelHandler.Sharable; import io.netty.channel.ChannelHandlerContext; import io.netty.channel.ChannelInitializer; import io.netty.channel.ChannelOption; import io.netty.channel.ChannelPipeline; import io.netty.channel.EventLoopGroup; import io.netty.channel.SimpleChannelInboundHandler; import io.netty.channel.nio.NioEventLoopGroup; import io.netty.channel.socket.SocketChannel; import io.netty.channel.socket.nio.NioSocketChannel; import io.netty.handler.timeout.IdleState; import io.netty.handler.timeout.IdleStateEvent; import io.netty.handler.timeout.IdleStateHandler; import io.netty.util.AttributeKey; import io.netty.util.concurrent.GenericFutureListener; /** * ??cim????? * * @author 3979434@qq.com */ @Sharable class CIMConnectorManager extends SimpleChannelInboundHandler<Object> { protected final Logger logger = LoggerFactory.getLogger(CIMConnectorManager.class.getSimpleName()); private final int CONNECT_TIMEOUT = 10 * 1000;// private final int WRITE_TIMEOUT = 10 * 1000;// private final int READ_IDLE_TIME = 120;// private final int HEARBEAT_TIME_OUT = (READ_IDLE_TIME + 20) * 1000;// ? private final String KEY_LAST_HEART_TIME = "KEY_LAST_HEART_TIME"; private Bootstrap bootstrap; private EventLoopGroup loopGroup; private Channel channel;; private ExecutorService executor = Executors.newCachedThreadPool(); private Semaphore semaphore = new Semaphore(1, true); private static CIMConnectorManager manager; private CIMConnectorManager() { bootstrap = new Bootstrap(); loopGroup = new NioEventLoopGroup(); bootstrap.group(loopGroup); bootstrap.channel(NioSocketChannel.class); bootstrap.option(ChannelOption.TCP_NODELAY, true); bootstrap.option(ChannelOption.CONNECT_TIMEOUT_MILLIS, CONNECT_TIMEOUT); bootstrap.handler(new ChannelInitializer<SocketChannel>() { @Override public void initChannel(SocketChannel ch) throws Exception { ChannelPipeline pipeline = ch.pipeline(); pipeline.addLast(new ClientMessageDecoder()); pipeline.addLast(new ClientMessageEncoder()); pipeline.addLast(new IdleStateHandler(READ_IDLE_TIME, 0, 0)); pipeline.addLast(CIMConnectorManager.this); } }); } public synchronized static CIMConnectorManager getManager() { if (manager == null) { manager = new CIMConnectorManager(); } return manager; } public void connect(final String host, final int port) { if (isConnected() || !semaphore.tryAcquire()) { return; } executor.execute(new Runnable() { @Override public void run() { logger.info("****************CIM? " + host + ":" + port + "......"); final InetSocketAddress remoteAddress = new InetSocketAddress(host, port); bootstrap.connect(remoteAddress).addListener(new GenericFutureListener<ChannelFuture>() { @Override public void operationComplete(ChannelFuture future) throws Exception { semaphore.release(); future.removeListener(this); if (!future.isSuccess() && future.cause() != null) { handleConnectFailure(future.cause(), remoteAddress); } if (future.isSuccess()) { channel = future.channel(); } } }); } }); } private void handleConnectFailure(Throwable error, InetSocketAddress remoteAddress) { long interval = CIMConstant.RECONN_INTERVAL_TIME - (5 * 1000 - new Random().nextInt(15 * 1000)); Intent intent = new Intent(); intent.setAction(CIMConstant.IntentAction.ACTION_CONNECTION_FAILED); intent.putExtra(Exception.class.getName(), error); intent.putExtra("interval", interval); sendBroadcast(intent); logger.warn("****************CIM? " + remoteAddress.getHostName() + ":" + remoteAddress.getPort() + "......" + interval / 1000 + "???"); } public void send(SentBody body) { boolean isSuccessed = false; Throwable exception = new SessionDisconnectedException(); if (channel != null && channel.isActive()) { isSuccessed = channel.writeAndFlush(body).awaitUninterruptibly(WRITE_TIMEOUT); } if (!isSuccessed) { Intent intent = new Intent(); intent.setAction(CIMConstant.IntentAction.ACTION_SENT_FAILED); intent.putExtra(Exception.class.getName(), exception); intent.putExtra(SentBody.class.getName(), body); sendBroadcast(intent); } else { Intent intent = new Intent(); intent.setAction(CIMConstant.IntentAction.ACTION_SENT_SUCCESSED); intent.putExtra(SentBody.class.getName(), (SentBody) body); sendBroadcast(intent); } } public void destroy() { if (channel != null) { channel.close(); } if (loopGroup != null) { loopGroup.shutdownGracefully(); } manager = null; } public boolean isConnected() { return channel != null && channel.isActive(); } public void closeSession() { if (channel != null) { channel.close(); } } @Override public void channelActive(ChannelHandlerContext ctx) throws Exception { logger.info("****************CIM??:" + ctx.channel().localAddress() + " NID:" + ctx.channel().id().asShortText()); setLastHeartbeatTime(ctx.channel()); Intent intent = new Intent(); intent.setAction(CIMConstant.IntentAction.ACTION_CONNECTION_SUCCESSED); sendBroadcast(intent); } @Override public void channelInactive(ChannelHandlerContext ctx) { logger.error("****************CIM?:" + ctx.channel().localAddress() + " NID:" + ctx.channel().id().asShortText()); Intent intent = new Intent(); intent.setAction(CIMConstant.IntentAction.ACTION_CONNECTION_CLOSED); sendBroadcast(intent); } @Override public void userEventTriggered(ChannelHandlerContext ctx, Object evt) { /** * wifi??? ??? * */ if (evt instanceof IdleStateEvent && ((IdleStateEvent) evt).state().equals(IdleState.READER_IDLE)) { logger.debug("****************CIM " + IdleState.READER_IDLE + ":" + ctx.channel().localAddress() + " NID:" + ctx.channel().id().asShortText()); Long lastTime = getLastHeartbeatTime(ctx.channel()); if (lastTime != null && System.currentTimeMillis() - lastTime > HEARBEAT_TIME_OUT) { channel.close(); logger.error("****************CIM ,??......" + " NID:" + ctx.channel().id().asShortText()); } } } @Override public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) { logger.error("****************CIM:" + ctx.channel().localAddress() + " NID:" + ctx.channel().id().asShortText()); if (cause != null && cause.getMessage() != null) { logger.error(cause.getMessage()); } Intent intent = new Intent(); intent.setAction(CIMConstant.IntentAction.ACTION_UNCAUGHT_EXCEPTION); intent.putExtra(Exception.class.getName(), cause); sendBroadcast(intent); } @Override public void channelRead0(ChannelHandlerContext ctx, Object msg) throws Exception { if (msg instanceof Message) { Intent intent = new Intent(); intent.setAction(CIMConstant.IntentAction.ACTION_MESSAGE_RECEIVED); intent.putExtra(Message.class.getName(), (Message) msg); sendBroadcast(intent); } if (msg instanceof ReplyBody) { Intent intent = new Intent(); intent.setAction(CIMConstant.IntentAction.ACTION_REPLY_RECEIVED); intent.putExtra(ReplyBody.class.getName(), (ReplyBody) msg); sendBroadcast(intent); } // ???? if (msg instanceof HeartbeatRequest) { ctx.writeAndFlush(HeartbeatResponse.getInstance()); setLastHeartbeatTime(ctx.channel()); } } private void setLastHeartbeatTime(Channel channel) { channel.attr(AttributeKey.valueOf(KEY_LAST_HEART_TIME)).set(System.currentTimeMillis()); } private Long getLastHeartbeatTime(Channel channel) { return (Long) channel.attr(AttributeKey.valueOf(KEY_LAST_HEART_TIME)).get(); } private void sendBroadcast(final Intent intent) { executor.execute(new Runnable() { @Override public void run() { CIMEventBroadcastReceiver.getInstance().onReceive(intent); } }); } }