package com.fzzy.gateway.hx2023.websocket;
|
|
import com.alibaba.fastjson.JSONObject;
|
import com.fzzy.gateway.hx2023.ScConstant;
|
import com.fzzy.gateway.hx2023.data.WebSocketPacket;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.lang.StringUtils;
|
import org.springframework.stereotype.Component;
|
|
import javax.websocket.*;
|
import javax.websocket.server.PathParam;
|
import javax.websocket.server.ServerEndpoint;
|
import java.util.Map;
|
import java.util.concurrent.ConcurrentHashMap;
|
|
/**
|
*
|
*/
|
@Slf4j
|
@Component
|
@ServerEndpoint(value = "/device-message-sender/{productId}/{deviceId}")
|
public class WebSocketDeviceLed {
|
|
private static Map<String, Session> sessionPool = new ConcurrentHashMap<>();
|
private static Map<String, String> sessionIds = new ConcurrentHashMap<>();
|
|
// 与某个客户端的连接会话，需要通过它来给客户端发送数据
|
private Session session;
|
|
|
@OnOpen
|
public void onOpen(Session session,
|
@PathParam("productId") String productId,
|
@PathParam("deviceId") String deviceId,
|
@PathParam("clientId") String clientId
|
) throws Exception {
|
|
this.session = session;
|
|
String key = productId + "-" + deviceId;
|
|
sessionPool.put(key, session);
|
sessionIds.put(session.getId(), key);
|
|
ScConstant.updateOnline(deviceId);
|
|
log.info("new webSocket,clientId={}", key);
|
}
|
|
@OnClose
|
public void onClose() {
|
|
String key = sessionIds.get(session.getId());
|
|
sessionPool.remove(key);
|
sessionIds.remove(session.getId());
|
|
String deviceId = key.substring(0, key.indexOf("-"));
|
|
ScConstant.updateOffOnline(deviceId);
|
|
log.info("WebSocket连接关闭={}", key);
|
|
|
}
|
|
/**
|
* 收到前端发送的信息
|
*
|
* @param message
|
* @param session
|
*/
|
@OnMessage
|
public void onMessage(String message, Session session) {
|
|
log.info("来自前端的信息:\n" + message);
|
}
|
|
@OnError
|
public void onError(Session session, Throwable error) {
|
log.error("发生错误");
|
|
String key = sessionIds.get(session.getId());
|
|
String deviceId = key.substring(0, key.indexOf("-"));
|
|
ScConstant.updateOffOnline(deviceId);
|
|
sessionPool.remove(key);
|
sessionIds.remove(session.getId());
|
error.printStackTrace();
|
}
|
|
|
/**
|
* @param packet
|
*/
|
public static void sendByPacket(WebSocketPacket packet) {
|
if (StringUtils.isEmpty(packet.getDeviceId())) {
|
log.error("WebSocket信息推送失败，设备编码为空。");
|
return;
|
}
|
|
String tag = packet.getDeviceId();
|
|
// 遍历推送
|
Session session;
|
for (String key : sessionPool.keySet()) {
|
if (key.indexOf(tag) != -1) {
|
session = sessionPool.get(key);
|
session.getAsyncRemote().sendText(
|
JSONObject.toJSONString(packet));
|
}
|
}
|
}
|
|
|
}
|
