liurunyu
2024-05-30 c929083f415ef98765afec91d5a8c88be4450f5b
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
package com.dy.rtuMw.server.forTcp;
 
import com.dy.rtuMw.server.ServerProperties;
import org.apache.mina.core.session.IoSession;
 
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.util.*;
import java.util.Map.Entry;
 
/**
 * 用Hashtable实现,但实测发现Hashtable并不线程安全
 */
public class TcpSessionCacheBk {
    
    /**
     * 用Hashtable而不用HashMap原因:
     * Hashtable线程安全的
     * HashMap线程不安全的
     * 多线程对sessionTable读出或存入,可能产生异常
     * TcpSessionCache是在多线程环境下运行
     */
    private static Hashtable<String, TcpSession> sessionTable = new Hashtable<String, TcpSession>() ;
    
    /**
     * 加入新的IoSession
     * @param rtuAddr
     * @param protocolName
     * @param ioSession
     */
    public static void putNewTcpSession(String rtuAddr, String protocolName, IoSession ioSession){
        TcpSession tcpSe = sessionTable.get(rtuAddr) ;
        if(tcpSe == null){
            tcpSe = new TcpSession() ;
            tcpSe.protocolName = protocolName ;
            tcpSe.ioSession = ioSession ;
            sessionTable.put(rtuAddr, tcpSe) ;
        }else{
            tcpSe.ioSession = ioSession ;
        }
    }
    
    /**
     * 更新IoSession对应的rtuAddr
     * @param oldRtuAddr
     * @param newRtuAddr
     * @param protocolName
     * @param ioSession
     */
    public static void changeRtuAddr(String oldRtuAddr, String newRtuAddr, String protocolName, IoSession ioSession){
        if(oldRtuAddr != null && newRtuAddr != null && !oldRtuAddr.equals(newRtuAddr)){
            TcpSession tcpSe = sessionTable.get(oldRtuAddr) ;
            if(tcpSe == null){
                putNewTcpSession(newRtuAddr, protocolName, ioSession) ;
            }else{
                sessionTable.remove(oldRtuAddr) ;
                sessionTable.put(newRtuAddr, tcpSe) ;
            }
        }
    }
    
    /**
     * 得到TcpSession
     * @param rtuAddr
     * @return
     */
    public static TcpSession getTcpSession(String rtuAddr){
        return sessionTable.get(rtuAddr) ;
    }
    
    /**
     * 得到Tcp通信协议名称
     * @param rtuAddr
     * @return
     */
    public static String getTcpProtocolName(String rtuAddr){
        TcpSession tcpSe = sessionTable.get(rtuAddr) ;
        if(tcpSe != null){
            return tcpSe.protocolName ;
        }else{
            return null ;
        }
    }
    
    /**
     * 得到所有在线情况
     * @return
     */
    public static HashMap<String, Boolean> allOnLine(){
        HashMap<String, Boolean> map = new HashMap<String, Boolean>();
        Iterator<Entry<String, TcpSession>> it = sessionTable.entrySet().iterator() ;
        Entry<String, TcpSession> entry = null ;
        while(it.hasNext()){
            entry = it.next() ;
            map.put(entry.getKey(), entry.getValue().ioSession.isConnected()) ;
        }
        return map ;
    }
    /**
     * 得到所有RTU连接状态情况
     * @return
     */
    public static List<RtuSessionStatus> allConnectStatus(){
        List<RtuSessionStatus> list = new ArrayList<RtuSessionStatus>();
        Iterator<Entry<String, TcpSession>> it = sessionTable.entrySet().iterator() ;
        Entry<String, TcpSession> entry = null ;
        while(it.hasNext()){
            entry = it.next() ;
            RtuSessionStatus vo = new RtuSessionStatus() ;
            vo.rtuAddr = entry.getKey() ;
            IoSession se = entry.getValue().ioSession ;
            vo.onTrueOffLine = se.isConnected() ;
            InetSocketAddress sa = (InetSocketAddress)se.getRemoteAddress() ;
            if(sa != null){
                InetAddress inetAddr = sa.getAddress() ;
                if(inetAddr != null){
                    vo.ip = inetAddr.getHostAddress() ;
                    vo.port = sa.getPort() ;
                }
            }
            list.add(vo) ;
        }
        return list ;
    }
    
    
    /**
     * 得到IoSession
     * @param rtuAddr
     * @return
     */
//    public IoSession getIoSession(String rtuAddr){
//        TcpSession tcpSe = sessionMap.get(rtuAddr) ;
//        if(tcpSe != null){
//            return tcpSe.ioSession ;
//        }
//        return null ;
//    }
    
    /**
     * 网络是否连接
     * @param rtuAddr
     * @return
     */
    public static Boolean isConnect(String rtuAddr){
        TcpSession tcpSe = sessionTable.get(rtuAddr) ;
        if(tcpSe != null){
            return tcpSe.ioSession.isConnected() ;
        }
        return null ;
    }
    
    /**
     * 通过IoSession输出数据
     * @param rtuAddr
     * @param data
     * @throws Exception
     */
    public static void write(String rtuAddr, byte[] data) throws Exception{
        TcpSession tcpSe = sessionTable.get(rtuAddr) ;
        if(tcpSe != null){
            if(tcpSe.ioSession.isConnected()){
                tcpSe.ioSession.write(data) ;
            }else{
                throw new Exception("Rtu连接已经关闭!") ;
            }
        }else{
            throw new Exception("Rtu未曾上线!") ;
        }
    }
    
    /**
     * 设置上行数据时刻
     * @param rtuAddr
     */
    public static void cacheUpDataTime(String rtuAddr){
        TcpSession tcpSe = sessionTable.get(rtuAddr) ;
        if(tcpSe != null){
            tcpSe.lastUpDataTime = System.currentTimeMillis() ;
        }
    }
    
    /**
     * 更新上行数据时刻
     * 当上行数据时刻已经过去一定时长,上行数据时刻清空
     */
    public static void updateUpDataTime(Long now){
        Iterator<TcpSession> it = sessionTable.values().iterator() ;
        TcpSession tcpSe = null ;
        while(it.hasNext()){
            tcpSe = it.next() ;
            if(tcpSe.lastUpDataTime != null){
                if(now - tcpSe.lastUpDataTime > ServerProperties.lastUpDataTimeLive){
                    tcpSe.lastUpDataTime = null ;
                }
            }
        }
    }
 
 
}