0


WebSocket详解(Java后端,基于Springboot)

WebSocket的介绍

WebSocket是一种在单个TCP连接上进行全双工通信的协议。这种协议在2011年被IETF定为标准RFC 6455,并由RFC7936补充规范,同时WebSocket API也被W3C定为标准。它使得客户端和服务器之间的数据交换变得更加简单,特别是允许服务端主动向客户端推送数据。

WebSocket技术具有一系列显著特点:

  1. 实时性更高:WebSocket技术可以实现实时通信,传输数据的延迟更低,因此可以更快地将数据传输到客户端。
  2. 减少网络带宽的使用:通过建立一个持久化的连接,WebSocket减少了HTTP协议中每次请求和响应所需要的网络带宽。
  3. 双向通信:WebSocket技术支持全双工通信模式,即客户端和服务器可以同时发送和接收数据。
  4. 跨域支持:WebSocket技术可以跨不同的域进行通信。
  5. 安全性更高:WebSocket可以通过SSL/TLS协议实现加密通信,确保数据传输的安全性。

WebSocket的应用场景非常广泛,包括但不限于:

  • 即时通讯:如聊天应用、在线客服系统等,用户可以实时地发送和接收消息。
  • 实时数据展示:如实时股票行情、实时天气更新等,可以实时地推送数据给前端。
  • 多人游戏:WebSocket可以实现多人在线游戏,玩家可以实时地进行交互和通信。
  • 实时协作:如实时协同编辑器,多个用户可以同时编辑一个文档,并实时地看到其他用户的操作。
  • 数据监控:WebSocket可以用于实时监控系统的运行状态、日志更新等,便于及时发现和解决问题。

注意,WebSocket也存在一些潜在的问题,如兼容性问题(特别是在一些旧版本的浏览器上),服务器资源占用(由于需要维护大量的长连接),以及安全性问题(需要特殊的安全设置以防止恶意攻击和数据泄漏)。

1. 添加依赖(基于idea)

首先,你需要在你的

pom.xml

(如果你使用Maven)或

build.gradle

(如果你使用Gradle)中添加Spring Boot WebSocket的依赖。

对于Maven,添加以下依赖:

<dependencies>  
    <!-- Spring Boot Starter WebSocket -->  
    <dependency>  
        <groupId>org.springframework.boot</groupId>  
        <artifactId>spring-boot-starter-websocket</artifactId>  
    </dependency>  
    <!-- 如果你还需要HTTP功能 -->  
    <dependency>  
        <groupId>org.springframework.boot</groupId>  
        <artifactId>spring-boot-starter-web</artifactId>  
    </dependency>  
    <!-- 其他依赖... -->  
</dependencies>

对于Gradle,添加以下依赖:

dependencies {  
    // Spring Boot Starter WebSocket  
    implementation 'org.springframework.boot:spring-boot-starter-websocket'  
    // 如果你还需要HTTP功能  
    implementation 'org.springframework.boot:spring-boot-starter-web'  
    // 其他依赖...  
}

2. 配置WebSocket

接下来,你需要配置WebSocket端点。我们继承WebMvcConfigurationSupport。以下是一个简单的配置示例:


import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.ResourceHandlerRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurationSupport;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;

/**
 * 开启WebSocket支持 配置websocket并开启
 */
@Configuration
public class WebSocketConfig extends WebMvcConfigurationSupport {
    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
        return new ServerEndpointExporter();
    }

    @Override
    protected void addResourceHandlers(ResourceHandlerRegistry registry) {
        super.addResourceHandlers(registry);
        // 相对路径
        registry.addResourceHandler("image/**").addResourceLocations("classpath:/static/image/");
        // 绝对路径
        // registry.addResourceHandler("image/**").addResourceLocations("file:" + "image/");
    }
}

上述代码的WebSocketConfig该类继承自

WebMvcConfigurationSupport

