程序员经验分享-hwhale

【WebSocket】WebSocket使用，看这篇就行

2023-11-09

一、WebSocket连接的建立、消息的接收和回复

  1. 当涉及到WebSocket框架的深度使用时，一个流行的选择是使用Java的Spring框架来实现。下面是一个基本的示例，演示了如何使用Spring WebSocket框架进行深度使用：

首先，确保你的项目中包含了Spring WebSocket的相关依赖。在 pom.xml文件中添加以下依赖：

<dependencies>
    <!-- Spring WebSocket -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-websocket</artifactId>
    </dependency>
</dependencies>

接下来，创建一个WebSocket配置类，用于配置和管理WebSocket连接：

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(new MyWebSocketHandler(), "/websocket").setAllowedOrigins("*");
    }
}

在上面的示例中，MyWebSocketHandler是自定义的WebSocket处理程序，它将处理来自客户端的WebSocket连接和消息。

接下来，创建一个WebSocket处理程序类来处理WebSocket连接和消息的逻辑：

import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;

public class MyWebSocketHandler extends TextWebSocketHandler {

    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        // 当WebSocket连接建立成功时调用
        System.out.println("WebSocket连接已建立");
    }

    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        // 处理接收到的WebSocket消息
        String payload = message.getPayload();
        System.out.println("接收到消息：" + payload);
  
        // 发送回复消息给客户端
        String replyMessage = "收到消息：" + payload;
        session.sendMessage(new TextMessage(replyMessage));
    }

    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        // 当WebSocket连接关闭时调用
        System.out.println("WebSocket连接已关闭");
    }
}

在上面的示例中，我们覆盖了 afterConnectionEstablished、handleTextMessage和 afterConnectionClosed等方法，以处理WebSocket连接的建立、接收消息和连接关闭等事件。

最后，你可以在你的应用程序中使用WebSocket连接，例如在控制器中处理WebSocket请求：

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.messaging.handler.annotation.MessageMapping;
import org.springframework.messaging.handler.annotation.SendTo;
import org.springframework.stereotype.Controller;

@Controller
public class WebSocketController {

    @MessageMapping("/send")
    @SendTo("/topic/messages")
    public String sendMessage(String message) {
        // 处理接收到的消息，并返回处理结果
        String replyMessage = "处理消息：" + message;
        return replyMessage;
    }
}

在上面的示例中，我们使用 @MessageMapping注解来指定处理客户端发送的消息的路径，然后使用 @SendTo注解将处理结果发送回指定的订阅路径。

二、广播消息、用户认证

  • 广播消息：除了在 handleTextMessage方法中直接发送回复消息给单个客户端外，你还可以使用 SimpMessagingTemplate来广播消息给所有订阅了特定主题的客户端。可以在 WebSocketConfig中注入 SimpMessagingTemplate，然后在处理程序中使用它发送消息。

    @Autowired
private SimpMessagingTemplate messagingTemplate;

@Override
protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
    // 处理接收到的WebSocket消息
    String payload = message.getPayload();
    System.out.println("接收到消息：" + payload);

    // 广播消息给所有订阅了特定主题的客户端
    messagingTemplate.convertAndSend("/topic/messages", "收到消息：" + payload);
}

  • 用户认证和授权：如果你的应用程序需要对WebSocket连接进行认证和授权，你可以使用Spring Security框架来实现。你可以在 WebSocketConfigurer的 registerWebSocketHandlers方法中添加适当的拦截器来处理认证和授权逻辑。例如，使用 HandshakeInterceptor进行握手阶段的认证，并使用 ChannelInterceptor来拦截和处理每个消息的授权逻辑。

  • 处理其他WebSocket事件：除了处理连接建立、消息接收和连接关闭等事件外，你还可以覆盖其他WebSocket事件的处理方法，例如 handleTransportError用于处理传输错误，handleBinaryMessage用于处理二进制消息等。根据你的需求，选择合适的方法进行覆盖和处理。

  • 集成其他功能：你可以将WebSocket与其他功能集成，例如数据库访问、消息队列、实时通知等。使用适当的组件和库，将WebSocket与你的应用程序的其他部分无缝集成，以实现更复杂的功能。

