在上篇中,我们探讨了使用Java实现websocket的方式。但显然,上篇的实现方式还有一些细节的东西需要打磨一下。那么在下篇中,我们就一起优化一下上篇中的实现方式吧。
用户sid的指定
在上篇中,我们用sid来区分每个用户(也可以理解为每个网页窗口)。但sid是由前端页面在发送websocket请求时随请求过来的,如果由用户自行指定sid,那很容易出现sid重复的情况(也就是串线)。所以需要由程序来指定每个用户的sid。
sid值可以由前端设置,也可以由后端指定再返回给前端。
如果websocket同时在线用户数比较少,对重复性要求不高的情况下,可以直接在前端使用随机数算法:
function random(lower, upper) {
return Math.floor(Math.random() * (upper - lower+1)) + lower;
}
在打开一个websocket页面窗口时,调用随机数算法指定该窗口的用户sid。比如:
var sid = random(1,10000) // 在1-10000中生成随机数
不过显然,在前端使用随机数算法仍不可避免的出现sid重复性的问题。如果用户数在线数比较多的话,可以用uuid作为代替。
除了在前端指定sid以外,也可以首次请求的时候不带sid,由后端指定sid返回给前端,前端再绑定sid。这里可以由后端分发uuid或者用雪花算法计算的值,也可以通过后端维护一个sid的稽核,这里就不展开来讲,如果大家有兴趣,可以自己动手试试。
断线重连机制
断线重连机制下WebSocketServer类调整
因网络原因或者其他原因,有可能会出现断线的情况,而客户端或者服务器没有监测机制,无法得知连接是否保持。这时候需要建立断线重连机制(也可以说心跳重连机制)。
简单而言,就是设置定时器,每隔一定时间发送一次特定的信息(心跳),当服务器返回任何信息(包括心跳回复或信息推送),都证明连接可靠,此时重置定时器。
我们首先调整一下服务器端WebSocketServer类,在onMessage方法中增加心跳相关的代码。作为心跳的特定字符串不能为常见的字符串,以免服务端将正常的信息和心跳混淆。
@ServerEndpoint(value = "/ws/{sid}")
@Component
public class WebSocketServer {
......
// 根据心跳机制调整onMessage方法
@OnMessage
public void onMessage(String message, Session session) {
// 对心跳进行回复(回复自己)
if (message.equals("$$")) {
// 心跳为$$
try {
sendInfo("1$",sid); // 假定心跳返回的字符串为1$
} catch (IOException e) {
e.printStackTrace();
}
} else {
// 对正常信息进行处理
log.info("收到来自窗口" + sid + "的信息:" + message);
// if(StringUtils.isNotBlank(message)){
// for(WebSocketServer server:websocketMap.values()) {
// try {
// server.sendMessage(message);
// } catch (IOException e) {
// e.printStackTrace();
// continue;
// }
// }
// }
try {
//从message中解析出toSid和content
Map map = JSONObject.parseObject(message, Map.class);
String toSid = (String) map.get("toSid");
String content = "sid" + sid + ":"