,并试图同时配置WebSocket和静态资源处理。然而,这种设计是有问题的,原因如下:

  1. 继承WebMvcConfigurationSupport: 继承WebMvcConfigurationSupport会关闭Spring Boot的自动配置功能,这意味着您将失去很多Spring Boot为MVC提供的默认配置。通常,如果您只是想要添加或覆盖某些配置,而不是完全自定义整个MVC配置,那么不推荐继承WebMvcConfigurationSupport
  2. WebSocket配置: 在这个配置类中,@Bean注解创建一个ServerEndpointExporter的bean,这通常用于配置JSR 356 (Java API for WebSocket) 的端点。但是,ServerEndpointExporter并不是Spring Boot的WebSocket消息代理的一部分,而是用于JSR 356 WebSocket端点的导出。如果您打算使用Spring的WebSocket消息代理(即STOMP over WebSocket),那么您不需要ServerEndpointExporter
  3. 静态资源处理: 通过覆盖addResourceHandlers方法来配置静态资源的位置。尽管这本身不是错误的,但如果您只是想要配置静态资源,那么您并不需要继承WebMvcConfigurationSupport。您可以直接在另一个配置类中通过实现WebMvcConfigurer接口来完成这个操作。

如果您打算使用Spring的WebSocket消息代理,那么您应该使用类似您之前的

WebSocketConfig

配置类(不继承

WebMvcConfigurationSupport

),并实现

WebSocketMessageBrokerConfigurer

接口。静态资源处理则可以通过实现

WebMvcConfigurer

接口来完成。

以下是一个改进后的配置示例:

WebSocket配置 (不使用

WebMvcConfigurationSupport

)

import org.springframework.context.annotation.Configuration;  
import org.springframework.messaging.simp.config.MessageBrokerRegistry;  
import org.springframework.web.socket.config.annotation.*;  
  
// 使用@Configuration注解来标识这个类是一个配置类  
@Configuration  
// 使用@EnableWebSocketMessageBroker注解来启用WebSocket消息代理支持  
@EnableWebSocketMessageBroker  
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {  
  
    @Override  
    public void configureMessageBroker(MessageBrokerRegistry config) {  
        // 配置消息代理的前缀,客户端订阅消息时需要使用这个前缀  
        config.enableSimpleBroker("/topic");  
        // 配置应用目标前缀,客户端发送消息到服务器时使用这个前缀  
        config.setApplicationDestinationPrefixes("/app");  
    }  
  
    @Override  
    public void registerStompEndpoints(StompEndpointRegistry registry) {  
        // 注册一个STOMP协议的WebSocket端点,客户端将连接到这个端点  
        // addEndpoint方法第一个参数是端点URL,第二个参数是使用的子协议,这里使用SockJS  
        registry.addEndpoint("/websocket").withSockJS();  
    }  
  
    // (可选)配置跨域策略  
    @Override  
    public void configureWebSocketTransport(WebSocketTransportRegistration registration) {  
        registration.setMessageSizeLimit(8192); // 设置消息大小限制  
        registration.setSendBufferSizeLimit(16384); // 设置发送缓冲区大小限制  
    }  
}

静态资源配置 (通过实现

WebMvcConfigurer

)

@Configuration  
public class WebMvcConfig implements WebMvcConfigurer {  
  
    @Override  
    public void addResourceHandlers(ResourceHandlerRegistry registry) {  
        registry.addResourceHandler("image/**")  
                .addResourceLocations("classpath:/static/image/");  
        // 如果需要绝对路径,可以添加另一个资源处理器  
    }  
}

WebSocketController.java

import org.springframework.beans.factory.annotation.Autowired;  
import org.springframework.messaging.handler.annotation.*;  
import org.springframework.stereotype.Controller;  
  
// 使用@Controller注解来标识这个类是一个控制器  
@Controller  
public class WebSocketController {  
  
    // 假设有一个服务来处理消息逻辑  
    @Autowired  
    private MessageService messageService;  
  
    // 使用@MessageMapping注解定义客户端发送消息到服务器的路由  
    @MessageMapping("/hello")  
    // 使用@SendTo注解定义服务器向客户端发送消息的路由  
    @SendTo("/topic/greetings")  
    public String greeting(String message) {  
        // 调用服务来处理消息  
        String processedMessage = messageService.processMessage(message);  
        // 返回处理后的消息给客户端  
        return processedMessage;  
    }  
  
    // (可选)定义一个方法处理连接事件  
    @EventHandler  
    public void handleConnect(SessionConnectedEvent event) {  
        // 处理客户端连接事件  
    }  
  
    // (可选)定义一个方法处理断开连接事件  
    @EventHandler  
    public void handleDisconnect(SessionDisconnectEvent event) {  
        // 处理客户端断开连接事件  
    }  
}

