package com.fzzy.gateway.sc2023.websocket;
|
|
import lombok.extern.slf4j.Slf4j;
|
|
import org.springframework.stereotype.Component;
|
|
import javax.websocket.*;
|
import javax.websocket.server.PathParam;
|
import javax.websocket.server.ServerEndpoint;
|
|
import java.util.HashMap;
|
import java.util.Map;
|
|
/**
|
*
|
*/
|
@Slf4j
|
@Component
|
@ServerEndpoint(value = "/mqtt")
|
public class WebSocketMqtt {
|
|
private static Map<String, Session> sessionPool = new HashMap<>();
|
|
// 与某个客户端的连接会话，需要通过它来给客户端发送数据
|
private Session session;
|
private String key;
|
|
|
@OnOpen
|
public void onOpen(Session session,
|
@PathParam("keepalive") String keepalive,
|
@PathParam("clientId") String clientId,
|
@PathParam("protocolId") String protocolId,
|
@PathParam("protocolVersion") String protocolVersion,
|
@PathParam("clean") String clean,
|
@PathParam("reconnectPeriod") String reconnectPeriod,
|
@PathParam("reconnectTimeout") String reconnectTimeout
|
) throws Exception {
|
|
this.session = session;
|
|
key = clientId;
|
|
sessionPool.put(key, session);
|
|
|
log.info("new webSocket,clientId={}", key);
|
}
|
|
@OnClose
|
public void onClose() {
|
|
sessionPool.remove(key);
|
|
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("发生错误");
|
|
sessionPool.remove(key);
|
|
error.printStackTrace();
|
}
|
|
}
|
