SpringBoot集成WebSokcet代码步骤

 更新时间:2026年01月16日 10:39:01   作者:盹猫  
Spring Boot是一个流行的Java框架,用于快速构建Spring应用程序,WebSocket是一种实现实时通信的协议,与HTTP协议结合使用,可以提供全双工通信,这篇文章主要介绍了SpringBoot集成WebSokcet代码步骤

前言

在Web开发时,有时我们需要一些实时性比较高的数据反馈,如设备监测、系统监测、聊天等,这时使用HTTP显然是不适合的,这时就需要用到WebSocket通过单次连接来获取长时间的数据获取。

什么是websocket?

WebSocket 是一种在单个 TCP 连接上实现全双工(双向)通信的网络协议,由 HTML5 规范定义,旨在它允许客户端(如浏览器)和服务器之间建立持久连接,双方可以随时向对方发送数据,而无需像 HTTP 那样每次通信都重新建立连接。

本篇文章就是记录SpringBoot如何集成WebSocket功能,废话不多说,让我们开始吧~

环境准备

因为要使用websocket,所以在除了集成基础的spring-boot-starter-web 依赖以外,还要集成

spring-boot-starter-websocket 以获得Websocket支持。

Pom.xml

<properties>
        <maven.compiler.source>8</maven.compiler.source>
        <maven.compiler.target>8</maven.compiler.target>
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
</properties>
<dependencies>
        <!--Web依赖-->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <!--WebSocket依赖-->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-websocket</artifactId>
            <version>2.3.12.RELEASE</version>
        </dependency>
</dependencies>

代码步骤

配置websocket节点

在配置类里,我们需要新建一个ServerEndpointExporter 并交给SpringBoot管理,即可开启WebSocket节点功能,内容如下:

@Configuration
public class WebSocketConfig {
    /**
     * 注入ServerEndpointExporter,
     * 这个bean会自动注册使用了@ServerEndpoint注解声明的Websocket endpoint
     */
    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
        return new ServerEndpointExporter();
    }
}

创建Websocket节点

首先需要在需使用到的WebSocket类上添加@ServerEndpoint(path,config) 以标记该类会作为一个websokcet的节点类,其中

  • path:该节点的访问路径
  • config: 可以做一些额外配置,如该节点的访问权限校验。

如下:

@Component
@Slf4j
@ServerEndpoint(value = "/detail", configurator = WebSocketAuthConfig.class)
public class DetailServer {}

下面就可以在该类中以事件标注的方式定义我们的方法,可以使用以下注解:

  • @OnOpen :在有新的连接接入时,触发方法。
  • @OnMessage: 当有新的消息时,触发方法。
  • @OnError:当发生错误时,触发方法。
  • @OnClose:当连接断开时,触发方法。

例如下面的方法定义:

    @OnOpen
    public void onOpen(Session session) {
        //这里一般会存储会话。
        log.info("【websocket消息】有新的连接");
    }

    /**
     * 链接关闭调用的方法
     */
    @OnClose
    public void onClose() {
        //这里会移除会话。
        log.info("【websocket消息】连接断开");
    }

    @OnMessage
    public void onMessage(String message) {
        log.info("【websocket消息】收到客户端消息:" + message);
    }

    @OnError
    public void onError(Session session, Throwable error) {
        log.error("用户错误,原因:" + error.getMessage());
    }

一般会创建一个用于存储当前当前Session的Map,用于会话的管理,并可以通过

session.getAsyncRemote().sendText(message);

将message数据发送到已连接的前端页面。

授权验证

当然不能允许所有人连接我们定义的WebSokcet节点,我们可以通过继承ServerEndpointConfig.Configurator 来对节点进行配置,一般可以用于添加请求配置或进行授权认证,如下实现了一个Jwt的token验证:

@Slf4j
public class WebSocketAuthConfig extends ServerEndpointConfig.Configurator {

