Mina与Spring整合

标签：

首先需要导入的包

applicationContext-mina.xml

1. <?xml version="1.0" encoding="UTF-8"?>
2. <beans xmlns="http://www.springframework.org/schema/beans"
3.     xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
4.     xsi:schemaLocation="http://www.springframework.org/schema/beans
5.   http://www.springframework.org/schema/beans/spring-beans-2.5.xsd">
6.     <!-- 配置地址 -->
7.     <bean class="com.sharp.slc.net.common.EncryptPropertyPlaceholderConfigurer">  
8.       <property name="order" value="1"/>  
9.       <property name="ignoreUnresolvablePlaceholders" value="true"/>  
10.       <property name="locations">  
11.         <list>  
12.             <value>classpath:ssl.properties</value>  
13.         </list>  
14.       </property>  
15.     </bean>
16.     <bean class="org.springframework.beans.factory.config.CustomEditorConfigurer">
17.         <property name="customEditors">
18.             <map>
19.                 <!-- spring升级后此配置已失效 会报错 -->
20.                 <!-- <entry key="java.net.SocketAddress"> -->
21.                 <!-- <bean class="org.apache.mina.integration.beans.InetSocketAddressEditor"
22.                     /> -->
23.                 <!-- </entry> -->
24.                 <!-- 修改这里 -->
25.                 <entry key="java.net.SocketAddress"
26.                     value="org.apache.mina.integration.beans.InetSocketAddressEditor">
27.                 </entry>
28.             </map>
29.         </property>
30.     </bean>
31.     <!-- 配置业务处理类 -->
32.     <bean id="serviceHandler" class="com.sharp.slc.net.core.TCPServerHandler" />
33.     <!-- 配置service -->
34.     <bean id="ioAcceptor" class="org.apache.mina.transport.socket.nio.NioSocketAcceptor"
35.         init-method="bind" destroy-method="unbind">
36.         <property name="defaultLocalAddress" value=":2001" />
37.         <property name="handler" ref="serviceHandler" />
38.         <property name="filterChainBuilder" ref="filterChainBuilder" />
39.     </bean>
41.     <!-- 配置解码器 -->
42.     <bean id="codecFilter" class="org.apache.mina.filter.codec.ProtocolCodecFilter">
43.         <constructor-arg>
44.             <bean class="com.sharp.slc.net.core.codec.TextCodecFactory" />
45.             
46. <!--     <bean class="org.apache.mina.filter.codec.textline.TextLineCodecFactory" /> -->
47.         </constructor-arg>
48.     </bean>
49.     <!-- 配置日志拦截器 -->
50.     <bean id="loggingFilter" class="org.apache.mina.filter.logging.LoggingFilter" />
51.     <!-- 配置SSL拦截器 -->
52.     <bean id="sslFilter" class="org.apache.mina.filter.ssl.SslFilter">
53.         <constructor-arg ref="sslContext" />
54. <!--         <property name="needClientAuth" value="true" /> -->
55.     </bean>
56.     <!-- The SSL configuration -->
57.     <bean id="keystoreFactory" class="org.apache.mina.filter.ssl.KeyStoreFactory">
58.         <property name="password">
59.             <value>${key_password}</value>
60.         </property>
61.     <!--     <property name="dataUrl" value="file:/d:/certs/server.jks" />  -->
62.         <property name="dataUrl" value="file:/usr/local/tomcat8/certs/server.jks" />
63.     </bean>
64.     <bean id="keyStore" factory-bean="keystoreFactory" factory-method="newInstance" />
65.     <!-- SSLContext to be used -->
66.     <bean id="sslContextFactory" class="org.apache.mina.filter.ssl.SslContextFactory">
67.         <property name="protocol" value="TLSV1.2" />
68.         <property name="keyManagerFactoryAlgorithm" value="SunX509" />
69.         <property name="keyManagerFactoryKeyStore">
70.             <ref local="keyStore" />
71.         </property>
72.         <property name="keyManagerFactoryKeyStorePassword">
73.             <value>${keyManager_password}</value>
74.         </property>
75.     </bean>
76.     <bean id="sslContext" factory-bean="sslContextFactory"
77.         factory-method="newInstance" />
79.     <!-- 将日志和解码器添加 -->
80.     <bean id="filterChainBuilder"
81.         class="org.apache.mina.core.filterchain.DefaultIoFilterChainBuilder">
82.         <property name="filters">
83.             <map>
84.                 <entry key="sslFilter" value-ref="sslFilter" />
85.                 <entry key="loggingFilter" value-ref="loggingFilter" />
86.                 <entry key="codecFilter" value-ref="codecFilter" />
87.             </map>
88.         </property>
89.     </bean>
90. </beans>

SessionCache.java

