【SOLUTION】Spring Boot 集成 WebSocket

🕗 发布于 2024-07-25 07:23 spring boot websocket 后端

1. Maven 依赖

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>

2. 注册ServerEndpointExporter、WebSocketConfigProperties

WebSocketConfigProperties：

建议代码中的第三方插件配置属性都通过@ConfigurationProperties注解类实现配置属性的定义，方便Spring管理和维护
建议第三方插件资源都设置一个enabel属性控制资源的加载，可以节省内存开销和代码安全

package xx.xx.xx.xx.xx;

import lombok.Data;
import org.springframework.boot.context.properties.ConfigurationProperties;

@Data
@ConfigurationProperties("spring.socket")
public class WebSocketConfigProperties {

    //是否开启WebSocket
    private boolean enabel;
}

ServerEndpointExporter：

WebSocket 需要创建一个ServerEndpointExporter的实例，ServerEndpointExporter负责注册ServerEndpoint和管理ServerContainer
配置类继承BeanFactoryAware 是因为WebSocket在每次连接时都会重新创建ServerEndpoint类，所以导致在ServerEndpoint类中无法通过@Autowrite自动注入IOC中的Bean，需要借助BeanFactory来获取对应的类

package xx.xx.xx.xx.xx;

import org.springframework.beans.BeansException;
import org.springframework.beans.factory.BeanFactory;
import org.springframework.beans.factory.BeanFactoryAware;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;

@ConditionalOnProperty(prefix = "spring.socket", name = "enabel", havingValue = "true")
@Configuration
@EnableConfigurationProperties(WebSocketConfigProperties.class)
public class WebSocketAutoConfig implements BeanFactoryAware {

    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
        return new ServerEndpointExporter();
    }

    @Override
    public void setBeanFactory(BeanFactory beanFactory) throws BeansException {
        SocketBaseService.setBeanFactory(beanFactory);
    }
}

配置SocketBaseService

这里配置SocketBaseService类的原因有以下几点：

每个Socket端点都需要创建一个ServerEndpoint类，所以在代码上可以节省时间成本且规范ServerEndpoint类
ServerEndpoint类有4个声明周期@OnOpen、@OnMessage、@OnError、@OnClose，而每次连接在四个周期方法内都需要统一处理一些逻辑，例如在@OnOpen时需要储存Session，在@OnClose时需要删除Session，@OnError时需要记录错误日志等；所以如果不写SocketBaseService类，则需要在每个ServerEndpoint中写重复的逻辑，导致代码可读性低，维护难等问题
子级只需要通过继承SocketBaseService 抽象类，实现内部4个生命周期的抽象方法，即可实现对应的业务逻辑，由于子级的4个生命周期方法是父级抽象出来的，所以子级方法上就无需在标注@OnOpen、@OnMessage、@OnError、@OnClose注解

package xx.xx.xx.xx.xx;

import cn.hutool.core.util.*;
import org.springframework.beans.factory.BeanFactory;
import org.springframework.beans.factory.annotation.Autowired;

import javax.websocket.*;
import java.lang.reflect.Field;
import java.util.Arrays;
import java.util.concurrent.ConcurrentHashMap;


public abstract class SocketBaseService {


    protected String randomId;

    protected Session session;

    protected static BeanFactory beanFactory;

    /**
     * 会话池
     */
    private static ConcurrentHashMap<String, Session> sessionPool = new ConcurrentHashMap<>();

    /**
     * 解析自动装配属性
     */
    private void processAutowriteFields() {
        Field[] fields = this.getClass().getDeclaredFields();
        Arrays.stream(fields).forEach(field -> {
            Autowired annotation = field.getAnnotation(Autowired.class);
            if (ObjectUtil.isEmpty(annotation)) {
                return;
            }
            Object bean = SocketBaseService.beanFactory.getBean(field.getType());
            if (ObjectUtil.isNotEmpty(bean)) {
                try {
                    field.setAccessible(true);
                    field.set(this, bean);
                } catch (IllegalAccessException e) {
                }
            }
        });
    }


    @OnOpen
    public void onConnect(Session session) {
        this.processAutowriteFields();

        this.randomId = IdUtil.nanoId();
        this.session = session;
        sessionPool.put(randomId, session);
        this.afterConnect(session);
    }

    /**
     * 连接回调
     *
     * @param session
     */
    public abstract void afterConnect(Session session);

    /**
     * 消息
     *
     * @param message
     */
    @OnMessage
    public void onMessage(String message) {
        this.afterMessage(message);
    }

    /**
     * 接受消息回调
     *
     * @param message
     */
    public abstract void afterMessage(String message);

    /**
     * 连接错误
     *
     * @param session
     * @param error
     */
    @OnError
    public void onError(Session session, Throwable error) {
        this.afterError(error);
    }

    /**
     * 连接错误回调
     *
     * @param error
     */
    public abstract void afterError(Throwable error);

    /**
     * 关闭连接
     */
    @OnClose
    public void onClose() {
        this.beforeClose();
        sessionPool.remove(randomId);
    }

    /**
     * 关联连接回调
     */
    public abstract void beforeClose();
}

案例：文件上传进度监听

在面对大文件上传时通常遇到无法即时获取上传进度的问题，在没有WebSocket之前都是异步上传，然后通过轮询的方式实时获取上传进度
而在有了WebSocket后就更方便了，可以实现持续连接，实现实时获取上传进度，步骤如下：

第一步：异步上传

由于userFileService.asynceUpload是异步方法，所以这里不会等待userFileService.asynceUpload方法执行，而是立即返回任务id