三、转换器、存储与过滤器

  1. 自定义消息转换器：Spring WebSocket框架支持使用不同的消息转换器来处理不同类型的消息。你可以自定义消息转换器，以实现自定义的消息格式和处理逻辑。通过实现 WebSocketMessageConverter接口，你可以定义自己的消息转换器，并在 WebSocketConfig中进行配置。

    @Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {

    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
                registry.addHandler(new MyWebSocketHandler(), "/websocket")
                .setAllowedOrigins("*")
                .addInterceptors(new HttpSessionHandshakeInterceptor())
                .withSockJS();
    }

    @Override
    public void configureMessageBroker(MessageBrokerRegistry registry) {
        registry.enableSimpleBroker("/topic");
        registry.setApplicationDestinationPrefixes("/app");
    }

    @Override
    public boolean configureMessageConverters(List<MessageConverter> messageConverters) {
        // 添加自定义的消息转换器
        messageConverters.add(new MyMessageConverter());
        return true;
    }
}

  2. 广播消息给特定用户：除了广播消息给所有订阅了特定主题的客户端，你还可以使用 SimpMessagingTemplate将消息发送给特定用户。通过将用户标识符作为消息的目的地，你可以确保消息仅发送给特定用户。

    @Autowired
private SimpMessagingTemplate messagingTemplate;

public void sendMessageToUser(String userId, String message) {
   messagingTemplate.convertAndSendToUser(userId, "/queue/messages", message);
}

  3. 会话管理和状态存储：WebSocket连接可以建立长时间的会话，你可能需要管理会话状态和存储用户相关的数据。你可以使用 WebSocketSession对象来管理会话，并使用适当的存储机制（例如数据库、缓存等）来存储和检索会话状态。

  4. 消息拦截器和过滤器：Spring WebSocket框架提供了拦截器和过滤器机制，允许你在处理消息之前或之后执行额外的逻辑。通过实现 HandshakeInterceptor和 ChannelInterceptor接口，你可以编写自定义的拦截器和过滤器来处理身份验证、消息转换、日志记录等操作。

四、拦截器、STOMP协议

  1. 处理连接错误：WebSocket连接可能会出现错误，例如连接断开、超时等情况。你可以在 handleTransportError方法中处理这些连接错误，并采取适当的措施，例如记录日志、重新连接等。

    @Override
public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
    // 处理连接错误
    System.out.println("连接错误：" + exception.getMessage());

    // 可以根据具体情况采取适当的措施，例如关闭连接、重新连接等
}

  2. 使用WebSocket拦截器：WebSocket拦截器允许你在建立连接之前和之后执行额外的逻辑。你可以实现 HandshakeInterceptor接口，并在 WebSocketConfig中注册拦截器来处理握手阶段的逻辑。

    @Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {

    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(new MyWebSocketHandler(), "/websocket")
                .setAllowedOrigins("*")
                .addInterceptors(new MyHandshakeInterceptor())
                .withSockJS();
    }

    // ...
}

  3. 使用自定义注解：你可以定义自己的注解，以便在WebSocket处理方法中进行更细粒度的控制。通过创建自定义注解并使用 @Target(ElementType.METHOD)将其应用于处理方法，你可以在运行时执行特定的逻辑。

    @Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface CustomWebSocketHandler {
    // 添加自定义属性
}




@CustomWebSocketHandler
public void handleWebSocketMessage(WebSocketSession session, TextMessage message) {
    // 执行自定义逻辑
}

  4. 使用STOMP协议：STOMP（Simple Text Oriented Messaging Protocol）是一种简单的文本导向消息协议，用于在WebSocket之上进行消息传递。你可以使用Spring的STOMP支持来实现更高级的消息传递功能，例如订阅和发布、消息头、错误处理等。

五、广播管理、定时任务

  1. 消息广播和群组管理：除了向特定用户发送消息，你可能还需要实现消息的广播和群组管理功能。Spring WebSocket框架提供了 SimpMessagingTemplate和 SimpUserRegistry等工具类来实现这些功能。你可以使用 SimpMessagingTemplate发送消息给特定主题或群组，而 SimpUserRegistry则用于管理连接的用户和会话。

    @Autowired
private SimpMessagingTemplate messagingTemplate;

@Autowired
private SimpUserRegistry userRegistry;

public void broadcastMessage(String topic, String message) {
    messagingTemplate.convertAndSend(topic, message);
}

