package org.springblade.common.handler;
|
|
import lombok.EqualsAndHashCode;
|
import lombok.extern.slf4j.Slf4j;
|
import okhttp3.WebSocket;
|
import org.springblade.core.tool.utils.StringUtil;
|
import org.springframework.stereotype.Component;
|
|
import javax.websocket.*;
|
import javax.websocket.server.PathParam;
|
import javax.websocket.server.ServerEndpoint;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.concurrent.ConcurrentHashMap;
|
import java.util.concurrent.CopyOnWriteArraySet;
|
import java.util.stream.Collectors;
|
|
@EqualsAndHashCode
|
@Component
|
@ServerEndpoint("/websocket/{code}/{targetCode}/{webType}")
|
@Slf4j
|
public class WebSocketServerService {
|
//当前设备编码
|
private String code;
|
|
//被控制的设备编码
|
private String targetCode;
|
|
//网页类型（1控制器，2显示器）
|
private String webType;
|
|
//实例一个session，这个session是websocket的session
|
private Session session;
|
|
//存放websocket的集合
|
private static CopyOnWriteArraySet<WebSocketServerService> webSocketSet = new CopyOnWriteArraySet<>();
|
|
//前端请求时一个websocket时
|
@OnOpen
|
public void onOpen(Session session, @PathParam("code") String code,@PathParam("targetCode") String targetCode,@PathParam("webType") String webType) {
|
List<WebSocketServerService> collect = webSocketSet.stream()
|
.filter(e -> e.code.equals(code))
|
.collect(Collectors.toList());
|
|
if (collect.size()>0){
|
log.info("【websocket消息】该设备已连接，无法再次连接，连接的设备id为{},连接类型为{},目标设备为{}，当前连接总数为{}",code,webType,targetCode,webSocketSet.size());
|
// sendErrorMessage("该设备已连接",code);
|
return;
|
}
|
|
this.session = session;
|
this.code = code;
|
this.targetCode = targetCode;
|
this.webType = webType;
|
webSocketSet.add(this);
|
log.info("【websocket消息】有新的连接，连接的设备id为{},连接类型为{},目标设备为{}, 当前连接总数:{}",code,webType,targetCode, webSocketSet.size());
|
}
|
|
//前端关闭时一个websocket时
|
@OnClose
|
public void onClose(@PathParam("code") String code,@PathParam("targetCode") String targetCode,@PathParam("webType") String webType) {
|
webSocketSet.remove(this);
|
log.info("【websocket消息】设备id为{}，类型为{}，连接断开, 总数:{}",code,webType, webSocketSet.size());
|
}
|
|
//前端向后端发送消息
|
@OnMessage
|
public void onMessage(String message,@PathParam("code") String code,@PathParam("targetCode") String targetCode,@PathParam("webType") String webType) {
|
log.info("【websocket消息】收到客户端发来的消息:{},发送消息的设备编码为{}，设备类型为{},目标设备为{}", message,code,webType,targetCode);
|
sendMessage(message,targetCode);
|
}
|
|
//新增一个方法用于主动向客户端发送消息
|
public static void sendMessage(String message,String targetCode) {
|
log.info("【websocket消息】广播消息, message={}", message);
|
for (WebSocketServerService webSocket: webSocketSet) {
|
try {
|
if (webSocket.code.equals(targetCode)){
|
webSocket.session.getBasicRemote().sendText(message);
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
|
public static void sendErrorMessage(String message,String code){
|
for (WebSocketServerService webSocket: webSocketSet) {
|
try {
|
if (webSocket.code.equals(code)){
|
webSocket.session.getBasicRemote().sendText(message);
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
|
|
}
|