Controller

    @PostMapping("/asynceUpload")
    @Transactional
    public Object asynceUpload(FileInfo info) {
    //获取一个异步上传任务id，方便后续跟踪文件上传进度
        String taskId = userFileService.getAsynceUploadTaskId(info);
        userFileService.asynceUpload(info, taskId);
        return taskId;
    }

Service

Spring如果想实现异步方法，需要在方法上添加@Async注解，并且在启动类或配置类上添加@EnabelAsync注解开启异步任务

@Async
@Override
public String asynceUpload(FileInfo info, String taskId) {
//处理文件
//更新进度
}

第二步：监听上传进度

监听步骤：

继承上文的SocketBaseService 抽象类，实现4个生命周期抽象方法
在afterConnect方法中注册定时任务，实现定时获取任务进度
在afterMessage中接受接口传的任务id
在beforeClose中关闭定时任务
在exce方法中编写获取任务进度的逻辑，并通过Session返回获取到的进度信息

注意：

子类中的teamDataService属性通过@Autowite注入成功是因为，在SocketBaseService 的onConnect方法中执行了processAutowriteFields方法，而processAutowriteFields通过获取当前类中标注了@Autowite注解的字段，再根据字段的类型通过BeanFactory在IOC获取对应的实例的方式注册

package xx.xx.xx.xx.xx.xx;

import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.cron.CronUtil;
import cn.hutool.cron.task.Task;
import cn.hutool.json.JSONUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import xx.common.web.socket.SocketBaseService;
import xx.xx.client.server.domain.TeamData;
import xx.xx.client.server.service.TeamDataService;

import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

@Service
@ServerEndpoint("/socket/getSyncFileTask")
public class UserFileSocketController extends SocketBaseService {

    private String scheduleId;

    private List<String> tasks = new ArrayList<>();

    @Autowired
    private TeamDataService teamDataService;

    @Override
    public void afterConnect(Session session) {

        scheduleId = CronUtil.schedule("*/2 * * * * *", new Task() {
            @Override
            public void execute() {
                exce();
            }
        });
    }

    @Override
    public void afterMessage(String message) {
        this.tasks = JSONUtil.toList(message, String.class);
    }

    @Override
    public void afterError(Throwable error) {

    }

    @Override
    public void beforeClose() {
        if (ObjectUtil.isNotEmpty(scheduleId)) {
            CronUtil.remove(scheduleId);
        }
    }

    private void exce() {
        if (session.isOpen()) {
        //获取任务进度
            List<TeamData> taskList = teamDataService.findByIds(tasks);
            //返回进度信息
            session.getAsyncRemote().sendText(JSONUtil.toJsonStr(taskList.stream().filter(el -> StrUtil.startWith(el.getKey(), "SYNCE_UPLOAD_")).collect(Collectors.toList())));
        }
    }
}

原文地址：https://blog.csdn.net/qq_40890233/article/details/140661156

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！

上一篇：16 敏捷开发实践（1）
下一篇：@Builder注释导致@RequestBody的前端json反序列化失败，HTTP400

【Linux】环境变量
环境变量与主函数参数
阅读更多2024-11-18
STL序列式容器之slist
STL是个双向链表（double linked list）。SGI STL另提供了一个单向链表（single linked list），名为slist。slist和list的主要差别在于，前者的迭代器
阅读更多2024-11-18
生产环境centos8 & Red Hat8部署ansible and 一键部署mysql两主两从ansible脚本预告
下载新的CentOS-Base.repo 到 /etc/yum.repos.d/ （这里用的CentOS 8.0）1.初始化磁盘为物理卷（PV）4.格式化逻辑卷为 XFS。5.挂载逻辑卷到 /db。3
阅读更多2024-11-18
AJAX入门
什么是 AJAX什么是 AJAX怎么用 AJAX语法：引入 axios.js：什么是 URL。
阅读更多2024-11-18
Visual Studio 2017 快捷键设置-批量注释和批量取消注释
设置Visual Studio 2017 快捷键，实现代码片段的批量注释和批量取消注释
阅读更多2024-11-18
uniapp自动注册机制：easycom
如果一个组件文件夹有多个组件文件，就需要在pages.json文件的easycom属性，批量引入注册到全局组件中。uniapp 框架提供了一种组件自动注册机制，只要你在 components 文件夹
阅读更多2024-11-18
动态规划 —— 子数组系列-等差数列划分
413. 等差数列划分 - 力扣（LeetCode）
阅读更多2024-11-18
单例模式详解
依赖注入是一种设计模式，通过将类的依赖项通过构造函数、属性或方法传入，而不是类自己创建依赖对象。
阅读更多2024-11-18
C++中的单例模式（Singleton）全面讲解与实际案例
通过单例模式，可以轻松实现全局统一管理、线程安全和资源共享。在多线程程序中，使用 C++11 的静态局部变量和互斥锁，可以确保日志记录器的安全性和稳定性。此代码是一个实际的应用案例，具有很高的实用性，
阅读更多2024-11-18
JVM--内存结构
PC Register，即程序计数器（Program Counter Register），是计算机处理器中的一个关键寄存器，也被称为指令计数器。它主要用于存放下一条指令所在单元的地址，是计算机能够连续
阅读更多2024-11-18

【SOLUTION】Spring Boot 集成 WebSocket

1. Maven 依赖

2. 注册ServerEndpointExporter、WebSocketConfigProperties

配置SocketBaseService

案例：文件上传进度监听

第一步：异步上传

Controller

Service

第二步：监听上传进度

相关文章