public Set<String> getConnectedUsers() {
    return userRegistry.getUsers().stream()
            .map(SimpUser::getName)
            .collect(Collectors.toSet());
}

  2. 定时任务和调度：在WebSocket应用中，你可能需要执行定时任务和调度任务，例如定时发送消息、定时清理会话等。你可以使用Spring框架提供的定时任务调度功能，结合WebSocket框架来实现这些任务。

    @Component
public class WebSocketScheduler {

    @Autowired
    private SimpMessagingTemplate messagingTemplate;

    @Scheduled(fixedDelay = 5000) // 每5秒执行一次
    public void sendScheduledMessage() {
        String message = "Scheduled message";
        messagingTemplate.convertAndSend("/topic/messages", message);
    }
}

  3. 跨域访问控制：如果你的WebSocket应用需要跨域访问控制，你可以配置相应的跨域策略。Spring WebSocket框架提供了 setAllowedOrigins方法来设置允许的跨域来源。

    @Override
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
    registry.addHandler(new MyWebSocketHandler(), "/websocket")
            .setAllowedOrigins("http://example.com")
            .addInterceptors(new HttpSessionHandshakeInterceptor())
            .withSockJS();
}

  4. 安全认证和授权：如果你的WebSocket应用需要安全认证和授权，你可以结合Spring Security框架来实现。通过配置适当的安全规则和认证机制，你可以确保只有经过授权的用户才能建立WebSocket连接和发送消息。

    @Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {

    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(new MyWebSocketHandler(), "/websocket")
                .setAllowedOrigins("*")
                .addInterceptors(new HttpSessionHandshakeInterceptor())
                .withSockJS();
    }

    // ...
}
@Configuration
@EnableWebSecurity
public class SecurityConfig extends WebSecurityConfigurerAdapter {

    @Override
    protected void configure(HttpSecurity http) throws Exception {
        http
            .authorizeRequests()
            .antMatchers("/websocket").authenticated()
            .anyRequest().permitAll()
            .and()
            .formLogin()
            .permitAll()
            .and()
            .logout()
            .permitAll();
    }
}

六、存储、多频道管理与部署

  1. 消息存储和持久化：如果你的应用需要存储和持久化消息，可以结合使用WebSocket和消息队列或数据库来实现。当接收到消息时，你可以将消息存储到数据库或消息队列中，并在需要的时候进行读取和处理。

    @Autowired
private MessageRepository messageRepository;

public void handleMessage(WebSocketSession session, TextMessage message) {
    // 存储消息到数据库或消息队列
    messageRepository.save(message.getPayload());
}

public List<String> getMessages() {
    // 从数据库或消息队列中读取消息
    return messageRepository.findAll();
}

  2. 多频道管理：如果你的应用需要管理多个频道或主题，可以考虑使用WebSocket的订阅和发布模式。你可以创建多个频道或主题，并在客户端订阅感兴趣的频道，以便接收相应的消息。

    @Autowired
private SimpMessagingTemplate messagingTemplate;

public void sendMessage(String channel, String message) {
    messagingTemplate.convertAndSend("/topic/" + channel, message);
}

public void subscribeChannel(String channel, WebSocketSession session) {
    messagingTemplate.subscribe("/topic/" + channel, session.getId());
}

public void unsubscribeChannel(String channel, WebSocketSession session) {
    messagingTemplate.unsubscribe("/topic/" + channel, session.getId());
}

  3. 跨服务器部署：如果你的应用需要在多个服务器上部署，可以考虑使用分布式消息代理来实现跨服务器的消息传递。一种常见的方案是使用RabbitMQ或Apache Kafka等消息队列作为消息代理，以确保消息在不同服务器之间的可靠传递。

  4. 错误处理和异常处理：在处理WebSocket连接和消息时，可能会出现错误和异常。你可以实现 WebSocketHandler接口的 handleTransportError和 handleMessageException方法来处理连接错误和消息异常，并进行相应的处理，例如记录日志、发送错误消息等。
    public class MyWebSocketHandler implements WebSocketHandler {

    public class MyWebSocketHandler implements WebSocketHandler {
 @Override
    public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
        // 处理连接错误
    }

    @Override
    public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception {
        try {
            // 处理接收到的消息
        } catch (Exception e) {
            // 处理消息异常
        }
    }
}

