首页 > 代码库 > 项目积累——Mina
项目积累——Mina
继续上一篇,这篇主要讲通过mina往B端发送消息,并接受消息,mina是一个网络通信框架,封装了javaNIO,简单易用,网上有很多关于他的介绍,在此不赘述了。
如上篇所介绍,完成功能,需要五个类:
PoolListener:监听,用来在系统启动的时候创建连接;
SessionPool:连接池;
SendHandler:处理类;
CharsetEncoder:编码;
CharsetDecoder:解码:
B为我们提供了6个端口,每个端口可建立3个长连接,因此,在系统时,就要创建长连接,下面是一个监听类:
import javax.servlet.ServletContextEvent; import javax.servlet.ServletContextListener; /** * 初始化连接 * @author yuanfubiao * */ public class PoolListener implements ServletContextListener { @Override public void contextDestroyed(ServletContextEvent sce) { } @Override public void contextInitialized(ServletContextEvent sce) { String nds_ip = sce.getServletContext().getInitParameter("nds_ip"); String nds_ports = sce.getServletContext().getInitParameter("nds_ports"); SessionPool pool = new SessionPool(); try { pool.init(nds_ip, nds_ports); } catch (Exception e) { e.printStackTrace(); } } }
下面是监听配置,是配置在web.xml中:
<display-name>Apache-Axis2</display-name> <context-param> <param-name>nds_ip</param-name> <param-value>XX.XXX.XXX.XXX</param-value> </context-param> <context-param> <param-name>nds_ports</param-name> <param-value>12210,12211,12212,12213,12214,12215</param-value> </context-param> <listener> <listener-class>cn.net.easyway.nds.PoolListener</listener-class> </listener>
下面是自己维护的一个连接池,同样使用并发包中的ConcurrentHashMap实现,他也是线程安全的,代码如下:
import java.net.InetSocketAddress; import java.util.HashMap; import java.util.Map; import java.util.concurrent.ConcurrentHashMap; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.apache.mina.core.future.ConnectFuture; import org.apache.mina.core.service.IoConnector; import org.apache.mina.core.session.IoSession; import org.apache.mina.filter.codec.ProtocolCodecFilter; import org.apache.mina.transport.socket.nio.NioSocketConnector; public class SessionPool { private static Log logger = LogFactory.getLog(SessionPool.class); private static int connNum = 0; private static String ip = null; private static Map<String,Integer> connNumPorts = new HashMap<String, Integer>(); private static ConcurrentHashMap<String, IoSession> pool = new ConcurrentHashMap<String, IoSession>(); /** * 初始化:读取配置文件,创建长连接 * @throws Exception */ public void init(String nds_ip,String nds_ports) throws Exception{ String[] ports = nds_ports.split(","); ip = nds_ip; for(int i=0;i<ports.length;i++){ int port = Integer.parseInt(ports[i]); ConnectFuture future = null; for(int j=0;j<3;j++){ String connNum = this.getConnNums(); logger.info("创建连接号---->>>>>" + connNum); connNumPorts.put(connNum, port); future = SessionPool.createConnect(ip, port); if(future.isConnected()){ logger.info("创建连接------->" + future.getSession()); pool.put(connNum, future.getSession()); }else{ logger.error("连接创建错误,请检查IP和端口配置!" + future); } } } } /** * 获取一个连接 * @param num * @return */ public static IoSession getSession(String strNum){ logger.info("IP端口号:" + ip + "连接序列号:" + strNum + "端口号:" + connNumPorts.get(strNum)); IoSession session = pool.get(strNum); if(null == session || !session.isClosing()){ ConnectFuture newConn = createConnect(ip, connNumPorts.get(strNum)); if(!newConn.isConnected()){ newConn = createConnect(ip,connNumPorts.get(strNum)); } session = newConn.getSession(); pool.replace(strNum, session); } return session; } /** * 创建连接 * @param ip * @param port * @return */ private static ConnectFuture createConnect(String strIp,int intPort){ IoConnector connector = new NioSocketConnector(); connector.getFilterChain().addLast("codec" ,new ProtocolCodecFilter(new CharsetCodecFactory())); connector.setHandler(new SendHandler()); ConnectFuture future = connector.connect(new InetSocketAddress(strIp,intPort)); connector.getSessionConfig().setReadBufferSize(128); future.awaitUninterruptibly(); return future; } /** * 生成连接序列号 * @return */ private synchronized String getConnNums(){ if(18 == connNum){ connNum = 0; } connNum++; return String.format("%02x", connNum); } }
因此,在项目启动的时候就会有18个连接自动创建,并放在pool中等待我们的使用。下面是业务处理类,需要继承IoHandlerAdapter类,并且实现以下几个方法:
import nds.framework.security.NDSMD5; import org.apache.commons.codec.binary.Hex; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.apache.mina.core.service.IoHandlerAdapter; import org.apache.mina.core.session.IdleStatus; import org.apache.mina.core.session.IoSession; import cm.custom.service.reception.RecResponse; import cm.custom.service.reception.ReceptionResponseServiceStub; /** * 业务处理 * @author yuanfubiao * */ public class SendHandler extends IoHandlerAdapter { private static Log logger = LogFactory.getLog(SendHandler.class); @Override public void exceptionCaught(IoSession session, Throwable cause) throws Exception { logger.error("连接出错", cause); } @Override /** * 设置空闲时间 */ public void sessionCreated(IoSession session) throws Exception { session.getConfig().setIdleTime(IdleStatus.BOTH_IDLE, 60); } /** * 接受到消息后,通过WS发送给用户管理系统 */ @Override public void messageReceived(IoSession session, Object message) throws Exception { String result = message.toString().trim(); String temp = result.substring(0, result.length()-16).trim(); logger.info("接受到的数据:" + result); //验证签名 String signature = null; String securityKey = "12345678"; try { byte binSignature[] = NDSMD5.signPacket(temp.getBytes(), securityKey); signature = new String(Hex.encodeHex(binSignature)); } catch (Exception e) { e.printStackTrace(); } String packet = temp + signature.toUpperCase().trim(); if(!result.equalsIgnoreCase(packet)){ logger.error("数字签名不正确!错误指令:" + result); return; } logger.info("接受到的数据:" + packet); RecResponse res = new RecResponse(); res.setResponse(temp); ReceptionResponseServiceStub stub = new ReceptionResponseServiceStub(); stub.recResponse(res); } /** * 连接空闲时,发送心跳包 */ @Override public void sessionIdle(IoSession session, IdleStatus status) throws Exception { if(status == IdleStatus.BOTH_IDLE){ session.write("heartbeat"); } } }
一般我们在写socket程序时,用阻塞的方式读取消息,一般是根据消息换行符或者特殊字符,或者对方关闭流来证明一条信息读取完成,在mina中,有默认的编解码方式,但也可以自定义,比如以长度来判断一条消息是否读取完毕:
编码
import java.nio.charset.Charset; import org.apache.mina.core.buffer.IoBuffer; import org.apache.mina.core.session.IoSession; import org.apache.mina.filter.codec.ProtocolEncoderAdapter; import org.apache.mina.filter.codec.ProtocolEncoderOutput; /** * 编码 * @author yuanfubiao * */ public class CharsetEncoder extends ProtocolEncoderAdapter{ private final static Charset charset = Charset.forName("utf-8"); @Override public void encode(IoSession session, Object message, ProtocolEncoderOutput out) throws Exception { IoBuffer buff = IoBuffer.allocate(100).setAutoExpand(true); buff.putString(message.toString(), charset.newEncoder()); buff.flip(); out.write(buff); } }
解码
import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.apache.mina.core.buffer.IoBuffer; import org.apache.mina.core.session.IoSession; import org.apache.mina.filter.codec.CumulativeProtocolDecoder; import org.apache.mina.filter.codec.ProtocolDecoderOutput; /** * 解码 * @author yuanfubiao * */ public class CharsetDecoder extends CumulativeProtocolDecoder{ private static Log logger = LogFactory.getLog(CharsetDecoder.class); @Override protected boolean doDecode(IoSession session, IoBuffer in, ProtocolDecoderOutput out) throws Exception { if(in.remaining() >= 9){ //心跳为最小传输长度 byte[] headBytes = new byte[in.limit()]; logger.info("接收到消息" + headBytes.toString()); in.get(headBytes, 0, 9); String head = new String(headBytes).trim(); if("heartbeat".equalsIgnoreCase(head)){ return true; } int lenPack = Integer.parseInt(head.substring(5, 9), 16)-9; if(in.remaining() == lenPack){ //验证消息长度 byte[] bodyBytes = new byte[in.limit()]; in.get(bodyBytes,0,lenPack); String body = new String(bodyBytes); out.write(head.trim()+body.trim()); return true; } in.flip(); return false; } return false; } }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。