当前项目中由于用到了webSocket 于是做一个总结,当前项目中由于需要后台监控日志程序,所以需要一直给前段发送请求,这就用到了webSocket以下是代码
依赖
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
配置bean
@Component
public class WebSocketConfig {
@Bean
public ServerEndpointExporter serverEndpointExporter() {
return new ServerEndpointExporter();
}
}
service
@Component
@ServerEndpoint("/fileWebSocket")
public class FileWebSocket {
private Logger log = LoggerFactory.getLogger(FileWebSocket.class);
private Session session;
private static CopyOnWriteArraySet<FileWebSocket> webSocketSet = new CopyOnWriteArraySet<>();
@OnOpen
public void onOpen(Session session) {
this.session = session;
webSocketSet.add(this);
log.info("【websocket消息】有新的连接," + "id为:" + session.getId() + " 总数:{}", webSocketSet.size());
}
@OnClose
public void onClose() {
webSocketSet.remove(this);
log.info("【websocket消息】连接断开, 总数:{}", webSocketSet.size());
}
@OnMessage
public void onMessage(Session session, String message){
log.info("【websocket消息】收到" + session.getId() + "发来的消息:{}", message);
log.info("收到来自" + session.getId() + "的信息" + message);
sendMessage(message);
}
public void sendMessage(String message) {
for (FileWebSocket webSocket : webSocketSet) {
// log.info(session.getId() + "发送的消息, message={}", message);
try {
webSocket.session.getBasicRemote().sendText(message);
} catch (Exception e) {
e.printStackTrace();
}
}
}
public void sendMessage(Map message) {
for (FileWebSocket webSocket : webSocketSet) {
// log.info(session.getId() + "发送的消息, message={}", message);
try {
webSocket.session.getBasicRemote().sendText(JSONUtil.toJsonStr(message));
} catch (Exception e) {
e.printStackTrace();
}
}
}
}