七、会话心跳

  1. 消息编解码：在WebSocket应用中，消息的编码和解码是一个重要的环节。你可以使用Spring提供的消息编解码器来实现消息的转换和处理。例如，可以使用 TextMessageEncoder和 TextMessageDecoder来进行文本消息的编解码。

    public class MyTextMessageEncoder implements MessageEncoder<String> {
    @Override
    public ByteBuffer encode(String message) throws EncodeException {
        // 实现消息的编码逻辑
    }
}

public class MyTextMessageDecoder implements MessageDecoder<String> {
    @Override
    public String decode(ByteBuffer message) throws DecodeException {
        // 实现消息的解码逻辑
    }
}

  2. 心跳检测：为了确保WebSocket连接的稳定性，你可以实现心跳检测机制来监控连接的状态。可以使用定时任务来发送心跳消息，并在一定时间内未收到心跳响应时，断开连接或进行相应的处理。

    @Component
public class HeartbeatScheduler {

    @Autowired
    private SimpMessagingTemplate messagingTemplate;

    @Scheduled(fixedDelay = 10000) // 每10秒发送一次心跳消息
    public void sendHeartbeatMessage() {
        messagingTemplate.convertAndSend("/topic/heartbeat", "Heartbeat");
    }
}

public class MyWebSocketHandler extends TextWebSocketHandler {

    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        if ("Heartbeat".equals(message.getPayload())) {
            // 收到心跳消息，进行相应处理
            // ...
        } else {
            // 处理其他消息
            // ...
        }
    }
}

  3. 自定义拦截器：你可以使用自定义拦截器来对WebSocket连接进行拦截和处理。拦截器可以用于认证、授权、日志记录等方面。通过实现 HandshakeInterceptor接口，你可以在建立连接之前和之后进行相应的操作。

    public class MyHandshakeInterceptor implements 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 exception) {
        // 在建立连接之后进行处理
        // ...
    }
}

  4. 会话管理：在WebSocket应用中，你可能需要管理和跟踪每个连接的会话信息。你可以通过维护一个会话管理器来实现会话的创建、销毁和查询等操作。

    public class SessionManager {
    private Map<String, WebSocketSession> sessions = new ConcurrentHashMap<>();

    public void addSession(WebSocketSession session) {
        sessions.put(session.getId(), session);
    }

    public void removeSession(WebSocketSession session) {
        sessions.remove(session.getId());
    }

    public WebSocketSession getSession(String sessionId) {
        return sessions.get(sessionId);
    }

    public List<WebSocketSession> getAllSessions() {
        return new ArrayList<>(sessions.values());
    }
}
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)

WebSocket

网络协议

【WebSocket】WebSocket使用，看这篇就行 的相关文章