MessageService.java

import org.springframework.stereotype.Service;  
  
// 使用@Service注解来标识这个类是一个服务类  
@Service  
public class MessageService {  
  
    public String processMessage(String message) {  
        // 这里可以添加消息处理的逻辑,比如格式化、验证等  
        // 返回一个处理后的消息  
        return "Hello, " + message + "!";  
    }  
}

以上代码配置了一个简单的WebSocket服务器,包括WebSocket消息代理、端点配置以及消息处理器。

  • WebSocketConfig 类是WebSocket的配置类,通过实现 WebSocketMessageBrokerConfigurer 接口,定义了消息代理和端点的配置。
  • WebSocketController 类是WebSocket的控制器,使用 @MessageMapping 注解来映射客户端发送的消息,并使用 @SendTo 注解来指定消息发送的路由。
  • MessageService 类是一个简单的服务类,用来处理WebSocket接收到的消息。

3.基于

@ServerEndpoint

的注解实现websocket通信

在Spring框架中,

@ServerEndpoint

注解用于声明一个WebSocket端点,这个端点可以被客户端通过WebSocket协议进行连接。然而,

@ServerEndpoint

注解本身并不与Spring容器集成,所以直接使用

@Component

注解来将其声明为一个Spring管理的bean是不起作用的。

要解决这个问题,你需要使用

ServerEndpointExporter

来手动注册

@ServerEndpoint

注解的类,以便它们能够被Spring容器管理。

ServerEndpointExporter

是一个Spring Boot组件,它会自动注册带有

@ServerEndpoint

注解的类,使其可以作为WebSocket端点。

以下是如何使用

@ServerEndpoint

@Component

注解,并通过

ServerEndpointExporter

来实现WebSocket长连接的步骤:

1.创建一个带有

@ServerEndpoint

注解的类,用于处理WebSocket连接。

import javax.websocket.OnClose;  
import javax.websocket.OnMessage;  
import javax.websocket.OnOpen;  
import javax.websocket.Session;  
import javax.websocket.server.ServerEndpoint;  
import org.springframework.stereotype.Component;  
  
@ServerEndpoint("/ws/{userId}/{device}")  
@Component  
public class MyWebSocketEndpoint {  
  
    @OnOpen  
    public void onOpen(Session session, @PathParam("userId") String userId, @PathParam("device") String device) {  
        // 连接建立时的处理逻辑  
        System.out.println("Client connected: " + session.getId() + " with userId=" + userId + " and device=" + device);  
    }  
  
    @OnMessage  
    public void onMessage(String message, Session session) {  
        // 收到消息时的处理逻辑  
        System.out.println("Message received: " + message + " from session: " + session.getId());  
    }  
  
    @OnClose  
    public void onClose(Session session) {  
        // 连接关闭时的处理逻辑  
        System.out.println("Client disconnected: " + session.getId());  
    }  
}

请注意,上面的代码中使用了

@PathParam

注解来提取URL模板参数

userId

device

。但是,

@PathParam

并不是

javax.websocket

包中的一部分,它是JAX-RS规范的一部分。在标准的WebSocket API中,你需要在

@OnOpen

方法中手动解析路径参数。

  1. 创建一个配置类来注册
    ServerEndpointExporter
    
import org.springframework.context.annotation.Bean;  
import org.springframework.context.annotation.Configuration;  
import org.springframework.web.socket.server.standard.ServerEndpointExporter;  
  
@Configuration  
public class WebSocketConfig {  
  
    @Bean  
    public ServerEndpointExporter serverEndpointExporter() {  
        return new ServerEndpointExporter();  
    }  
}

在这个配置类中,我们创建了一个

ServerEndpointExporter

的bean,Spring Boot会自动扫描并注册所有带有

@ServerEndpoint

注解的类。

请注意

@PathParam

注解在标准的WebSocket API中并不适用。如果你需要处理URL模板参数,你可能需要手动解析它们,或者使用Spring的

@RequestMapping

HandlerAdapter

来创建一个自定义的WebSocket端点处理器。


本文转载自: https://blog.csdn.net/Android_data/article/details/137144297
版权归原作者 一键曾当百万狮 所有, 如有侵权,请联系我们删除。

“WebSocket详解(Java后端,基于Springboot)”的评论:

还没有评论