    /**
     * DES:[拦截websocket的请求头,并验证]
     *
     * @param originHeaderValue 请求头
     * @return boolean
     * @link [java.lang.String]
     */
    @Override
    public boolean checkOrigin(final String originHeaderValue) {
        final ServletRequestAttributes servletRequestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        try {
            assert servletRequestAttributes != null;
            /*
              此处使用反射的方式获取对象,无法@AutoWiredW自动注入
             */
            final JwtUtil jwtUtil = SpringUtil.getBean(JwtUtil.class);
            HttpServletResponse response = servletRequestAttributes.getResponse();
            HttpServletRequest request = servletRequestAttributes.getRequest();
            //获取并验证请求参数中的token参数            
final String token = servletRequestAttributes.getRequest().getParameter("token");
            final Claims claims = jwtUtil.getClaimsByToken(token);
            if (ObjectUtil.isNull(claims)) {
                /*
                  Token验证失败,拒绝链接
                 */
                log.info("Token错误,拒绝连接......");
                request.getSession().setAttribute("loginFill",true);
                return  false;
            }else {
                request.getSession().setAttribute("loginFill",false);
            }
        } catch (final Exception e) {
            log.error("WebSocket-auth-error", e);
        }

        return true;
    }
}

当然这里也可以通过Shiro的根据路径配置拦截请求,这里也是支持的,这样ServerEndpointConfig.Configurator就可以不用单独配置。

总结

通过上述方式,就可以建立一个比较完善的WebSocket节点,在实际开发过程中可以解决一些实时性要求较高的数据传输。当然,实现WebSocket并不只是有这一种方式,也可以通过Netty框架去自己实现,这样在高并发系统下的效率会更高。

到此这篇关于SpringBoot集成WebSokcet的文章就介绍到这了,更多相关SpringBoot集成WebSokcet内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • Java 客户端操作 FastDFS 实现文件上传下载替换删除功能

    Java 客户端操作 FastDFS 实现文件上传下载替换删除功能

    这篇文章主要介绍了Java 客户端操作 FastDFS 实现文件上传下载替换删除功能,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2020-10-10
  • Elasticsearch查询之Match Query示例详解

    Elasticsearch查询之Match Query示例详解

    这篇文章主要为大家介绍了Elasticsearch查询之Match查询示例详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-04-04
  • 面试官问如何启动Java 线程

    面试官问如何启动Java 线程

    这篇文章主要介绍了面试官问如何启动Java 线程,Java 的线程创建和启动非常简单,但如果问一个线程是怎么启动起来的往往并不清楚,甚至不知道为什么启动时是调用start(),而不是调用run()方法呢?下面我们就一起进入文章学习这个问题吧
    2021-12-12
  • 深入探究Java中的HashMap为什么会产生死循环

    深入探究Java中的HashMap为什么会产生死循环

    HashMap 死循环发生在 JDK 1.8 之前的版本中,这篇文章主要来和大家深入探究一下为什么Java中HashMap会产生死循环,感兴趣的小伙伴可以了解一下
    2023-05-05
  • 详解Java编程中final,finalize,finally的区别

    详解Java编程中final,finalize,finally的区别

    这篇文章主要介绍了详解Java编程中final,finalize,finally的区别,这个在Java面试题中简直是太常见了...需要的朋友可以参考下
    2015-11-11
  • Java 深入浅出解析面向对象之抽象类和接口

    Java 深入浅出解析面向对象之抽象类和接口

    本章具体介绍了抽象类和接口,整篇文章用目前流行的手机来举例,图解穿插代码案例。 JAVA成仙路从基础开始讲,后续会讲到JAVA高级,中间会穿插面试题和项目实战,希望能给大家带来帮助
    2022-03-03
  • Java Zip文件读写操作详解

    Java Zip文件读写操作详解

    这篇文章主要为大家详细介绍了如何利用Java ZipInputstream、ZipOutputStream实现获取每个文件中的内容与写入内容,感兴趣的可以动手尝试一下
    2022-11-11
  • 2020年IntelliJ IDEA最新最详细配置图文教程详解

    2020年IntelliJ IDEA最新最详细配置图文教程详解

    这篇文章主要介绍了2020年IntelliJ IDEA最新最详细配置图文教程详解,本文通过图文并茂的形式给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2021-02-02
  • JVM生产环境调优实战案例讲解

    JVM生产环境调优实战案例讲解

    这篇文章主要介绍了JVM生产环境调优实战,本文通过实例代码给大家介绍的非常详细,感兴趣的朋友跟随小编一起看看吧
    2025-04-04
  • springboot接入mq的方法示例

    springboot接入mq的方法示例

    本文主要介绍了springboot接入mq的方法示例,主要实现配置以及实现一个简单的发送、接收消息的例子,具有一定的参考价值,感兴趣的可以了解一下
    2023-09-09

最新评论