springboot集成Websocket

目录

一、集成

1、pom依赖

2、webSocket端口处理消息 

3、注册websocket端口

二、demo

1、pom依赖

2、dto

3、constant 

4、webSocket端口处理

5、webSocket端口注册

6、controller接口

7、前端文件

8、application.properties配置文件

9、启动类

10、测试

10.1、user消息

10.2、menu消息 


一、集成

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)的没有接收到

SpringBoot集成WebSocket可以使用Spring框架提供的WebSocket API来实现。在SpringBoot中,使用WebSocket需要进行以下几个步骤: 1. 添加依赖:在pom.xml文件中添加spring-boot-starter-websocket依赖。 2. 创建WebSocket配置类:创建一个WebSocket配置类,用于配置WebSocket相关的参数,如注册EndPoint、消息编解码器、拦截器等。 3. 创建EndPoint:创建一个WebSocket的EndPoint类,用于处理WebSocket连接、消息发送、关闭等操作。可以通过实现Spring提供的WebSocketHandler接口或者继承TextWebSocketHandler来实现。 4. 添加拦截器:可以添加自定义的拦截器,用于处理WebSocket连接建立、消息发送等事件。 5. 配置WebSocket消息代理:使用Spring提供的消息代理,可以实现WebSocket消息的广播和点对点传递。 以下是一个简单的SpringBoot集成WebSocket的示例代码: 1. 添加依赖 ``` <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-websocket</artifactId> </dependency> ``` 2. 创建WebSocket配置类 ``` @Configuration @EnableWebSocket public class WebSocketConfig implements WebSocketConfigurer { @Override public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) { registry.addHandler(myHandler(), "/myHandler") .addInterceptors(new WebSocketInterceptor()); } @Bean public WebSocketHandler myHandler() { return new MyHandler(); } } ``` 3. 创建EndPoint ``` public class MyHandler extends TextWebSocketHandler { @Override public void afterConnectionEstablished(WebSocketSession session) throws Exception { // 处理连接建立事件 } @Override protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception { // 处理消息事件 } @Override public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception { // 处理连接关闭事件 } } ``` 4. 添加拦截器 ``` public class WebSocketInterceptor extends HandshakeInterceptor { @Override public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map<String, Object> attributes) throws Exception { // 处理连接建立前事件 return true; } @Override public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Exception ex) { // 处理连接建立后事件 } } ``` 5. 配置WebSocket消息代理 ``` @Configuration @EnableWebSocketMessageBroker public class WebSocketMessageBrokerConfig implements WebSocketMessageBrokerConfigurer { @Override public void configureMessageBroker(MessageBrokerRegistry registry) { registry.enableSimpleBroker("/topic"); registry.setApplicationDestinationPrefixes("/app"); } @Override public void registerStompEndpoints(StompEndpointRegistry registry) { registry.addEndpoint("/myEndpoint") .setAllowedOrigins("*") .withSockJS(); } } ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

w_t_y_y

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值