Spring Boot整合WebSocket实现实时通信,前端实时通信,前后端实时通信

Spring Boot整合WebSocket实现实时通信

实时通信在现代Web应用中扮演着越来越重要的角色,无论是在线聊天、股票价格更新还是实时通知,WebSocket都是实现这些功能的关键技术之一。Spring Boot作为一个简化企业级应用开发的框架,其对WebSocket的支持也非常友好。本文将详细介绍如何在Spring Boot中整合WebSocket,实现一个简单的实时通信服务。

引言

在 Spring Boot 中使用 WebSocket 有 2 种方式。第 1 种是使用由 Jakarta EE 规范提供的 Api,也就是 jakarta.websocket 包下的接口。第 2 种是使用 spring 提供的支持,也就是 spring-websocket 模块。前者是一种独立于框架的技术规范,而后者是 Spring 生态系统的一部分,可以与其他 Spring 模块(如 Spring MVC、Spring Security)无缝集成,共享其配置和功能。因为是进行学习进行使用webSocket所以进行选择第一种方式来进行实现.

本文将使用第 1 种方式,也就是使用 jakarta.websocket 来开发 WebSocket 应用。

1. 环境准备

  • JDK 1.8 或更高版本
  • Spring Boot 2.x
  • IntelliJ IDEA / Eclipse (或其他Java IDE)
  • Maven 或 Gradle 作为构建工具
  • springboot版本:3.2.5

2. 添加依赖

打开pom.xml文件,并添加WebSocket相关的依赖.


        <dependency>
            <groupId>org.springframework.bootgroupId>
            <artifactId>spring-boot-starter-websocketartifactId>
        dependency>

3. 开发 ServerEndpoint 端点

ServerEndpoint是一个注解,用于定义WebSocket服务器端点。使用这个注解,你可以指定WebSocket的URL模式,以便客户端能够连接到你的服务.

package com.fs.webSocket;

import jakarta.websocket.*;
import jakarta.websocket.server.ServerEndpoint;

// 这个和Controller的作用是类似的
@ServerEndpoint(value = "/websocket") // 此类是一个webSocket端点 服务端,监听/websocket的路径
public class MyServerEndpoint {
    // 定义一个Session对象
    private Session session;
    // 客户端与服务端建立连接触发
    @OnOpen
    public void onOpen(Session session) {
        // 保存session
        this.session = session;
        System.out.println("websocket连接成功");
    }
    // 客户端向服务端发送消息
    @OnMessage
    public void onMessage(String message) {
        System.out.println("收到客户端消息:" + message);
    }

    // 客户端服务端连接异常触发
    @OnError
    public void onError(Session session, Throwable error) {
        System.out.println("websocket连接异常");
    }

    // 客户端与服务端断开连接触发
    @OnClose
    public void onClose(Session session) {
        System.out.println("websocket连接关闭");
    }

}

注意:所有事件方法,都支持使用 Session 作为参数,表示当前连接参数。但是为了更加方便,我们在 @OnOpen 事件中直接把 Session 存储到了当前对象中,可以在任意方法中使用 this 访问。服务器会为每个连接创建一个端点对象,所以这是线程安全的。

4. 配置webSocket

@Configuration
public class WebSocketConfig {
    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
        ServerEndpointExporter exporter = new ServerEndpointExporter();
        exporter.setAnnotatedEndpointClasses(MyServerEndpoint.class); // 进行手动注册webSocket端点
        return exporter;
    }
}

当然也可以在MyServerEndpoint这个类上面进行添加 @Component注解,Spring 自动扫描,这样的话不需要手动调用 setAnnotatedEndpointClasses 方法进行注册。

5.前端集成测试

在资源文件夹resorces/static/index.html中书写一个简单的前端websocket进行连接到服务器通信,

DOCTYPE html>
<html>
<head>
    <meta charset="UTF-8">
    <title>WebSockettitle>
head>
<body>
<script type="text/javascript">
    let websocket = new WebSocket("ws://localhost:8080/websocket");

    // 连接断开
    websocket.onclose = e => {
        console.log(`连接关闭: code=${e.code}, reason=${e.reason}`)
    }
    // 收到消息
    websocket.onmessage = e => {
        console.log(`收到消息:${e.data}`);
    }
    // 异常
    websocket.onerror = e => {
        console.log("连接异常")
        console.error(e)
    }
    // 连接打开
    websocket.onopen = e => {
        console.log("连接打开");

        // 创建连接后,往服务器连续写入3条消息
        websocket.send("sprigdoc.cn");
        websocket.send("sprigdoc.cn");
        websocket.send("sprigdoc.cn");

        // 最后发送 bye,由服务器断开连接
        websocket.send("bye");

        // 也可以由客户端主动断开
        // websocket.close();
    }
script>
body>
html>

启动应用,打开浏览器(先打开控制台),然后访问 http://localhost:8080/,查看控制台服务端输出的日志:

websocket连接成功
收到客户端消息:sprigdoc.cn
收到客户端消息:sprigdoc.cn
收到客户端消息:sprigdoc.cn
收到客户端消息:bye

此时代表连接成功!!!

结语

通过以上步骤,我们已经创建了一个简单的Spring Boot WebSocket的连接。展示了如何使用Spring Boot和WebSocket实现实时通信的基础。在实际项目中,你可能需要考虑更多的因素,如安全性、连接管理、广播优化等。

希望这篇教程能够帮助你开始探索Spring Boot与WebSocket的整合。如果你有任何疑问或建议,请在评论区告诉我。祝你编程愉快!

附录

  • Spring Boot官方文档: https://docs.spring.io/spring-boot/docs/current/reference/html/websocket.html
  • WebSocket协议规范: https://tools.ietf.org/html/rfc6455

你可能感兴趣的:(spring,boot,websocket,前端,java,后端)