1. public class SessionCache {
2.     private final static Log log = LogFactory.getLog(SessionCache.class);
3.     private static SessionCache session = null;
4.     /** 设备连接信息[sessions] */
5.     private Map<String, IoSession> sessions = new HashMap<String, IoSession>();
7.     private static final String value = KVStoreConfig.getIpAddress();
9.     /**
10.      * 获取唯一实例
11.      */
12.     public static SessionCache getInstance() {
13.         log.debug(" <<< Session单例获取 <<< ");
14.         if (session == null) {
15.             session = new SessionCache();
16.         }
17.         return session;
18.     }
20.     /**
21.      * 获取设备连接信息[sessions]
22.      *
23.      * @return 设备连接信息[sessions]
24.      */
25.     public Map<String, IoSession> getSessions() {
26.         return sessions;
27.     }
29.     /**
30.      * 设置设备连接信息[sessions]
31.      *
32.      * @param 设备连接信息
33.      *            [sessions]
34.      */
35.     public void setSessions(Map<String, IoSession> sessions) {
36.         this.sessions = sessions;
37.     }
39.     /**
40.      * 增加设备连接信息[sessions]
41.      *
42.      * @param MAC地址作为键
43.      *            [mac]
44.      * @param IoSession对象作为值
45.      *            [session]
46.      */
47.     public void save(String mac, IoSession session) {
48.         sessions.put(mac, session);
49.         String clientIp = session.getRemoteAddress().toString();
50.         KVStoreUtils.putHashmapField(mac, value);
51.         String kvIp = KVStoreUtils.getHashmapField(NetConstants.KV_PORT_KEY, mac);
52.         if (StringUtil.isNotBlank(kvIp)) {
53.             if (!kvIp.contains(clientIp)) {
54.                 String ipvalue = kvIp + ";" + clientIp;
55.                 KVStoreUtils.putHashmapField(NetConstants.KV_PORT_KEY, mac, ipvalue);
56.             }
57.         } else {
58.             KVStoreUtils.putHashmapField(NetConstants.KV_PORT_KEY, mac, clientIp);
59.         }
60.     }
62.     /**
63.      * 查找设备连接信息[sessions]
64.      *
65.      * @param MAC地址作为键
66.      *            [key]
67.      * @return 连接信息
68.      */
69.     public IoSession isExists(String key) {
70.         if (sessions.containsKey(key)) {
71.             return sessions.get(key);
72.         }
73.         return null;
74.     }
76.     /**
77.      * 删除设备连接信息[sessions]
78.      *
79.      * @param MAC地址作为键
80.      *            [mac]
81.      */
82.     public void remove(String mac, String clientIp) {
83.         sessions.remove(mac);
84.         String ipvalue = KVStoreUtils.getHashmapField(NetConstants.KV_PORT_KEY, mac);
85.         if (clientIp.equals(ipvalue)) {
86.             KVStoreUtils.removeHahmapField(mac);
87.             KVStoreUtils.removeHahmapField(NetConstants.KV_PORT_KEY, mac);
88.         } else {
89.             String ipvalueTemp = ipvalue.replaceAll(clientIp + ";", "");
90.             KVStoreUtils.putHashmapField(NetConstants.KV_PORT_KEY, mac, ipvalueTemp);
91.         }
92.     }
93. }

TCPServerHandler.java

