WebSocketServer.java
3.11 KB
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
package com.huaheng.mobile.websocket;
import org.apache.juli.logging.Log;
import org.apache.juli.logging.LogFactory;
import org.springframework.stereotype.Component;
import javax.websocket.OnClose;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.concurrent.CopyOnWriteArrayList;
@ServerEndpoint("/websocket/{vmcNo}") // 客户端URI访问的路径
@Component
public class WebSocketServer {
/** 保存所有连接的webSocket实体
* CopyOnWriteArrayList使用了一种叫写时复制的方法，
* 当有新元素添加到CopyOnWriteArrayList时，
* 先从原有的数组中拷贝一份出来，然后在新的数组做写操作，
* 写完之后，再将原来的数组引用指向到新数组。
* 具备线程安全，并且适用于高并发场景
*/
private static CopyOnWriteArrayList<WebSocketServer> sWebSocketServers = new CopyOnWriteArrayList<>();
private Session mSession; // 与客户端连接的会话，用于发送数据
private long mVmcNo; // 客户端的标识(这里以机器编号)
private Log mLog = LogFactory.getLog(WebSocketServer.class);
@OnOpen
public void onOpen(Session session, @PathParam("vmcNo") long vmcNo){
mSession = session;
sWebSocketServers.add(this); // 将回话保存
System.out.println("-->onOpen new connect vmcNo is "+vmcNo);
mVmcNo = vmcNo;
}
@OnClose
public void onClose(){
sWebSocketServers.remove(this);
System.out.println("-->onClose a connect");
}
@OnMessage
public void onMessage(String message, Session session){
System.out.println("-->onMessage "+message);
// 这里选择的是让其他客户端都知道消息，类似于转发的聊天室，可根据使用场景使用
// for (WebSocketServer socketServer : sWebSocketServers){
// socketServer.sendMessage("i have rcv you message");
// }
}
/** 对外发送消息
* @param message
*/
public boolean sendMessage(String message){
try {
mSession.getBasicRemote().sendText(message);
} catch (IOException e) {
mLog.info(e.toString());
return false;
}
return true;
}
/** 对某个机器发送消息
* @param message
* @param vmcNo 机器编号
* @return true,返回发送的消息,false，返回failed字符串
*/
public static String sendMessage(String message, long vmcNo){
boolean success = false;
for (WebSocketServer server : sWebSocketServers){
if (server.mVmcNo == vmcNo){
success = server.sendMessage(message);
break;
}
}
return success ? message : "failed";
}
public static String send(String message){
boolean success = false;
for (WebSocketServer server : sWebSocketServers){
success = server.sendMessage(message);
}
return success ? message : "failed";
}
}