Springboot项目Websocket

第一步&＃xff1a;pom文件依赖

org.springframework.bootspring-boot-starter-websocket

第二步&＃xff1a;配置websocket

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;/*** &＃64;author lrx* &＃64;create 2020-08-14 19:33* websocket配置类*/
&＃64;Configuration
public class WebSocketConfig {&＃64;Beanpublic ServerEndpointExporter serverEndpointExporter() {return new ServerEndpointExporter();}}

第三步&＃xff1a;去除验证

第四步&＃xff0c;编写websocket

package com.mp.core.websocket;import com.alibaba.fastjson.JSONObject;
import com.mp.core.model.dto.pressure.RealDataDTO;
import com.mp.core.service.pressure.realdata.RealDataService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;import javax.annotation.PostConstruct;
import javax.websocket.*;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.Map;
import java.util.concurrent.CopyOnWriteArraySet;
import java.util.concurrent.atomic.AtomicInteger;/*** &＃64;author lrx* &＃64;description: TODO app站点实时数据* &＃64;date 2021/3/29 15:43*/
&＃64;Slf4j
&＃64;ServerEndpoint(value &＃61; "/websocketServer/appRealData/{userId}")
&＃64;Component
public class AppRealDataSocketServer {&＃64;PostConstructpublic void init() {System.out.println("websocket 加载");}&＃64;Autowiredprivate RealDataService realDataService;private static final AtomicInteger OnlineCount &＃61; new AtomicInteger(0);// concurrent包的线程安全Set&＃xff0c;用来存放每个客户端对应的Session对象。private static CopyOnWriteArraySet sessionSet &＃61; new CopyOnWriteArraySet();/*** 连接建立成功调用的方法*/&＃64;OnOpenpublic void onOpen(Session session) {sessionSet.add(session);int cnt &＃61; OnlineCount.incrementAndGet(); // 在线数加1log.info("有连接加入&＃xff0c;当前连接数为&＃xff1a;{}", cnt);}/*** 连接关闭调用的方法*/&＃64;OnClosepublic void onClose(Session session) {sessionSet.remove(session);int cnt &＃61; OnlineCount.decrementAndGet();log.info("有连接关闭&＃xff0c;当前连接数为&＃xff1a;{}", cnt);}/*** 收到客户端消息后调用的方法** &＃64;param message 客户端发送过来的消息*/&＃64;OnMessagepublic void onMessage(String message, Session session) {log.info("来自客户端的消息&＃xff1a;{}", message);}/*** 出现错误** &＃64;param session* &＃64;param error*/&＃64;OnErrorpublic void onError(Session session, Throwable error) {log.error("发生错误&＃xff1a;{}&＃xff0c;Session ID&＃xff1a; {}", error.getMessage(), session.getId());error.printStackTrace();}/*** 发送消息&＃xff0c;实践表明&＃xff0c;每次浏览器刷新&＃xff0c;session会发生变化。** &＃64;param session*/public void sendMessage(Session session) {try {// 获取参数Map pathParameters &＃61; session.getPathParameters();//获取用户idLong userId &＃61; Long.valueOf(pathParameters.get("userId"));Map> appRealDataDTOMap &＃61; realDataService.appRealDataDTOList(userId);log.info("会话 " &＃43; session.getId() &＃43; " 推送消息&＃xff1a;" &＃43; JSONObject.toJSONString(appRealDataDTOMap));session.getBasicRemote().sendText(JSONObject.toJSONString(appRealDataDTOMap));} catch (IOException e) {log.error("发送消息出错&＃xff1a;{}", e.getMessage());e.printStackTrace();} catch (Exception e) {e.printStackTrace();}}/*** 群发消息** &＃64;throws IOException*/public void broadCastInfo() throws IOException {log.info("&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;开始群发消息&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;&＃61;");for (Session session : sessionSet) {if (session.isOpen()) {sendMessage(session);}}}
}