随机推荐

  • MySQL高可用工具heartbeat简介

    MySQL高可用工具heartbeat简介 官网 http www linux ha org wiki Heartbeat 一 HeartBeat的作用 通过HeartBeat 可以将资源 IP以及程序服务等资源 从一台已经故障的计算机快速

  • ETL基础认知

    1 ETL基础认知 了解 问题1 如何将零散的数据 集中输入到数据仓库 ETL E 数据抽取 抽取的是其他数据源中的数据 T 数据转换 将数据转换为统一的格式 消除异常值 缺失值 对于错误的逻辑进行修改 L 数据加载 将不同数据源的数据处理

  • OpenGL GLFW入门篇 - 画矩形

    效果图 主体代码 void DrawRectangle void GLfloat xl yt xr yb w h glPushMatrix glLoadIdentity glTranslatef 0 0 0 0 0 f w 1 2 h 1

  • osgEarth中opengl版本的确定

    osgEarth VirtualProgram if defined OSG GLES2 AVAILABLE define GLSL VERSION 100 define GLSL VERSION STR 100 define GLSL D

  • 聚类算法总结

    最近整理一下聚类相关的东西 数据说明 凸集 在欧氏空间中 凸集是对于集合内的每一对点 连接该对点的直线段上的每个点也在该集合内 非凸 non convex 数据 类比上述可知 距离 相似度 首先我们要了解衡量对象间差异的方法对象可能是一个值

  • 关于带MinGW版本的codeblocks

    MinGW就是Windows移植版的GCC编译器 Codeblocks是IDE 这个软件的特点是可以让你自由选择想要使用的编译器 Code Blocks是一个免费 开源 跨平台的C C IDE 支持Windows Linux MacOSX

  • Boost电路的结构及工作原理

    Boost电路定义 Boost升压电路的英文名称为 theboostconverter 或者叫 step upconverter 是一种开关直流升压电路 它能够将直流电变为另一固定电压或可调电压的直流电 也称为直流 直流变换器 DC DCC

  • COPU陆首群教授应邀在开放原子全球开源峰会上做主旨演讲

    各位领导 各位专家 同志们 朋友们 大家下午好 祝贺开放原子开源基金会首届全球开源峰会成功举办 1970年是为人们称道的UNIX元年 也是开源在全球诞生之日 开源在全球流行至今已有52年了 自从1991年我国引进UNIX现代计算系统以来 中

  • DS内排—直插排序

    目录 题目描述 思路分析 AC代码 题目描述 给定一组数据 使用直插排序完成数据的升序排序 程序要求 若使用C 只能include一个头文件iostream 若使用C语言只能include一个头文件stdio 程序中若include多过一个

  • 在java中重复一个字符串n次的几种方法

    方法一 String format 0 n d 0 replace 0 s 方法二 new String new char n replace 0 s 方法三 JAVA 8 String join Collections nCopies n

  • （三）Unity开发Vision Pro——入门

    3 入门 1 入门 本节涵盖了几个重要主题 可帮助您加快visionOS 平台开发速度 在这里 您将找到构建第一个 Unity PolySpatial XR 应用程序的分步指南的链接 以及 PolySpatial XR 开发时的一些开发最佳

  • 目标检测数据集PASCAL VOC笔记

    PASCAL VOC 数据集的应用领域有Object Classification Object Detection Object Segmentation Human Layout Action Classification等 它的常用版

  • Acwing 116. 飞行员兄弟

    枚举所有开关的状态 0 2 16 1 16位二进制数 若某一位为1表示按一下 为0表示不按 按照该方案 对所有灯泡进行操作 所在行 所在列全部按一下 判断灯泡是否全亮 如果全亮的话 记录方案 include

  • 美团客户端技术团队招人啦

    非广告哈 帮好友发一则招聘 美团客户端团队在北京招人了 性能优化 基础组件相关的岗位都有 在看机会的或者想了解一下的 都可以通过文章最后面的联系方式进行联系 或者私信我 我拉个群你们细聊 想必大家都看过美团技术团队的博客 美团技术团队 1

  • SQL中DML语句（数据操作语言）

    表示数据操作语言 凡是对表当中的数据进行增删改的都是DML 目录 insert 插入数据 update 修改数据 delete 删除数据 insert 插入数据 语法格式 insert into 表名 字段名1 字段名2 字段名3 valu

  • 如何用burpsuite进行攻击

    一 使用Burpsuite进行攻击 1 第一步打开burpsuite 2 第二部点击Repeater 3 第三步点击粉笔形状的按钮 4 输入要攻击目标的ip地址与端口号 5 添加攻击报文 进行攻击 6 查看响应结果 完整界面展示如下 注意

  • 今日头条2017校招（出题数目）

    题目描述 头条的2017校招开始了 为了这次校招 我们组织了一个规模宏大的出题团队 每个出题人都出了一些有趣的题目 而我们现在想把这些题目组合成若干场考试出来 在选题之前 我们对题目进行了盲审 并定出了每道题的难度系数 一场考试包含3道开放

  • 使用共享 MVI 架构实现高效的 Kotlin Multiplatform Mobile (KMM) 开发

    使用共享 MVI 架构实现高效的 Kotlin Multiplatform Mobile KMM 开发 文章中探讨了 Google 提供的应用架构指南在多平台上的实现 通过共享视图模型 View Models 和共享 UI 状态 UI St

  • Python3 面向对象

    文章目录 面向对象基础 类及类的定义 对象的创建 成员变量 定义格式一 常用 定义格式二 不常用 区别 成员方法 定义格式一 定义格式二 init 方法 str 方法 成员方法调成员 成员 方法 调成员 变量 成员 方法 调成员 方法 手机

  • 【WebSocket】WebSocket使用，看这篇就行

    一 WebSocket连接的建立 消息的接收和回复 当涉及到WebSocket框架的深度使用时 一个流行的选择是使用Java的Spring框架来实现 下面是一个基本的示例 演示了如何使用Spring WebSocket框架进行深度使用 首先

热门标签