package com.fzzy.gateway.hx2023.websocket;
|
|
|
import com.alibaba.fastjson.JSON;
|
import com.alibaba.fastjson.JSONObject;
|
import org.springframework.stereotype.Component;
|
import org.springframework.stereotype.Service;
|
import org.springframework.web.bind.annotation.CrossOrigin;
|
|
import javax.websocket.*;
|
import javax.websocket.server.PathParam;
|
import javax.websocket.server.ServerEndpoint;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.concurrent.CopyOnWriteArraySet;
|
|
@Component
|
@CrossOrigin
|
@Service
|
@ServerEndpoint(value = "/mqtt")
|
public class SocketService {
|
|
/**
|
* 用来存放每个客户端对应的MyWebSocket对象。
|
**/
|
private static CopyOnWriteArraySet<SocketService> socketSet = new CopyOnWriteArraySet<>();
|
/**
|
* 与某个客户端的连接会话，需要通过它来给客户端发送数据
|
**/
|
private Session session;
|
/**
|
* 用户名称
|
**/
|
private String nickname;
|
/**
|
* 用来记录sessionId和该session进行绑定
|
**/
|
private static Map<String,Session> map = new HashMap<String, Session>();
|
|
/**
|
* 连接建立成功调用的方法
|
*/
|
@OnOpen
|
public void onOpen(Session session,@PathParam("nickname") String nickname) {
|
this.session = session;
|
this.nickname=nickname;
|
|
map.put(nickname, session);
|
socketSet.add(this);
|
|
System.out.println("有新连接加入:"+nickname+",当前在线人数为" + socketSet.size());
|
}
|
|
/**
|
* 连接关闭调用的方法
|
*/
|
@OnClose
|
public void onClose() {
|
socketSet.remove(this);
|
List<String> nickname = this.session.getRequestParameterMap().get("nickname");
|
for(String nick:nickname) {
|
map.remove(nick);
|
}
|
System.out.println("有一连接关闭！当前在线人数为" + socketSet.size());
|
}
|
|
/**
|
* 收到客户端消息后调用的方法
|
*/
|
@OnMessage
|
public void onMessage(String message, Session session,@PathParam("nickname") String nickname) {
|
System.out.println("来自客户端的消息-->"+nickname+": " + message);
|
//将mqtt发送过来的数据返回给前端
|
try {
|
// JSONObject parse = JSONObject.parse(message);
|
// Map<String,Object> valuesMap = (Map<String, Object>) parse.getByPath("values");
|
// String tag1 = valuesMap.get("tag1").toString();
|
// String replace = tag1.replace("\r\n", "");
|
// String substring = replace.substring(replace.indexOf(":") + 1).trim().replace(" ",":");
|
// String tag2 = valuesMap.get("tag2").toString();
|
// String substring1=substring+"tag2:"+tag2;
|
// String result="{"+substring1+"}";
|
// //发送给前端，用作页面渲染
|
// Session fromSession = map.get(nickname);
|
// fromSession.getAsyncRemote().sendText(result);
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
}
|
|
/**
|
* 发生错误时调用
|
*/
|
@OnError
|
public void onError(Session session, Throwable error) {
|
System.out.println("发生错误");
|
error.printStackTrace();
|
}
|
|
|
}
|