1. /**
2.  * 服务端业务处理类
3.  *
4.  * @author Pactera.He
5.  */
6. public class PurifierTCPServerHandler extends IoHandlerAdapter {
7.     private final static Log log = LogFactory.getLog(TCPServerHandler.class);
8.     private final static Map<String, String> firstHeartBeatMap = new HashMap<String, String>();
9.     public TCPServerHandler() {
10.     }
11.     @Override
12.     public void exceptionCaught(IoSession session, Throwable cause) throws Exception {
13.         log.error(session.getRemoteAddress().toString() + " : " + cause.toString());
14.     }
15.     @Override
16.     public void messageReceived(IoSession session, Object message) throws Exception {
17.         // 消息内容
18.         String text = message.toString();
19.         // 获取客户端发过来的消息内容
20.         log.info("接收消息内容 : " + text);
21.         if (StringUtil.isNotBlank(text)) {
22.             // 解析XML文件
23.             readXML(text, session);
24.         } else {
25.             // 返回错误信息
26.             session.write("error heartbeat");
27.         }
28.     }
29.     @Override
30.     public void messageSent(IoSession session, Object message) throws Exception {
31.         log.debug("发送消息内容 : \n" + message.toString());
32.     }
33.     @Override
34.     public void sessionCreated(IoSession session) throws Exception {
35.         log.debug("连接创建 : " + session.getRemoteAddress().toString());
36.     }
37.     @Override
38.     public void sessionIdle(IoSession session, IdleStatus status) throws Exception {
39.         log.info(MessageFormat.format("连接Idle [{0}] from {1} ", status, session.getRemoteAddress()));
40.         if (status == IdleStatus.READER_IDLE) {
41.             session.close(true);
42.         }
43.     }
44.     @Override
45.     public void sessionOpened(IoSession session) throws Exception {
46.         if (checkIsBlocked(session)) {
47.             session.close(true);
48.             return;
49.         }
50.         log.info("连接打开：" + session.getRemoteAddress().toString());
51.     }
52.     @Override
53.     public void sessionClosed(IoSession session) throws Exception {
54.         if (checkIsBlocked(session)) {
55.             return;
56.         }
57.         log.info("连接关闭 : " + session.getRemoteAddress().toString());
58.         // 关闭通知
59.         if (session.getAttribute("mac") != null) {
60.             String mac = session.getAttribute("mac").toString();
61.             firstHeartBeatMap.remove(mac);
62.             this.notifyWeChat(mac, session.getRemoteAddress().toString());
63.         }
64.     }
65.     /**
66.      * session关闭通知WeChat
67.      */
68.     private void notifyWeChat(String mac, String clientIp) {
69.         log.info("连接关闭：" + mac + ",通知微信 . . .");
70.         if (StringUtil.isNotBlank(mac)) {
71.             // 清除信息
72.             SessionCache.getInstance().remove(mac, clientIp);
73.         }
74.     }
75.     /**
76.      * 解析XML文件
77.      *
78.      * @param contents
79.      *            解析数据内容
80.      * @param session
81.      *            处理对象
82.      */
83.     private void readXML(String contents, IoSession session) throws DocumentException {
84.         Document readDoc = DocumentHelper.parseText(contents);
85.         Element rootNode = readDoc.getRootElement();
86.         String mac = "";
87.         // 匹配节点：tcp_msg
88.         if (NetConstants.NODE_TCP_MSG.equals(rootNode.getName())) {
89.             // 匹配节点（msg）内容
90.             if (NetConstants.MSG_VALUE_HEARTBEAT.equals(rootNode.elementTextTrim(NetConstants.NODE_MSG))) {
91.                 // mac address
92.                 Element node = rootNode.element(NetConstants.NODE_DATA);
93.                 mac = node.elementTextTrim(NetConstants.NODE_MAC_ADDRESS);
94.                 // 生成所需XML文件
95.                 createXML(mac, session, wifiVersion);
96.             }
97.         }
98.     }
99.     /**
100.      * 生成XML文件
101.      *
102.      * @param mac
103.      *            MAC地址
104.      * @param session
105.      *            处理对象
106.      */
107.     private void createXML(String mac, IoSession session, String wifiVersion) {
108.         log.info("当前处理Mac地址 : " + mac);
109.         if (StringUtil.isNotBlank(mac)) {
110.             // 保存客户端的会话session
111.             SessionCache.getInstance().save(mac, session);
112.             // DocumentHelper提供了创建Document对象的方法
113.             Document writeDoc = DocumentHelper.createDocument();
114.             // 添加节点：tcp_msg
115.             Element root = writeDoc.addElement(NetConstants.NODE_TCP_MSG);
116.             // 添加节点：msg
117.             Element msg = root.addElement(NetConstants.NODE_MSG);
118.             // 设置节点信息
119.             msg.setText(NetConstants.MSG_VALUE_HEARTBEATRES);
120.             // 添加节点:cmd
121.             Element cmd = root.addElement(NetConstants.NODE_CMD);
122.             // 不再检查每个心跳包
123.             cmd.setText(NetConstants.VALUE_NULL);
124.             // 添加节点:data
125.             Element data = root.addElement(NetConstants.NODE_DATA);
126.             // 添加节点:server_time
127.             Element server_time = data.addElement(NetConstants.NODE_SERVER_TIME);
128.             // 设置节点信息
129.             server_time.setText(String.valueOf(System.currentTimeMillis()));
130.             // 将document文档对象直接转换成字符串
131.             log.debug("发送内容: " + writeDoc.asXML());
132.             session.write(writeDoc.asXML());
133.         }
134.     }
135.     /**
136.      * 检查对面地址是否在黑名单中
137.      *
138.      * @param session
139.      *            连接session
140.      * @return 是：true，否：false
141.      */
142.     private boolean checkIsBlocked(IoSession session) {
143.         boolean ret = false;
144.         String[] blockList = {};
145.         for (String blocked : blockList) {
146.             if (session.getRemoteAddress().toString().indexOf(blocked) > 0) {
147.                 ret = true;
148.                 break;
149.             }
150.         }
151.         return ret;
152.     }
153. }

Mina与Spring整合

标签：

原文地址：http://www.cnblogs.com/codeydt/p/5126220.html