目录
一、集成
1、pom依赖
<!--WebSocket的依赖-->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
2、webSocket端口处理消息
一般extends TextWebSocketHandler即可
3、注册websocket端口
implements WebSocketConfigurer
二、demo
1、pom依赖
<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion>
<groupId>org.example</groupId>
<artifactId>websocket-demo</artifactId>
<version>1.0-SNAPSHOT</version>
<properties>
<maven.compiler.source>17</maven.compiler.source>
<maven.compiler.target>17</maven.compiler.target>
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
</properties>
<parent>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-parent</artifactId>
<version>3.2.4</version>
<relativePath/>
</parent>
<dependencies>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter</artifactId>
</dependency>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-web</artifactId>
</dependency>
<dependency>
<groupId>org.projectlombok</groupId>
<artifactId>lombok</artifactId>
</dependency>
<dependency>
<groupId>com.google.code.gson</groupId>
<artifactId>gson</artifactId>
<version>2.10.1</version>
</dependency>
<!--WebSocket的依赖-->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
<!-- <dependency>-->
<!-- <groupId>javax.websocket</groupId>-->
<!-- <artifactId>javax.websocket-api</artifactId>-->
<!-- <version>1.1</version>-->
<!-- <scope>provided</scope>-->
<!-- </dependency>-->
</dependencies>
</project>
2、dto
package org.example.dto;
import lombok.Data;
@Data
public class MenuDTO {
private String url;
private String code;
private String name;
}
package org.example.dto;
import lombok.Data;
@Data
public class UserDTO {
private String userName;
private Integer age;
private String userAccount;
}
3、constant
这里常量是为了所有线程共享WebSocketSession
package org.example.constant;
import lombok.Getter;
import org.springframework.web.socket.WebSocketSession;
import java.util.HashSet;
import java.util.Set;
public class MenuSessionConstant {
@Getter
private static Set<WebSocketSession> sessions = new HashSet<>();
public static void add(WebSocketSession session) {
sessions.add(session);
}
public static void remove(WebSocketSession session) {
sessions.remove(session);
}
}
package org.example.constant;
import lombok.Getter;
import org.springframework.web.socket.WebSocketSession;
import java.util.HashSet;
import java.util.Set;
public class UserSessionConstant {
@Getter
private static Set<WebSocketSession> sessions = new HashSet<>();
public static void add(WebSocketSession session) {
sessions.add(session);
}
public static void remove(WebSocketSession session) {
sessions.remove(session);
}
}
4、webSocket端口处理
包括连接、断开连接、发送消息
package org.example.config;
import lombok.extern.slf4j.Slf4j;
import org.example.constant.UserSessionConstant;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;
import java.io.IOException;
@Component
@Slf4j
public class UserWebSocketHandler extends TextWebSocketHandler {
@Override
public void afterConnectionEstablished(WebSocketSession session) throws Exception {
UserSessionConstant.add(session);
log.info("user有新的连接,sessionId:{}",session.getId());
}
@Override
public void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
// 处理接收到的消息
log.info("user服务端Received message: {}",message.getPayload());
}
@Override
public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
UserSessionConstant.remove(session);
log.info("user连接已断开sessionId:{}",session.getId());
}
public void sendMessageToAll(String message) {
for(WebSocketSession session : UserSessionConstant.getSessions()) {
if (session.isOpen()) {
try {
session.sendMessage(new TextMessage(message));
log.info("user发送消息给{}成功",session.getId());
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
}
package org.example.config;
import lombok.extern.slf4j.Slf4j;
import org.example.constant.MenuSessionConstant;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;
import java.io.IOException;
@Component
@Slf4j
public class MenuWebSocketHandler extends TextWebSocketHandler {
@Override
public void afterConnectionEstablished(WebSocketSession session) throws Exception {
MenuSessionConstant.add(session);
log.info("menu有新的连接,sessionId:{}",session.getId());
}
@Override
public void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
// 处理接收到的消息
log.info("menu服务端Received message: {}",message.getPayload());
}
@Override
public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
MenuSessionConstant.remove(session);
log.info("menu连接已断开sessionId:{}",session.getId());
}
public void sendMessageToAll(String message) {
for(WebSocketSession session : MenuSessionConstant.getSessions()) {
if (session.isOpen()) {
try {
session.sendMessage(new TextMessage(message));
log.info("menu发送消息给{}成功",session.getId());
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
}
5、webSocket端口注册
package org.example.config;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;
@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
@Override
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
registry.addHandler(userWebSocketHandler(), "/user-websocket-endpoint").setAllowedOrigins("*");
registry.addHandler(menuWebSocketHandler(), "/menu-websocket-endpoint").setAllowedOrigins("*");
}
public UserWebSocketHandler userWebSocketHandler() {
return new UserWebSocketHandler();
}
public MenuWebSocketHandler menuWebSocketHandler() {
return new MenuWebSocketHandler();
}
}
6、controller接口
package org.example.controller;
import org.example.config.UserWebSocketHandler;
import org.example.dto.UserDTO;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
@RestController
@RequestMapping("/userMsg")
public class UserMsgController {
@Autowired
private UserWebSocketHandler userWebSocketHandler;
@RequestMapping("/send")
public void sendMessage(UserDTO messageDTO) {
userWebSocketHandler.sendMessageToAll(messageDTO.toString());
}
}
package org.example.controller;
import org.example.config.MenuWebSocketHandler;
import org.example.config.UserWebSocketHandler;
import org.example.dto.MenuDTO;
import org.example.dto.UserDTO;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
@RestController
@RequestMapping("/menuMsg")
public class MenuMsgController {
@Autowired
private MenuWebSocketHandler menuWebSocketHandler;
@RequestMapping("/send")
public void sendMessage(MenuDTO messageDTO) {
menuWebSocketHandler.sendMessageToAll(messageDTO.toString());
}
}
7、前端文件
在resource下面新建static文件夹,在static下新建user.html、menu.html
(1)user.html
<!DOCTYPE html>
<html>
<head>
<title>WebSocket Example</title>
</head>
<body>
<script>
var socket = new WebSocket("ws://localhost:5555/websocketDemo/user-websocket-endpoint");
socket.onopen = function(event) {
console.log("User WebSocket is open now.");
socket.send("Hello Server! This is user client");
};
socket.onmessage = function(event) {
console.log("Message from server ", event.data);
};
socket.onclose = function(event) {
console.log("WebSocket is closed now.");
};
socket.onerror = function(error) {
console.log("WebSocket Error: " + error);
};
</script>
</body>
</html>
(2) menu.html
<!DOCTYPE html>
<html>
<head>
<title>WebSocket Example</title>
</head>
<body>
<script>
var socket = new WebSocket("ws://localhost:5555/websocketDemo/menu-websocket-endpoint");
socket.onopen = function(event) {
console.log("Menu WebSocket is open now.");
socket.send("Hello Server! This is menu client");
};
socket.onmessage = function(event) {
console.log("Message from server ", event.data);
};
socket.onclose = function(event) {
console.log("WebSocket is closed now.");
};
socket.onerror = function(error) {
console.log("WebSocket Error: " + error);
};
</script>
</body>
</html>
8、application.properties配置文件
server.port=5555
server.servlet.context-path=/websocketDemo
9、启动类
package org.example;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.context.annotation.EnableMBeanExport;
import org.springframework.jmx.support.RegistrationPolicy;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
@SpringBootApplication
public class WebsocketApplication {
public static void main(String[] args) {
SpringApplication.run(WebsocketApplication.class, args);
}
}
10、测试
启动项目。
10.1、user消息
(1)先在浏览器中f12,浏览器访问http:5555/websocketDemo/user.html
idea控制台打印:
2024-08-27T10:40:07.316+08:00 INFO 40188 --- [nio-5555-exec-2] o.example.config.UserWebSocketHandler : user有新的连接,sessionId:745333c6-5f19-3fc1-bc18-9f9a377fa539
2024-08-27T10:40:07.321+08:00 INFO 40188 --- [nio-5555-exec-2] o.example.config.UserWebSocketHandler : user服务端Received message: Hello Server! This is user client
刷新浏览器,idea再次打印新的连接信息,可见每次刷新浏览器都相当于产生一个新的连接
(2)postman调用/userMsg/send接口
查看浏览器,成功接收
(3)重复(1)打开一个窗口
(4)重复(2),查看两个浏览器都接收到消息了:
10.2、menu消息
打开一个浏览器窗口,访问http://localhost:5555/websocketDemo/menu.html
postman访问
查看浏览器消息:menu的窗口成功接收了,非menu(user)的没有接收到