SpringBoot+WebSocket实现在线客服咨询功能

package com.yzx.websocketdemo.server; import com.alibaba.fastjson2.JSON; import com.yzx.websocketdemo.pojo.Message; import org.omg.PortableInterceptor.INACTIVE; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Component; import javax.websocket.*; import javax.websocket.server.PathParam; import javax.websocket.server.ServerEndpoint; import java.io.IOException; import java.security.SecureRandom; import java.util.List; import java.util.Map; import java.util.Set; import java.util.concurrent.ConcurrentHashMap; import java.util.concurrent.CopyOnWriteArrayList; import java.util.concurrent.CopyOnWriteArraySet; @Component @ServerEndpoint("/chat/{type}") public class ChetServer { //存放每个客户端对应的Session对象 //ConcurrentHashMap实现了 Map 接口，基本功能与普通的 HashMap一样。但它具备在多线程环境下高效且安全地进行并发访问和操作的能力。 //private static Map<String, Session> onlineUsersSession = new ConcurrentHashMap<>(); //随机生成userID private static SecureRandom secureRandom = new SecureRandom(); private Session session; //用户信息 //存放每个客户端对应的MyWebSocket对象 private static CopyOnWriteArraySet<ChetServer> webSocketSet = new CopyOnWriteArraySet(); //接收sid private String sid = ""; /** * 连接建立成功调用的方法 * @param session * @param type 0代表user/1代表service * @throws IOException */ @OnOpen public void onOpen(Session session,@PathParam("type") Integer type) throws IOException { String ID = ""; if(type==0){ byte[] buffer = new byte[10]; secureRandom.nextBytes(buffer);// 将随机字节填充到指定的字节数组中 ID = buffer.toString(); }else if(type==1){ ID = "service"; } //设置当前用户信息 this.sid = ID; this.session = session; webSocketSet.add(this); //向客服端发送在线用户id Session serviceSession = getSessionById("service"); if(serviceSession!=null){ //获取当前所有在线用户的id List<String> userIds = getOnlineUsersId(); //String message = JSON.toJSONString(userIds); String message = userIds.toString(); serviceSession.getBasicRemote().sendText(message); } } /** * 连接关闭调用的方法 */ @OnClose public void onClose(Session session) throws IOException { webSocketSet.remove(this); //从set中删除 //当前用户下线后,使用客服端的session再给前端发送一下当前在线用户id Session serviceSession = getSessionById("service"); if(serviceSession!=null){ //获取当前在线用户id List<String> userIds = getOnlineUsersId(); //String message = JSON.toJSONString(userIds); String message = userIds.toString(); serviceSession.getBasicRemote().sendText(message); } } /** * 当浏览器发消息到服务器时，执行该方法 * * @param message = {"ToId":"***","msg":"***"} */ @OnMessage public void onMessage(String message) throws IOException { System.out.println(message.toString()); //msg01表示客户端发来的消息，需要服务端转发给目标用户 Message msg01 = JSON.parseObject(message, Message.class); //要发的消息 String mess = msg01.getMsg(); //发给谁 String toId = msg01.getToId(); //谁发的 String fromId = this.sid; //获取到目标用户的session Session toSession = getSessionById(toId); if(toSession==null){ switch (this.sid){ case "service": this.session .getBasicRemote().sendText("当前用户已下线~"); System.out.println("当前用户已下线~"); break; default: this.session .getBasicRemote().sendText("客服已下线~"); System.out.println(this.sid+"你好，客服已下线~"); break; } }else { Message msg02 = new Message(toId, fromId, mess); String json_msg02 = JSON.toJSONString(msg02); toSession.getBasicRemote().sendText(json_msg02); System.out.println(json_msg02); } } /** * @ Param session * @ Param error */ @OnError public void onError(Throwable error) { System.out.println("发生错误"); error.printStackTrace(); } //获取所有在线用户Id public List<String> getOnlineUsersId(){ List<String> userIds = new CopyOnWriteArrayList<String>(); for (ChetServer C:webSocketSet){ if(!C.sid.equals("service")) userIds.add(C.sid); } return userIds; } //通过用户id获取session public Session getSessionById(String userId) { for (ChetServer C:webSocketSet){ if (C.sid.equals(userId)){ return C.session; } } return null; } }