2025-05-31

Vue 实战:利用 StompJS + WebSocket 实现后端高效通信


目录

  1. 前言
  2. 技术选型与环境准备

    • 2.1 WebSocket 与 STOMP 简介
    • 2.2 项目环境与依赖安装
  3. 后端示例(Spring Boot + WebSocket)

    • 3.1 WebSocket 配置
    • 3.2 STOMP 端点与消息处理
  4. 前端集成 StompJS

    • 4.1 安装 stompjssockjs-client
    • 4.2 封装 WebSocket 服务(stompService.js
  5. Vue 组件实战:消息发布与订阅

    • 5.1 ChatRoom.vue:聊天室主组件
    • 5.2 MessageList.vue:消息列表展示组件
    • 5.3 MessageInput.vue:消息发送组件
  6. 数据流示意图解
  7. 连接管理与断线重连

    • 7.1 连接状态监控
    • 7.2 心跳与自动重连策略
  8. 订阅管理与消息处理
  9. 常见问题与优化建议
  10. 总结

前言

在现代前端开发中,借助 WebSocket 可以实现客户端和服务器的双向实时通信,而 STOMP(Simple Text Oriented Messaging Protocol) 则为消息传递定义了更高层次的约定,便于我们管理主题订阅、消息广播等功能。StompJS 是一款流行的 JavaScript STOMP 客户端库,配合 SockJS 能在浏览器中可靠地与后端进行 WebSocket 通信。

本文将以 Vue 3 为例,演示如何利用 StompJS + WebSocket 实现一个最基础的实时聊天室场景,涵盖后端 Spring Boot+WebSocket 配置、以及 Vue 前端如何封装连接服务、组件化实现消息订阅与发布。通过代码示例流程图解详细说明,帮助你快速掌握实战要点。


技术选型与环境准备

2.1 WebSocket 与 STOMP 简介

  • WebSocket:在单个 TCP 连接上实现双向通信协议,浏览器与服务器通过 ws://wss:// 建立连接,可实时收发消息。
  • STOMP:类似于 HTTP 的文本协议,定义了具体的消息头部格式、订阅机制,并在 WebSocket 之上构建消息队列与主题订阅功能。
  • SockJS:浏览器端的 WebSocket 兼容库,会在浏览器不支持原生 WebSocket 时自动退回到 xhr-streaming、xhr-polling 等模拟方式。
  • StompJS:基于 STOMP 协议的 JavaScript 客户端,实现了发送、订阅、心跳等功能。

使用 StompJS + SockJS,前端可以调用类似:

const socket = new SockJS('/ws-endpoint');
const client = Stomp.over(socket);
client.connect({}, () => {
  client.subscribe('/topic/chat', message => {
    console.log('收到消息:', message.body);
  });
  client.send('/app/chat', {}, JSON.stringify({ user: 'Alice', content: 'Hello' }));
});

2.2 项目环境与依赖安装

2.2.1 后端(Spring Boot)

  • JDK 8+
  • Spring Boot 2.x
  • 添加依赖:

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

2.2.2 前端(Vue 3)

# 1. 初始化 Vue 3 项目(使用 Vite 或 Vue CLI)
npm create vite@latest vue-stomp-chat -- --template vue
cd vue-stomp-chat
npm install

# 2. 安装 StompJS 与 SockJS 依赖
npm install stompjs sockjs-client --save

目录结构示例:

vue-stomp-chat/
├─ src/
│  ├─ services/
│  │   └─ stompService.js
│  ├─ components/
│  │   ├─ ChatRoom.vue
│  │   ├─ MessageList.vue
│  │   └─ MessageInput.vue
│  ├─ App.vue
│  └─ main.js
├─ package.json
└─ ...

后端示例(Spring Boot + WebSocket)

为了让前后端完整配合,后端先示例一个简单的 Spring Boot WebSocket 设置,提供一个 STOMP 端点以及消息广播逻辑。

3.1 WebSocket 配置

// src/main/java/com/example/config/WebSocketConfig.java
package com.example.config;

import org.springframework.context.annotation.Configuration;
import org.springframework.messaging.simp.config.MessageBrokerRegistry;
import org.springframework.web.socket.config.annotation.*;

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {

    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        // 前端将通过 /ws-endpoint 来建立 SockJS 连接
        registry.addEndpoint("/ws-endpoint").setAllowedOrigins("*").withSockJS();
    }

    @Override
    public void configureMessageBroker(MessageBrokerRegistry registry) {
        // 应用层消息前缀,前端发送到 /app/**
        registry.setApplicationDestinationPrefixes("/app");
        // 启用简单内存消息代理,广播到 /topic/**
        registry.enableSimpleBroker("/topic");
    }
}
  • registerStompEndpoints:注册一个 /ws-endpoint STOMP 端点,启用 SockJS。
  • setApplicationDestinationPrefixes("/app"):前端发送到 /app/... 的消息将被标记为需路由到 @MessageMapping 处理。
  • enableSimpleBroker("/topic"):启用基于内存的消息代理,向所有订阅了 /topic/... 的客户端广播消息。

3.2 STOMP 端点与消息处理

// src/main/java/com/example/controller/ChatController.java
package com.example.controller;

import com.example.model.ChatMessage;
import org.springframework.messaging.handler.annotation.MessageMapping;
import org.springframework.messaging.handler.annotation.SendTo;
import org.springframework.stereotype.Controller;

@Controller
public class ChatController {

    // 当前端发送消息到 /app/chat 时,此方法被调用
    @MessageMapping("/chat")
    @SendTo("/topic/messages")
    public ChatMessage handleChatMessage(ChatMessage message) {
        // 可以在此做保存、过滤、存储等逻辑
        return message;  // 返回的内容会被广播到 /topic/messages
    }
}
// src/main/java/com/example/model/ChatMessage.java
package com.example.model;

public class ChatMessage {
    private String user;
    private String content;
    // 构造、getter/setter 省略
}
  • 客户端发送到 /app/chat 的消息,Spring 会调用 handleChatMessage,并将其广播到所有订阅了 /topic/messages 的客户端。

前端集成 StompJS

4.1 安装 stompjssockjs-client

npm install stompjs sockjs-client --save

stompjs 包含 Stomp 客户端核心;sockjs-client 用于兼容不同浏览器。


4.2 封装 WebSocket 服务(stompService.js

src/services/stompService.js 中统一管理 STOMP 连接、订阅、发送、断开等逻辑:

// src/services/stompService.js

import { Client } from 'stompjs';
import SockJS from 'sockjs-client';

// STOMP 客户端实例
let stompClient = null;

// 订阅回调列表:{ topic: [ callback, ... ] }
const subscriptions = new Map();

/**
 * 初始化并连接 STOMP
 * @param {Function} onConnect - 连接成功回调
 * @param {Function} onError - 连接出错回调
 */
export function connect(onConnect, onError) {
  // 如果已存在连接,直接返回
  if (stompClient && stompClient.connected) {
    onConnect();
    return;
  }
  const socket = new SockJS('/ws-endpoint'); // 与后端 registerStompEndpoints 对应
  stompClient = Client.over(socket);
  stompClient.debug = null; // 关闭日志打印,生产可去除

  // 连接时 configuration(headers 可写 token 等)
  stompClient.connect(
    {}, 
    () => {
      console.log('[Stomp] Connected');
      onConnect && onConnect();
      // 断线重连后自动恢复订阅
      resubscribeAll();
    },
    (error) => {
      console.error('[Stomp] Error:', error);
      onError && onError(error);
    }
  );
}

/**
 * 断开 STOMP 连接
 */
export function disconnect() {
  if (stompClient) {
    stompClient.disconnect(() => {
      console.log('[Stomp] Disconnected');
    });
    stompClient = null;
    subscriptions.clear();
  }
}

/**
 * 订阅某个 topic
 * @param {String} topic - 形如 '/topic/messages'
 * @param {Function} callback - 收到消息的回调,参数 message.body
 */
export function subscribe(topic, callback) {
  if (!stompClient || !stompClient.connected) {
    console.warn('[Stomp] 订阅前请先 connect()');
    return;
  }
  // 第一次该 topic 订阅时先订阅 STOMP
  if (!subscriptions.has(topic)) {
    const subscription = stompClient.subscribe(topic, (message) => {
      try {
        const body = JSON.parse(message.body);
        callback(body);
      } catch (e) {
        console.error('[Stomp] JSON 解析错误', e);
      }
    });
    subscriptions.set(topic, { callbacks: [callback], subscription });
  } else {
    // 已经存在订阅,仅追加回调
    subscriptions.get(topic).callbacks.push(callback);
  }
}

/**
 * 取消对 topic 的某个回调订阅
 * @param {String} topic
 * @param {Function} callback
 */
export function unsubscribe(topic, callback) {
  if (!subscriptions.has(topic)) return;
  const entry = subscriptions.get(topic);
  entry.callbacks = entry.callbacks.filter((cb) => cb !== callback);
  // 如果回调数组为空,则取消 STOMP 订阅
  if (entry.callbacks.length === 0) {
    entry.subscription.unsubscribe();
    subscriptions.delete(topic);
  }
}

/**
 * 发送消息到后端
 * @param {String} destination - 形如 '/app/chat'
 * @param {Object} payload - JS 对象,会被序列化为 JSON
 */
export function send(destination, payload) {
  if (!stompClient || !stompClient.connected) {
    console.warn('[Stomp] 未连接,无法发送消息');
    return;
  }
  stompClient.send(destination, {}, JSON.stringify(payload));
}

/**
 * 重连后恢复所有订阅
 */
function resubscribeAll() {
  for (const [topic, entry] of subscriptions.entries()) {
    // 重新订阅 STOMP,回调列表保持不变
    const subscription = stompClient.subscribe(topic, (message) => {
      const body = JSON.parse(message.body);
      entry.callbacks.forEach((cb) => cb(body));
    });
    entry.subscription = subscription;
  }
}

说明

  1. connect:建立 SockJS+Stomp 连接,onConnect 回调中可开始订阅。
  2. subscriptions:使用 Map<topic, { callbacks: [], subscription: StompSubscription }> 管理同一 topic 下的多个回调,避免重复调用 stompClient.subscribe(topic)
  3. 断线重连:当后端重启或网络断开后 Stomp 会触发 stompClient.error,可在页面中捕捉并 connect() 重新连接,resubscribeAll() 保证恢复所有订阅。
  4. send:发送至后端的 destination 必须与后端 @MessageMapping 配置对应。

Vue 组件实战:消息发布与订阅

基于已封装好的 stompService.js,下面实现一个最基础的聊天样例,由三部分组件组成:

  1. ChatRoom.vue:负责整体布局,连接/断开、登录、展示当前状态;
  2. MessageList.vue:展示从后端 /topic/messages 接收到的消息列表;
  3. MessageInput.vue:提供输入框,发送消息到 /app/chat

5.1 ChatRoom.vue:聊天室主组件

<!-- src/components/ChatRoom.vue -->
<template>
  <div class="chat-room">
    <div class="header">
      <span v-if="!loggedIn">未登录</span>
      <span v-else>用户:{{ username }}</span>
      <button v-if="!loggedIn" @click="login">登录</button>
      <button v-else @click="logout">退出</button>
      <span class="status">状态:{{ connectionStatus }}</span>
    </div>
    <MessageList v-if="loggedIn" :messages="messages" />
    <MessageInput v-if="loggedIn" @sendMessage="handleSendMessage" />
  </div>
</template>

<script setup>
import { ref, computed, onUnmounted } from 'vue';
import * as stompService from '@/services/stompService';
import MessageList from './MessageList.vue';
import MessageInput from './MessageInput.vue';

// 本地状态
const loggedIn = ref(false);
const username = ref('');
const messages = ref([]);

// 订阅回调:将接收到的消息推入列表
function onMessageReceived(msg) {
  messages.value.push(msg);
}

// 连接状态
const connectionStatus = computed(() => {
  if (!stompService.stompClient) return 'DISCONNECTED';
  return stompService.stompClient.connected ? 'CONNECTED' : 'CONNECTING';
});

// 登录:弹出 prompt 输入用户名,connect 后订阅
function login() {
  const name = prompt('请输入用户名:', '用户_' + Date.now());
  if (!name) return;
  username.value = name;
  stompService.connect(
    () => {
      // 连接成功后订阅 /topic/messages
      stompService.subscribe('/topic/messages', onMessageReceived);
      // 发送加入消息
      stompService.send('/app/chat', { user: name, content: `${name} 加入了聊天室` });
      loggedIn.value = true;
    },
    (error) => {
      console.error('连接失败:', error);
    }
  );
}

// 退出:发送离开消息,取消订阅并断开
function logout() {
  stompService.send('/app/chat', { user: username.value, content: `${username.value} 离开了聊天室` });
  stompService.unsubscribe('/topic/messages', onMessageReceived);
  stompService.disconnect();
  loggedIn.value = false;
  username.value = '';
  messages.value = [];
}

// 发送消息:由子组件触发
function handleSendMessage(content) {
  const msg = { user: username.value, content };
  // 本地回显
  messages.value.push(msg);
  // 发送给后端
  stompService.send('/app/chat', msg);
}

// 组件卸载时若登录则退出
onUnmounted(() => {
  if (loggedIn.value) {
    logout();
  }
});
</script>

<style scoped>
.chat-room {
  max-width: 600px;
  margin: 0 auto;
  padding: 16px;
}
.header {
  display: flex;
  align-items: center;
  gap: 12px;
  margin-bottom: 12px;
}
.status {
  margin-left: auto;
  font-weight: bold;
}
button {
  padding: 4px 12px;
  background: #409eff;
  color: white;
  border: none;
  border-radius: 4px;
  cursor: pointer;
}
button:hover {
  background: #66b1ff;
}
</style>

说明:

  • login() 中调用 stompService.connect 并在 onConnect 回调里订阅 /topic/messages,并发送一条 “加入” 通知。
  • MessageListMessageInput 仅在 loggedIntrue 时渲染。
  • onMessageReceived 作为订阅回调,将接收到的消息追加到 messages 数组。
  • 退出时先发送“离开”通知,再 unsubscribedisconnect

5.2 MessageList.vue:消息列表展示组件

<!-- src/components/MessageList.vue -->
<template>
  <div class="message-list" ref="listRef">
    <div v-for="(msg, idx) in messages" :key="idx" class="message-item">
      <span class="user">{{ msg.user }}:</span>
      <span class="content">{{ msg.content }}</span>
    </div>
  </div>
</template>

<script setup>
import { watch, nextTick, ref } from 'vue';

const props = defineProps({
  messages: {
    type: Array,
    default: () => []
  }
});

const listRef = ref(null);

// 监听 messages 变化后自动滚到底部
watch(
  () => props.messages.length,
  async () => {
    await nextTick();
    const el = listRef.value;
    if (el) {
      el.scrollTop = el.scrollHeight;
    }
  }
);
</script>

<style scoped>
.message-list {
  height: 300px;
  overflow-y: auto;
  border: 1px solid #ccc;
  padding: 8px;
  background: #fafafa;
}
.message-item {
  margin-bottom: 6px;
}
.user {
  font-weight: bold;
  margin-right: 4px;
}
</style>

说明:

  • 通过 props.messages 渲染聊天记录;
  • 使用 watch 监听 messages.length,每次新消息到来后 scrollTop = scrollHeight 自动滚到底部。

5.3 MessageInput.vue:消息发送组件

<!-- src/components/MessageInput.vue -->
<template>
  <div class="message-input">
    <input
      v-model="inputText"
      placeholder="输入消息,按回车发送"
      @keydown.enter.prevent="send"
    />
    <button @click="send">发送</button>
  </div>
</template>

<script setup>
import { ref } from 'vue';

const emit = defineEmits(['sendMessage']);
const inputText = ref('');

// 触发父组件的 sendMessage 事件
function send() {
  const text = inputText.value.trim();
  if (!text) return;
  emit('sendMessage', text);
  inputText.value = '';
}
</script>

<style scoped>
.message-input {
  display: flex;
  margin-top: 12px;
}
.message-input input {
  flex: 1;
  padding: 6px;
  font-size: 14px;
  border: 1px solid #ccc;
  border-radius: 4px;
}
.message-input button {
  margin-left: 8px;
  padding: 6px 12px;
  background: #67c23a;
  color: white;
  border: none;
  cursor: pointer;
  border-radius: 4px;
}
button:hover {
  background: #85ce61;
}
</style>

说明:

  • inputText 绑定输入框内容,按回车或点击 “发送” 后,通过 emit('sendMessage', text) 通知父组件 ChatRoom 发送。

数据流示意图解

为帮助理解前后端通信流程,下面用 ASCII 图展示一次完整的“发送 → 广播 → 接收”过程:

┌────────────────────────────────────────────────┐
│                客户端(浏览器)                 │
│                                                │
│  ChatRoom.vue                                 │
│    ├── login() → stompService.connect()        │
│    │        └────────────┐                     │
│    ├── stompClient.connect() (STOMP handshake) │
│    │        └───────────→│                     │
│                                                │
│  MessageInput.vue                              │
│    └─ 用户输入 “Hello” → sendMessage(“Hello”)  │
│                ↓                               │
│       stompService.send('/app/chat', {user,content})   │
│                ↓                               │
└────────────────── WebSocket ────────────────────┘
           (STOMP 格式) ─▶
┌────────────────────────────────────────────────┐
│                后端(Spring Boot)              │
│                                                │
│  WebSocketConfig 注册 /ws-endpoint            │
│  STOMP 协议升级完成                          │
│    └─ 触发 ChatController.handleChatMessage()  │
│         收到 { user, content }                 │
│    └─ 返回该对象并通过 broker 广播到 /topic/messages │
│                                                │
└────────────────── WebSocket ────────────────────┘
      ◀─ (STOMP /topic/messages) “Hello” ────────
┌────────────────────────────────────────────────┐
│              客户端(浏览器)                  │
│                                                │
│  stompService.onMessage('/topic/messages')     │
│    └─ 调用 onMessageReceived({user, content})  │
│           ↓                                    │
│  ChatRoom.vue: messages.push({user, content})  │
│           ↓                                    │
│  MessageList 渲染新消息                         │
└────────────────────────────────────────────────┘
  • 阶段 1:客户端 send('/app/chat', payload)
  • 阶段 2:后端 /app/chat 路由触发 @MessageMapping("/chat") 方法,将消息广播到 /topic/messages
  • 阶段 3:客户端订阅 /topic/messages,收到 “Hello” 并更新视图

连接管理与断线重连

7.1 连接状态监控

stompService.js 中,我们并未实现自动重连。可以在客户端检测到连接丢失后,自动尝试重连。示例改造:

// 在 stompService.js 中增加

let reconnectAttempts = 0;
const MAX_RECONNECT_ATTEMPTS = 5;
const RECONNECT_DELAY = 5000; // 毫秒

export function connect(onConnect, onError) {
  const socket = new SockJS('/ws-endpoint');
  stompClient = Client.over(socket);
  stompClient.debug = null;

  stompClient.connect(
    {},
    () => {
      console.log('[Stomp] Connected');
      reconnectAttempts = 0;
      onConnect && onConnect();
      resubscribeAll();
    },
    (error) => {
      console.error('[Stomp] 连接出错', error);
      if (reconnectAttempts < MAX_RECONNECT_ATTEMPTS) {
        reconnectAttempts++;
        console.log(`[Stomp] 第 ${reconnectAttempts} 次重连将在 ${RECONNECT_DELAY}ms 后尝试`);
        setTimeout(() => connect(onConnect, onError), RECONNECT_DELAY);
      } else {
        onError && onError(error);
      }
    }
  );
}

要点

  • 每次 connect 出错后,如果重连次数小于 MAX_RECONNECT_ATTEMPTS,则 setTimeout 延迟后再次调用 connect()
  • 成功连接后,重置 reconnectAttempts = 0,并在回调中恢复订阅。

7.2 心跳与自动重连策略

StompJS 库支持心跳检测,可在 connect 时传入心跳参数:

stompClient.connect(
  { heartbeat: '10000,10000' }, // “发送心跳间隔,接收心跳间隔” 单位 ms
  onConnect,
  onError
);
  • 第一位(10000):10 秒后向服务器发送心跳;
  • 第二位(10000):若 10 秒内未接收到服务器心跳,则认为连接失效;

确保服务器端也启用了心跳,否则客户端会自动断开。


订阅管理与消息处理

8.1 主题订阅与回调分发

stompService.js 中,通过 subscriptions Map 维护多个订阅回调。例如多个组件都需要监听 /topic/notifications,只会注册一次 STOMP 订阅,并在收到消息后依次调用各自回调函数:

// 第一次 subscribe('/topic/notifications', cb1) 时,
stompClient.subscribe('/topic/notifications', ...);
subscriptions.set('/topic/notifications', { callbacks: [cb1], subscription });

// 再次 subscribe('/topic/notifications', cb2) 时,
subscriptions.get('/topic/notifications').callbacks.push(cb2);
// 不会重复调用 stompClient.subscribe()

当需要取消时,可按回调移除:

unsubscribe('/topic/notifications', cb1);

如无回调剩余,则真正调用 subscription.unsubscribe() 取消 STOMP 订阅。


常见问题与优化建议

  1. 消息体过大导致性能瓶颈

    • 若消息 JSON 包含大量字段,可考虑仅传输必要数据,或对二进制数据做 Base64/压缩处理;
  2. 批量消息/快速涌入

    • 若服务器在短时间内向客户端推送大量消息,前端渲染可能卡顿。可对渲染做防抖节流处理,亦可分页加载消息;
  3. 多浏览器兼容

    • stompjs + sockjs-client 在大多数现代浏览器都能正常工作。若需要支持 IE9 以下,请额外引入 setTimeout polyfill,并确保服务器端 SockJS 配置兼容;
  4. 安全与权限校验

    • 建议在 HTTP 握手阶段或 STOMP header 中带上授权 Token,后端在 WebSocketConfig 中做 HandshakeInterceptor 验证;
  5. Session 粘性与跨集群

    • 若后端部署在多实例集群,需确保 WebSocket 连接的 Session 粘性或使用共享消息代理(如 RabbitMQ、ActiveMQ)做集群消息广播;

总结

本文从理论与实践两方面讲解了如何在 Vue 3 项目中,利用 StompJS + WebSocket 与后端进行高效实时通信。主要内容包括:

  1. 后端(Spring Boot + WebSocket)配置:注册 STOMP 端点 /ws-endpoint,设置消息前缀与订阅代理;
  2. StompJS 封装:在 stompService.js 中集中管理连接、订阅、发送、重连、心跳,避免多个组件各自管理连接;
  3. Vue 组件化实战ChatRoom.vue 负责登录/订阅与断开,MessageList.vue 实现实时渲染,MessageInput.vue 实现发布;
  4. 数据流示意:从 send('/app/chat') 到后端 @MessageMapping 再广播到 /topic/messages,最后前端接收并更新视图的完整流程;
  5. 断线重连与心跳:在 connect() 中加入心跳配置与自动重连逻辑,提高连接稳定性;
  6. 订阅管理:使用 Map<topic, callbacks[]> 防止重复订阅,并在重连后恢复;
  7. 常见问题及优化建议:包括批量消息渲染、消息体过大、跨集群 Session 粘性、安全校验等注意事项。

通过此方案,你可以快速为 Vue 应用接入后端实时通信功能,搭建简单的聊天室、通知系统、实时数据推送等场景。希望本文的代码示例图解说明能让你更容易上手,掌握 StompJS + WebSocket 的实战应用。

2025-05-31

Vue 中高效使用 WebSocket 的实战教程


目录

  1. 前言
  2. WebSocket 基础回顾

    • 2.1 什么是 WebSocket
    • 2.2 与 HTTP 长轮询对比
  3. 高效使用的必要性

    • 3.1 常见问题与挑战
    • 3.2 心跳、重连、订阅管理
  4. Vue 项目初始化与依赖准备

    • 4.1 创建 Vue 3 + Vite 项目
    • 4.2 安装 Pinia(或 Vuex,可选)
  5. WebSocket 服务封装:websocketService.js

    • 5.1 单例模式设计
    • 5.2 连接、断开、重连与心跳
    • 5.3 事件订阅与广播机制
  6. Vue 组件实战示例:简易实时聊天室

    • 6.1 目录结构与组件划分
    • 6.2 ChatManager 组件:管理连接与会话
    • 6.3 MessageList 组件:显示消息列表
    • 6.4 MessageInput 组件:输入并发送消息
    • 6.5 数据流动图解
  7. 多组件/模块共享同一个 WebSocket 连接

    • 7.1 在 Pinia 中维护状态
    • 7.2 通过 provide/inject 或全局属性共享实例
  8. 心跳检测与自动重连实现

    • 8.1 心跳机制原理
    • 8.2 在 websocketService 中集成心跳与重连
    • 8.3 防抖/节流发送心跳
  9. 性能优化与最佳实践

    • 9.1 减少重复消息订阅
    • 9.2 批量发送与分包
    • 9.3 合理关闭与清理监听器
  10. 常见问题与调试技巧
  11. 总结

前言

在现代 Web 应用中,实时通信 已经成为许多场景的核心需求:聊天室、协同协作、股票行情更新、IoT 设备状态监控等。与传统的 HTTP 轮询相比,WebSocket 提供了一个持久化、双向的 TCP 连接,使得客户端和服务器可以随时互相推送消息。对 Vue 开发者而言,如何在项目中高效、稳定地使用 WebSocket,是一门必备技能。

本篇教程将带你从零开始,基于 Vue 3 + Composition API,配合 Pinia(或 Vuex),一步步实现一个“简易实时聊天室”的完整示例,涵盖:

  • WebSocket 服务的单例封装
  • 心跳检测自动重连机制
  • 多组件/模块共享同一连接
  • 批量发送与防抖/节流
  • 常见坑与调试技巧

通过代码示例、ASCII 图解与详细说明,让你快速掌握在 Vue 中高效使用 WebSocket 的核心要点。


WebSocket 基础回顾

2.1 什么是 WebSocket

WebSocket 是一种在单个 TCP 连接上进行全双工(bidirectional)、实时通信的协议。其典型握手流程如下:

Client                                   Server
  |    HTTP Upgrade Request              |
  |------------------------------------->|
  |    HTTP 101 Switching Protocols      |
  |<-------------------------------------|
  |                                      |
  |<========== WebSocket 双向数据 ==========>|
  |                                      |
  • 客户端发起一个 HTTP 请求,头部包含 Upgrade: websocket,请求将协议升级为 WebSocket。
  • 服务器返回 101 Switching Protocols,确认切换。
  • 从此,客户端与服务器通过同一个连接可以随时互发消息,无需每次都重新建立 TCP 连接。

2.2 与 HTTP 长轮询对比

特性HTTP 轮询WebSocket
建立连接每次请求都要新建 TCP 连接单次握手后复用同一 TCP 连接
消息开销请求/响应头大,开销高握手后头部极小,帧格式精简
双向通信只能客户端发起请求客户端/服务器都可随时推送
延迟取决于轮询间隔近乎“零”延迟
适用场景简单场景,实时性要求不高聊天、游戏、协同编辑、高频数据

高效使用的必要性

3.1 常见问题与挑战

在实际项目中,直接在组件里这样写会遇到很多问题:

// 简易示例:放在组件内
const ws = new WebSocket('wss://example.com/socket');
ws.onopen = () => console.log('已连接');
ws.onmessage = (ev) => console.log('收到消息', ev.data);
ws.onerror = (err) => console.error(err);
ws.onclose = () => console.log('已关闭');

// 在组件卸载时要调用 ws.close()
// 如果有多个组件,需要避免重复创建多个连接

挑战包括:

  1. 重复连接:若多个组件各自创建 WebSocket,会造成端口被占用、服务器负载过高,且消息处理分散。
  2. 意外断开:网络波动、服务器重启会导致连接断开,需要自动重连
  3. 心跳检测:服务器通常需要客户端周期性发送“心跳”消息以断定客户端是否在线,客户端也要监测服务器响应。
  4. 消息订阅管理:某些频道消息只需订阅一次,避免重复订阅导致“重复消息”
  5. 资源清理:组件卸载时,需解绑事件、关闭连接,防止内存泄漏和无效消息监听。

3.2 心跳、重连、订阅管理

为了满足以上需求,我们需要将 WebSocket 的逻辑进行集中封装,做到:

  • 全局单例:整个应用只维护一个 WebSocket 实例,消息通过发布/订阅或状态管理分发给各组件。
  • 心跳机制:定期(如 30 秒)向服务器发送“ping”消息,若一定时间内未收到“pong” 或服务端响应,认为断线并触发重连。
  • 自动重连:连接断开后,间隔(如 5 秒)重试,直到成功。
  • 订阅管理:对于需要向服务器“订阅频道”的场景,可在封装里记录已订阅频道列表,断线重连后自动恢复订阅。
  • 错误处理与退避策略:若短时间内多次重连失败,可采用指数退避,避免服务器或网络被压垮。

Vue 项目初始化与依赖准备

4.1 创建 Vue 3 + Vite 项目

# 1. 初始化项目
npm create vite@latest vue-websocket -- --template vue
cd vue-websocket
npm install

# 2. 安装 Pinia(状态管理,可选)
npm install pinia --save

# 3. 运行开发
npm run dev

项目结构:

vue-websocket/
├─ public/
├─ src/
│  ├─ assets/
│  ├─ components/
│  ├─ services/
│  │   └─ websocketService.js
│  ├─ store/
│  ├─ App.vue
│  └─ main.js
└─ package.json

4.2 安装 Pinia(或 Vuex,可选)

本教程示范用 Pinia 存放“连接状态”、“消息列表”等全局数据,当然也可以用 Vuex。

npm install pinia --save

main.js 中引入并挂载:

// src/main.js
import { createApp } from 'vue';
import { createPinia } from 'pinia';
import App from './App.vue';

const app = createApp(App);
const pinia = createPinia();
app.use(pinia);
app.mount('#app');

WebSocket 服务封装:websocketService.js

关键思路:通过一个独立的模块,集中管理 WebSocket 连接、事件订阅、心跳、重连、消息分发等逻辑,供各组件或 Store 调用。

5.1 单例模式设计

// src/services/websocketService.js

import { ref } from 'vue';

/**
 * WebSocket 服务单例
 * - 负责创建连接、断开、重连、心跳
 * - 支持消息订阅/广播
 */
class WebSocketService {
  constructor() {
    // 1. WebSocket 实例
    this.ws = null;

    // 2. 连接状态
    this.status = ref('CLOSED'); // 'CONNECTING' / 'OPEN' / 'CLOSING' / 'CLOSED'

    // 3. 心跳与重连机制
    this.heartBeatTimer = null;
    this.reconnectTimer = null;
    this.reconnectInterval = 5000;    // 重连间隔
    this.heartBeatInterval = 30000;   // 心跳间隔

    // 4. 已订阅频道记录(可选)
    this.subscriptions = new Set();

    // 5. 回调映射:topic => [callback, ...]
    this.listeners = new Map();
  }

  /** 初始化并连接 */
  connect(url) {
    if (this.ws && this.status.value === 'OPEN') return;
    this.status.value = 'CONNECTING';

    this.ws = new WebSocket(url);

    this.ws.onopen = () => {
      console.log('[WebSocket] 已连接');
      this.status.value = 'OPEN';
      this.startHeartBeat();
      this.resubscribeAll(); // 断线重连后恢复订阅
    };

    this.ws.onmessage = (event) => {
      this.handleMessage(event.data);
    };

    this.ws.onerror = (err) => {
      console.error('[WebSocket] 错误:', err);
    };

    this.ws.onclose = (ev) => {
      console.warn('[WebSocket] 已关闭,原因:', ev.reason);
      this.status.value = 'CLOSED';
      this.stopHeartBeat();
      this.tryReconnect(url);
    };
  }

  /** 关闭连接 */
  disconnect() {
    if (this.ws) {
      this.status.value = 'CLOSING';
      this.ws.close();
    }
    this.clearTimers();
    this.status.value = 'CLOSED';
  }

  /** 发送消息(自动包裹 topic 与 payload) */
  send(topic, payload) {
    if (this.status.value !== 'OPEN') {
      console.warn('[WebSocket] 连接未打开,无法发送消息');
      return;
    }
    const message = JSON.stringify({ topic, payload });
    this.ws.send(message);
  }

  /** 订阅指定 topic  */
  subscribe(topic, callback) {
    if (!this.listeners.has(topic)) {
      this.listeners.set(topic, []);
      // 首次订阅时向服务器发送订阅指令
      this.send('SUBSCRIBE', { topic });
      this.subscriptions.add(topic);
    }
    this.listeners.get(topic).push(callback);
  }

  /** 取消订阅 */
  unsubscribe(topic, callback) {
    if (!this.listeners.has(topic)) return;
    const arr = this.listeners.get(topic).filter(cb => cb !== callback);
    if (arr.length > 0) {
      this.listeners.set(topic, arr);
    } else {
      this.listeners.delete(topic);
      // 向服务器发送取消订阅
      this.send('UNSUBSCRIBE', { topic });
      this.subscriptions.delete(topic);
    }
  }

  /** 断线重连后,恢复所有已订阅的 topic */
  resubscribeAll() {
    for (const topic of this.subscriptions) {
      this.send('SUBSCRIBE', { topic });
    }
  }

  /** 处理收到的原始消息 */
  handleMessage(raw) {
    let msg;
    try {
      msg = JSON.parse(raw);
    } catch (e) {
      console.warn('[WebSocket] 无法解析消息:', raw);
      return;
    }
    const { topic, payload } = msg;
    if (topic === 'HEARTBEAT') {
      // 收到心跳响应(pong)
      return;
    }
    // 如果有对应 topic 的监听者,分发回调
    if (this.listeners.has(topic)) {
      for (const cb of this.listeners.get(topic)) {
        cb(payload);
      }
    }
  }

  /** 启动心跳:定时发送 PING */
  startHeartBeat() {
    this.heartBeatTimer && clearInterval(this.heartBeatTimer);
    this.heartBeatTimer = setInterval(() => {
      if (this.status.value === 'OPEN') {
        this.send('PING', { ts: Date.now() });
      }
    }, this.heartBeatInterval);
  }

  /** 停止心跳 */
  stopHeartBeat() {
    this.heartBeatTimer && clearInterval(this.heartBeatTimer);
    this.heartBeatTimer = null;
  }

  /** 自动重连 */
  tryReconnect(url) {
    if (this.reconnectTimer) return;
    this.reconnectTimer = setInterval(() => {
      console.log('[WebSocket] 尝试重连...');
      this.connect(url);
      // 若成功连接后清除定时器
      if (this.status.value === 'OPEN') {
        clearInterval(this.reconnectTimer);
        this.reconnectTimer = null;
      }
    }, this.reconnectInterval);
  }

  /** 清除所有定时器 */
  clearTimers() {
    this.stopHeartBeat();
    this.reconnectTimer && clearInterval(this.reconnectTimer);
    this.reconnectTimer = null;
  }
}

// 导出单例
export default new WebSocketService();

5.2 关键说明

  1. status:用 ref 将 WebSocket 状态(CONNECTING/OPEN/CLOSED 等)暴露给 Vue 组件,可用于界面显示或禁用按钮。
  2. listeners:维护一个 Map,将不同的 topic(频道)映射到回调数组,方便多处订阅、取消订阅。
  3. 心跳机制:定时发送 { topic: 'PING', payload: { ts } },服务器应在收到后回显 { topic: 'HEARTBEAT' },单纯返回。例如:

    // 服务器伪代码
    ws.on('message', (msg) => {
      const { topic, payload } = JSON.parse(msg);
      if (topic === 'PING') {
        ws.send(JSON.stringify({ topic: 'HEARTBEAT' }));
      }
      // 其它逻辑…
    });
  4. 重连策略:当 onclose 触发时,开启定时器周期重连;如果连接成功,清除重连定时器。简单而有效。
  5. 恢复订阅:在重连后重发所有 SUBSCRIBE 指令,确保服务器推送相关消息。

Vue 组件实战示例:简易实时聊天室

下面我们基于上述 websocketService,搭建一个“简易实时聊天室”示例,展示如何在多个组件间高效分发消息。

6.1 目录结构与组件划分

src/
├─ components/
│  ├─ ChatManager.vue    # 负责打开/关闭 WebSocket、登录与管理会话
│  ├─ MessageList.vue    # 实时显示收到的聊天消息
│  └─ MessageInput.vue   # 输入并发送聊天消息
├─ services/
│  └─ websocketService.js
└─ store/
   └─ chatStore.js       # Pinia

6.2 ChatManager 组件:管理连接与会话

负责:登录(获取用户名)、连接/断开 WebSocket、订阅“chat”频道,将收到的消息写入全局 Store。

<template>
  <div class="chat-manager">
    <div v-if="!loggedIn" class="login">
      <input v-model="username" placeholder="输入用户名" />
      <button @click="login">登录并连接</button>
    </div>
    <div v-else class="controls">
      <span>当前用户:{{ username }}</span>
      <span class="status">状态:{{ status }}</span>
      <button @click="logout">退出并断开</button>
    </div>
  </div>
</template>

<script setup>
import { ref, computed, onBeforeUnmount } from 'vue';
import { useChatStore } from '@/store/chatStore';
import websocketService from '@/services/websocketService';

// Pinia Store 管理全局消息列表与用户状态
const chatStore = useChatStore();

// 本地状态:是否已登录
const loggedIn = ref(false);
const username = ref('');

// 连接状态映射显示
const status = computed(() => websocketService.status.value);

// 登录函数:连接 WebSocket 并订阅聊天频道
async function login() {
  if (!username.value.trim()) return alert('请输入用户名');
  chatStore.setUser(username.value.trim());

  try {
    // 1. 连接 WebSocket
    await websocketService.connect('wss://example.com/chat');
    // 2. 订阅 “chat” 频道
    websocketService.subscribe('chat', (payload) => {
      chatStore.addMessage(payload);
    });
    // 3. 向服务器发送“用户加入”系统消息
    websocketService.send('chat', { user: username.value, text: `${username.value} 加入了聊天室` });

    loggedIn.value = true;
  } catch (err) {
    console.error('连接失败:', err);
    alert('连接失败,请重试');
  }
}

// 注销并断开
async function logout() {
  // 发送退出消息
  websocketService.send('chat', { user: username.value, text: `${username.value} 离开了聊天室` });
  // 取消订阅并断开
  websocketService.unsubscribe('chat', chatStore.addMessage);
  await websocketService.disconnect();
  chatStore.clearMessages();
  loggedIn.value = false;
  username.value = '';
}

onBeforeUnmount(() => {
  if (loggedIn.value) {
    logout();
  }
});
</script>

<style scoped>
.chat-manager {
  margin-bottom: 16px;
}
.login input {
  padding: 6px;
  margin-right: 8px;
}
.status {
  margin: 0 12px;
  font-weight: bold;
}
button {
  padding: 6px 12px;
  background: #409eff;
  border: none;
  border-radius: 4px;
  color: white;
  cursor: pointer;
}
button:hover {
  background: #66b1ff;
}
</style>

6.2.1 关键说明

  1. 登录后:调用 websocketService.connect('wss://example.com/chat'),然后 subscribe('chat', callback) 订阅“chat”频道。
  2. 收到服务器广播的聊天消息时,callback 会把消息推入 Pinia Store。
  3. status 计算属性实时反映 websocketService.status 状态(CONNECTING/OPEN/CLOSED)。
  4. logout() 先发送离开消息,再 unsubscribe()disconnect()。在组件卸载前调用 logout(),保证资源释放。

6.3 MessageList 组件:显示消息列表

<template>
  <div class="message-list">
    <div v-for="(msg, idx) in messages" :key="idx" class="message-item">
      <span class="user">{{ msg.user }}:</span>
      <span class="text">{{ msg.text }}</span>
      <span class="time">{{ msg.time }}</span>
    </div>
  </div>
</template>

<script setup>
import { computed, nextTick, ref } from 'vue';
import { useChatStore } from '@/store/chatStore';

const chatStore = useChatStore();
const messages = computed(() => chatStore.messages);

const listRef = ref(null);

// 自动滚到底部
watch(messages, async () => {
  await nextTick();
  const el = listRef.value;
  if (el) el.scrollTop = el.scrollHeight;
});

</script>

<style scoped>
.message-list {
  border: 1px solid #ccc;
  height: 300px;
  overflow-y: auto;
  padding: 8px;
  background: #fafafa;
}
.message-item {
  margin-bottom: 6px;
}
.user {
  font-weight: bold;
  margin-right: 4px;
}
.time {
  color: #999;
  font-size: 12px;
  margin-left: 8px;
}
</style>

6.3.1 关键说明

  • 从 Pinia Store 中读取 messages 数组,并通过 v-for 渲染。
  • 使用 watch 监听 messages 变化,在新消息到来后自动滚动到底部。

6.4 MessageInput 组件:输入并发送消息

<template>
  <div class="message-input">
    <input
      v-model="text"
      placeholder="输入消息后按回车"
      @keydown.enter.prevent="send"
    />
    <button @click="send">发送</button>
  </div>
</template>

<script setup>
import { ref } from 'vue';
import { useChatStore } from '@/store/chatStore';
import websocketService from '@/services/websocketService';

const chatStore = useChatStore();
const text = ref('');

// 发送聊天消息
function send() {
  const content = text.value.trim();
  if (!content) return;
  const msg = {
    user: chatStore.user,
    text: content,
    time: new Date().toLocaleTimeString()
  };
  // 1. 先本地回显
  chatStore.addMessage(msg);
  // 2. 发送给服务器
  websocketService.send('chat', msg);
  text.value = '';
}
</script>

<style scoped>
.message-input {
  display: flex;
  margin-top: 8px;
}
.message-input input {
  flex: 1;
  padding: 6px;
  font-size: 14px;
  margin-right: 8px;
  border: 1px solid #ccc;
  border-radius: 4px;
}
.message-input button {
  padding: 6px 12px;
  background: #67c23a;
  border: none;
  color: white;
  border-radius: 4px;
  cursor: pointer;
}
button:hover {
  background: #85ce61;
}
</style>

6.4.1 关键说明

  1. 用户输入后,send() 会先将消息推入本地 chatStore.addMessage(msg),保证“即时回显”;
  2. 再执行 websocketService.send('chat', msg),将消息广播给服务器;
  3. 服务器收到后再广播给所有在线客户端(包括发送者自己),其他客户端即触发 handleMessage 并更新 Store。

6.5 数据流动图解

┌─────────────────────────────┐
│        ChatManager.vue      │
│  ┌───────────────────────┐  │
│  │ 点击“登录”:login()   │  │
│  │   ↓                   │  │
│  │ websocketService.connect │  │
│  │   ↓                   │  │
│  │ ws.onopen → startHeartBeat() │
│  │   ↓                   │  │
│  │ subscribe('chat', callback)  │
│  └───────────────────────┘  │
│                              │
│                              │
│  服务器推送 → ws.onmessage    │
│      ↓                         │
│ ChatManager.handleMessage     │
│      ↓                         │
│ chatStore.addMessage(payload) │
└─────────────────────────────┘
           ↑
           │
┌─────────────────────────────┐
│      MessageList.vue         │
│   watch(messages) → 渲染列表  │
└─────────────────────────────┘

┌─────────────────────────────┐
│     MessageInput.vue        │
│ 用户输入 → send()            │
│   ↓                          │
│ chatStore.addMessage(local) │
│   ↓                          │
│ websocketService.send(...)   │
└─────────────────────────────┘
  1. 登录ChatManager.vue 调用 websocketService.connect(),再 subscribe('chat', cb)
  2. 接收消息ws.onmessagehandleMessagechatStore.addMessageMessageList.vue 渲染。
  3. 发送消息MessageInput.vuechatStore.addMessage 提前回显 → 然后 websocketService.send('chat', msg) 发给服务器 → 服务器再广播给所有客户端。(发送者自己也会再次收到并重复渲染,注意过滤或去重)

多组件/模块共享同一个 WebSocket 连接

在大型项目里,往往有多个功能模块都需要通过 WebSocket 通信。我们要保证全局只有一个 WebSocket 实例,避免重复连接。

7.1 在 Pinia 中维护状态

可以将 websocketService 作为单独模块,也可以在 Pinia Store 中维护 WebSocket 实例引用并封装调用。

// src/store/chatStore.js
import { defineStore } from 'pinia';
import websocketService from '@/services/websocketService';

export const useChatStore = defineStore('chat', {
  state: () => ({
    user: '',
    messages: []
  }),
  actions: {
    setUser(name) {
      this.user = name;
    },
    addMessage(msg) {
      this.messages.push(msg);
    }
  },
  getters: {
    messageCount: (state) => state.messages.length
  }
});
注意websocketService 本身是个独立单例,只要在组件或 Store 里 import 一次,无论何处调用,都是同一个实例。

7.2 通过 provide/inject 或全局属性共享实例

除了 Store,也可以在 App.vueprovide('ws', websocketService),让子组件通过 inject('ws') 直接访问该实例。上述示例在 ChatManager.vueMessageInput.vueMessageList.vue 中直接 import websocketService 即可,因为它是单例。


心跳检测与自动重连实现

为了保证连接的持久性与可靠性,需要在 websocketService 中内置心跳与重连逻辑。

8.1 心跳机制原理

  • 发送心跳:客户端定期(如 30 秒)向服务器发送 { topic: 'PING' },服务器需监听并回应 { topic: 'HEARTBEAT' }
  • 超时检测:如果在两倍心跳间隔时长内没有收到服务器的 HEARTBEAT,则判定连接已断,主动触发 disconnect() 并启动重连。
// 在 websocketService.handleMessage 中
handleMessage(raw) {
  const msg = JSON.parse(raw);
  const { topic, payload } = msg;
  if (topic === 'HEARTBEAT') {
    this.lastPong = Date.now();
    return;
  }
  // 其余逻辑…
}

// 心跳定时器
startHeartBeat() {
  this.stopHeartBeat();
  this.lastPong = Date.now();
  this.heartBeatTimer = setInterval(() => {
    if (Date.now() - this.lastPong > this.heartBeatInterval * 2) {
      // 认为已断开
      console.warn('[WebSocket] 心跳超时,尝试重连');
      this.disconnect();
      this.tryReconnect(this.url);
      return;
    }
    this.send('PING', { ts: Date.now() });
  }, this.heartBeatInterval);
}

8.2 在 websocketService 中集成心跳与重连

// 修改 connect 方法,保存 url
connect(url) {
  this.url = url;
  // …其余不变
}

tryReconnect(url) {
  // 断开时自动重连
  this.clearTimers();
  this.reconnectTimer = setInterval(() => {
    console.log('[WebSocket] 重连尝试...');
    this.connect(url);
  }, this.reconnectInterval);
}

8.2.1 防抖/节流发送心跳

如果业务场景复杂,可能有大量频繁消息进出,心跳不应与常规消息冲突。可通过节流控制心跳发送:

import { throttle } from 'lodash';

startHeartBeat() {
  this.stopHeartBeat();
  this.lastPong = Date.now();
  const sendPing = throttle(() => {
    this.send('PING', { ts: Date.now() });
  }, this.heartBeatInterval);
  this.heartBeatTimer = setInterval(() => {
    if (Date.now() - this.lastPong > this.heartBeatInterval * 2) {
      this.disconnect();
      this.tryReconnect(this.url);
      return;
    }
    sendPing();
  }, this.heartBeatInterval);
}

性能优化与最佳实践

9.1 减少重复消息订阅

  • 某些需求下,不同模块需要监听同一个 topic,直接在 websocketServicelisteners 将多个回调并存即可。
  • 若尝试重复 subscribe('chat', cb),封装里会判断 this.listeners 是否已有该 topic,若已有则不重新 send('SUBSCRIBE')

9.2 批量发送与分包

  • 当要发送大量数据(如一次性发很多消息),应考虑批量分包,避免一次性写入阻塞。可以用 setTimeoutrequestIdleCallback 分段写入。
  • 亦可将多条消息合并成一个对象 { topic:'BATCH', payload: [msg1, msg2, …] },在服务器端解包后再分发。

9.3 合理关闭与清理监听器

  • 在组件卸载或离开页面时,务必调用 unsubscribe(topic, callback)disconnect(),避免无效回调积累,导致内存泄漏。
  • 对于短期连接需求(例如只在某些页面使用),可在路由守卫 beforeRouteLeave 中断开连接。

常见问题与调试技巧

  1. 浏览器报 SecurityErrorReferenceError: navigator.serial is undefined

    • 确保在 HTTPS 或本地 localhost 环境下运行;
    • 在不支持 Web Serial API 的浏览器(如 Firefox、Safari)会找不到 navigator.serial,需做兼容性提示。
  2. WebSocket 握手失败或频繁重连

    • 检查服务端地址是否正确,是否开启了 SSL(如果用 wss://);
    • 服务器是否允许跨域,WebSocket endpoint 是否配置了 Access-Control-Allow-Origin: *
    • 在 DevTools Network → WS 面板检查握手 HTTP 请求/响应。
  3. 心跳无效无法保持连接

    • 确保服务器代码正确响应 PING,返回 { topic: 'HEARTBEAT' }
    • 客户端记得在 handleMessage 内及时更新 lastPong 时间戳。
  4. 重复消息

    • 如果在重连后没有调用 unsubscribe 就重新 subscribe,可能会收到多份同一消息;
    • 解决:在 connect() 之前先清空 listeners 或使用 Set 去重回调。
  5. 消息 JSON 解析失败

    • 确保客户端与服务端约定好消息格式,都是 { topic, payload } 串行化的 JSON;
    • handleMessage 中用 try…catch 捕获解析错误并打印原始 raw 数据。

总结

本文从WebSocket 原理讲起,重点演示了在 Vue 3 + Pinia 项目中,如何通过单例服务 + 心跳/重连 + 订阅管理机制,实现一个高效、稳定、易维护的 WebSocket 通信层。我们完整演示了一个“简易实时聊天室”示例,包括:

  • websocketService.js:集中管理连接、断开、重连、心跳、消息订阅/广播。
  • ChatManager.vue:负责登录/登出、订阅频道、更新全局 Store。
  • MessageList.vue:实时渲染收到的聊天消息,自动滚到底部。
  • MessageInput.vue:发送聊天消息并本地回显。

同时讨论了心跳检测与自动重连的实现思路,演示了性能优化(节流、批量分包)和常见问题的排查方法。通过本文内容,你应该能够在自己的 Vue 应用里灵活、高效地集成 WebSocket,实现各种实时通信场景,如:聊天室、实时监控、股价行情、游戏对战等。

2024-11-27

Python Socket 详解,最全教程

Socket 是计算机网络编程的基础工具,它提供了跨网络通信的能力。在 Python 中,socket 模块是开发网络应用的核心库。本教程将详细介绍 Python socket 模块的基础知识、用法及应用场景,并通过代码示例和图解帮助你快速入门。


一、什么是 Socket?

Socket 是网络中不同程序间通信的桥梁。它允许程序发送或接收数据,通常用于构建服务器与客户端模型。

常见 Socket 类型

  1. TCP(传输控制协议): 提供可靠的、基于连接的通信。
  2. UDP(用户数据报协议): 提供不可靠、无连接的通信,但速度快。

二、Python Socket 基本用法

1. 导入模块

在使用 socket 前,需导入模块:

import socket

2. 创建 Socket

基本语法:

s = socket.socket(family, type)
  • family: 地址族,例如 AF_INET(IPv4)或 AF_INET6(IPv6)。
  • type: 套接字类型,例如 SOCK_STREAM(TCP)或 SOCK_DGRAM(UDP)。

示例:

# 创建一个 TCP 套接字
tcp_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 创建一个 UDP 套接字
udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

3. 客户端通信流程

TCP 客户端通信的基本步骤如下:

1. 创建套接字

client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

2. 连接到服务器

server_address = ('127.0.0.1', 65432)  # 地址和端口
client_socket.connect(server_address)

3. 发送和接收数据

client_socket.sendall(b'Hello, Server!')
response = client_socket.recv(1024)  # 接收数据,最大字节数
print(f'Received: {response}')

4. 关闭套接字

client_socket.close()

完整示例:

import socket

# 创建客户端套接字
client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 连接服务器
server_address = ('127.0.0.1', 65432)
client_socket.connect(server_address)

try:
    # 发送数据
    message = b'Hello, Server!'
    client_socket.sendall(message)

    # 接收响应
    response = client_socket.recv(1024)
    print(f'Received: {response.decode()}')
finally:
    client_socket.close()

4. 服务器通信流程

TCP 服务器通信的基本步骤如下:

1. 创建套接字

server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

2. 绑定地址

server_socket.bind(('127.0.0.1', 65432))  # 绑定 IP 和端口

3. 开始监听

server_socket.listen(5)  # 最大连接数

4. 接收连接和处理

connection, client_address = server_socket.accept()
print(f'Connection from {client_address}')

data = connection.recv(1024)  # 接收数据
print(f'Received: {data.decode()}')

connection.sendall(b'Hello, Client!')  # 发送响应
connection.close()

完整示例:

import socket

# 创建服务器套接字
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server_socket.bind(('127.0.0.1', 65432))
server_socket.listen(5)

print('Server is listening...')

while True:
    connection, client_address = server_socket.accept()
    try:
        print(f'Connection from {client_address}')
        data = connection.recv(1024)
        print(f'Received: {data.decode()}')

        if data:
            connection.sendall(b'Hello, Client!')
    finally:
        connection.close()

运行结果:

  1. 启动服务器。
  2. 启动客户端发送数据。
  3. 客户端收到响应。

三、UDP 通信

与 TCP 不同,UDP 是无连接协议,不需要建立连接。

1. UDP 客户端

示例:

import socket

udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

server_address = ('127.0.0.1', 65432)
udp_socket.sendto(b'Hello, UDP Server!', server_address)

data, server = udp_socket.recvfrom(1024)
print(f'Received: {data.decode()}')

udp_socket.close()

2. UDP 服务器

示例:

import socket

udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
udp_socket.bind(('127.0.0.1', 65432))

print('UDP server is listening...')

while True:
    data, address = udp_socket.recvfrom(1024)
    print(f'Received {data.decode()} from {address}')

    udp_socket.sendto(b'Hello, UDP Client!', address)

四、图解 Socket 通信

1. TCP 通信模型

+------------+       +-------------+
|  Client    |       |  Server     |
+------------+       +-------------+
| Connect()  | <-->  | Accept()    |
| Send()     | <-->  | Receive()   |
| Receive()  | <-->  | Send()      |
| Close()    | <-->  | Close()     |
+------------+       +-------------+

2. UDP 通信模型

+------------+         +-------------+
|  Client    |         |  Server     |
+------------+         +-------------+
| SendTo()   | ----->  | RecvFrom()  |
| RecvFrom() | <-----  | SendTo()    |
+------------+         +-------------+

五、Socket 编程的常见问题

1. Address already in use

原因: 套接字未关闭或正在使用。
解决:

server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

2. Connection reset by peer

原因: 客户端过早断开连接。
解决: 检查连接和数据流逻辑。

3. Timeout

原因: 通信超时。
解决:

socket.settimeout(5)  # 设置超时时间

六、Socket 的高级用法

  1. 多线程/多进程支持: 使用 threadingmultiprocessing 模块实现并发。
  2. SSL/TLS 支持: 使用 ssl 模块实现加密通信。
  3. 非阻塞 Socket: 设置套接字为非阻塞模式,适用于高性能应用。
  4. WebSocket 支持: 可结合 websockets 库构建实时通信。

七、总结

通过本文的介绍,你已经掌握了 Python socket 的基本概念和使用方法。无论是实现简单的客户端-服务器通信,还是构建复杂的网络应用,socket 都是不可或缺的工具。

练习建议:

  1. 使用 TCP 创建一个聊天室应用。
  2. 使用 UDP 构建一个简单的文件传输工具。
  3. 探索 SSL 加密通信。

拓展阅读:

  • 官方文档:Python socket
  • 实战项目:用 socket 构建 HTTP 服务。

快动手尝试吧!Socket 是网络编程的基石,掌握它将为你打开更广阔的编程世界。

2024-11-26

Python的WebSocket方法教程

WebSocket 是一种通信协议,允许客户端和服务器之间的双向实时通信。它常用于需要实时交互的应用场景,例如在线聊天、实时数据更新和在线游戏。在 Python 中,有多种库支持 WebSocket,其中 websockets 是一款简单易用的库。

本文将全面介绍 WebSocket 的基本原理、安装配置以及 Python 中 WebSocket 的使用方法,配以代码示例和图解,帮助你快速掌握 WebSocket 的开发。


一、WebSocket简介

1. 什么是WebSocket?

  • WebSocket 是一种在单个 TCP 连接上实现全双工通信的协议。
  • 它的通信方式不同于传统 HTTP 请求-响应模式,WebSocket 建立后,客户端和服务器可以随时互发消息。

传统 HTTP 和 WebSocket 的区别:

特性HTTPWebSocket
通信模式请求-响应全双工
连接保持每次请求建立连接,完成后断开连接建立后持续
实时性较差
场景静态数据传输实时互动应用

2. WebSocket 工作流程

  1. 客户端向服务器发送 WebSocket 握手请求。
  2. 服务器返回响应,确认协议升级。
  3. 握手成功后,客户端和服务器可以进行双向通信。
  4. 双方可以在连接期间随时发送消息。
  5. 连接关闭后,通信结束。

图解:WebSocket工作流程

客户端               服务器
  |----握手请求----->|
  |<----握手确认-----|
  |<====建立连接====>|
  |<====数据交换====>|
  |<----关闭连接---->|

二、Python 中的 WebSocket 使用

1. 安装依赖

我们使用 websockets 库,它是 Python 中功能强大且易用的 WebSocket 库。

安装方式:

pip install websockets

2. 创建 WebSocket 服务器

下面是一个简单的 WebSocket 服务器示例,监听客户端连接并与之通信。

示例代码

import asyncio
import websockets

# 处理客户端连接
async def echo(websocket, path):
    print("客户端已连接")
    try:
        async for message in websocket:
            print(f"收到消息: {message}")
            await websocket.send(f"服务端回复: {message}")
    except websockets.ConnectionClosed:
        print("客户端断开连接")

# 启动服务器
start_server = websockets.serve(echo, "localhost", 12345)

print("WebSocket服务器已启动,监听端口12345")

asyncio.get_event_loop().run_until_complete(start_server)
asyncio.get_event_loop().run_forever()

运行说明

  • 服务端会监听 localhost:12345,并等待客户端连接。
  • 当客户端发送消息时,服务端会回显消息。

3. 创建 WebSocket 客户端

我们用客户端连接服务器并发送消息。

示例代码

import asyncio
import websockets

async def communicate():
    uri = "ws://localhost:12345"
    async with websockets.connect(uri) as websocket:
        await websocket.send("你好,服务器!")
        response = await websocket.recv()
        print(f"收到服务端回复: {response}")

# 运行客户端
asyncio.run(communicate())

运行说明

  • 客户端连接到 ws://localhost:12345
  • 客户端发送消息后接收服务端的回显。

三、WebSocket 实战应用

1. 实现简单聊天室

通过 WebSocket 实现一个多人聊天的服务器。

服务端代码

import asyncio
import websockets

connected_users = set()

async def chat_handler(websocket, path):
    connected_users.add(websocket)
    print(f"新用户加入,当前用户数: {len(connected_users)}")
    try:
        async for message in websocket:
            print(f"收到消息: {message}")
            # 广播消息给所有用户
            for user in connected_users:
                if user != websocket:
                    await user.send(message)
    except websockets.ConnectionClosed:
        print("用户断开连接")
    finally:
        connected_users.remove(websocket)

start_server = websockets.serve(chat_handler, "localhost", 12345)

print("聊天服务器启动中...")
asyncio.get_event_loop().run_until_complete(start_server)
asyncio.get_event_loop().run_forever()

客户端代码

import asyncio
import websockets

async def chat_client():
    uri = "ws://localhost:12345"
    async with websockets.connect(uri) as websocket:
        print("已连接到聊天室。输入消息并按回车发送:")
        while True:
            message = input("你:")
            await websocket.send(message)
            response = await websocket.recv()
            print(f"其他人:{response}")

# 运行客户端
asyncio.run(chat_client())

2. 服务端性能优化

  • 心跳检测:定期发送 Ping 来检测连接状态。
  • 连接限流:限制并发用户数。
  • 日志记录:记录每个连接的活动。

四、WebSocket 常见问题与解决

1. 为什么连接会失败?

  • 服务端未启动或地址错误。
  • 网络不通或防火墙阻断。

2. 如何处理连接中断?

  • 在客户端设置重连机制。
  • 使用 try...except 捕获 ConnectionClosed 异常。

示例:客户端重连机制

async def reconnect(uri):
    while True:
        try:
            async with websockets.connect(uri) as websocket:
                print("已连接到服务器")
                while True:
                    message = input("请输入消息:")
                    await websocket.send(message)
                    print(await websocket.recv())
        except websockets.ConnectionClosed:
            print("连接断开,尝试重连...")
            await asyncio.sleep(5)

五、WebSocket 应用场景

  • 实时聊天:支持多人实时聊天功能。
  • 实时数据更新:如股票价格、物联网数据监控。
  • 游戏通信:实现低延迟的多人在线游戏。
  • 通知推送:服务端主动推送消息到客户端。

六、总结

WebSocket 是实现实时通信的重要工具,Python 提供了功能强大的库来帮助我们快速开发 WebSocket 应用。通过 websockets,我们可以轻松实现双向通信、多人聊天和实时数据更新等功能。

学习要点

  1. 掌握 WebSocket 的基本原理和通信流程。
  2. 学会搭建 WebSocket 服务器和客户端。
  3. 理解 WebSocket 的实战应用场景。

希望本文对你学习 WebSocket 的方法和技巧有所帮助!如果你有更多问题,欢迎交流讨论!

2024-09-09

在Spring Boot中使用WebSocket可以通过Spring的WebSocket支持来实现。以下是一个简单的例子,展示了如何在Spring Boot应用程序中设置WebSocket端点。

  1. 添加依赖到你的pom.xml



<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
  1. 配置WebSocket:



import org.springframework.context.annotation.Configuration;
import org.springframework.messaging.simp.config.MessageBrokerRegistry;
import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
 
@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
    
    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        registry.addEndpoint("/ws").withSockJS();
    }
 
    @Override
    public void configureMessageBroker(MessageBrokerRegistry registry) {
        registry.enableSimpleBroker("/topic");
        registry.setApplicationDestinationPrefixes("/app");
    }
}
  1. 创建WebSocket控制器:



import org.springframework.messaging.handler.annotation.MessageMapping;
import org.springframework.messaging.handler.annotation.SendTo;
import org.springframework.stereotype.Controller;
 
@Controller
public class WebSocketController {
 
    @MessageMapping("/hello")
    @SendTo("/topic/greetings")
    public String greeting(String message) {
        return "Hello, " + message + "!";
    }
}
  1. 前端使用WebSocket的例子(使用JavaScript和SockJS客户端):



<script src="https://cdn.jsdelivr.net/npm/sockjs-client@1.7.0/dist/sockjs.min.js"></script>
<script>
    var socket = new SockJS('/ws');
    var stompClient = Stomp.over(socket);
    stompClient.connect({}, function(frame) {
        console.log('Connected: ' + frame);
        stompClient.subscribe('/topic/greetings', function(greeting){
            console.log(greeting.body);
        });
    });
    
    function sendName() {
        var name = document.getElementById('name').value;
        stompClient.send("/app/hello", {}, name);
    }
</script>
 
<input type="text" id="name" />
<button onclick="sendName()">Send</button>

这个例子展示了如何在Spring Boot应用程序中设置WebSocket端点,并使用SockJS和STOMP协议与客户端进行通信。在前端,用户可以在文本框输入名字,点击发送按钮,将消息发送到后端,后端处理这个消息并通过WebSocket发送回客户端。

2024-09-09

在Spring Boot中集成WebSocket以实现WebRTC功能,你可以使用Spring的@EnableWebSocketMessageBroker注解来配置WebSocket作为信息代理。以下是一个简化的例子:

  1. 添加Maven依赖:



<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
  1. 配置WebSocketMessageBroker:



import org.springframework.context.annotation.Configuration;
import org.springframework.messaging.simp.config.MessageBrokerRegistry;
import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
 
@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
    
    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        registry.addEndpoint("/ws").withSockJS();
    }
 
    @Override
    public void configureMessageBroker(MessageBrokerRegistry registry) {
        registry.enableSimpleBroker("/topic");
        registry.setApplicationDestinationPrefixes("/app");
    }
}
  1. 创建WebSocket控制器处理WebRTC信令:



import org.springframework.messaging.handler.annotation.MessageMapping;
import org.springframework.messaging.handler.annotation.SendTo;
import org.springframework.stereotype.Controller;
 
@Controller
public class WebSocketController {
 
    @MessageMapping("/webrtc/offer")
    @SendTo("/topic/webrtc/offers")
    public String handleOffer(String offer) {
        // 处理Offer信令并返回应答信令
        return "应答信令";
    }
 
    // 其他WebRTC信令处理方法...
}

在这个例子中,我们定义了一个WebSocket端点/ws,并且配置了一个简单的代理,将应用程序目标前缀设置为/app。然后,我们创建了一个控制器,其中包含处理WebRTC Offer信令的方法。这个例子展示了如何使用Spring Boot和WebSocket实现WebRTC信令服务的基本框架。在实际应用中,你需要实现完整的WebRTC信令处理逻辑,以及任何必要的安全措施。

2024-09-09

解释:

Spring Cloud Gateway是Spring Cloud生态中的一个项目,它提供了一个API网关,用于转发请求。当Spring Cloud Gateway尝试转发WebSocket请求时,如果遇到404错误,通常意味着Gateway没有正确配置来处理WebSocket请求,或者目标WebSocket端点不存在。

解决方法:

  1. 确保你的Gateway路由配置正确。你需要为WebSocket流量指定正确的路由,并确保路径匹配是正确的。例如:



spring:
  cloud:
    gateway:
      routes:
        - id: websocket_route
          uri: ws://websocket-service:8080
          predicates:
            - Path=/ws/**
  1. 确保WebSocket服务是可达的。Gateway需要能够连接到配置的WebSocket服务地址。
  2. 如果你使用的是Spring WebFlux,确保你的WebSocket处理是基于WebFlux的。
  3. 检查安全配置。如果你使用了Spring Security,确保WebSocket端点不被安全规则阻止。
  4. 如果你使用的是HTTP来代理WebSocket请求,确保你没有在Gateway中配置任何只适用于HTTP请求的过滤器或者路由策略,因为WebSocket是一个持久化的连接,可能需要不同的处理流程。
  5. 查看日志文件,以获取更多关于404错误的详细信息,这可能会提供额外的线索。
  6. 如果你已经确认以上配置都没有问题,但问题依然存在,可以考虑检查WebSocket服务是否已经正确启动并且监听在预期的端口上。

总结,解决Spring Cloud Gateway转发WebSocket时出现404错误,需要检查路由配置、WebSocket服务可达性、WebFlux支持以及安全配置。如果问题依然存在,检查服务启动状态和日志文件。

2024-09-09

以下是一个简化的Spring Boot WebSocket解决方案示例,使用Redis作为消息代理进行发布/订阅:




import org.springframework.context.annotation.Configuration;
import org.springframework.data.redis.connection.RedisConnectionFactory;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.listener.RedisMessageListenerContainer;
import org.springframework.messaging.simp.config.MessageBrokerRegistry;
import org.springframework.web.socket.config.annotation.*;
 
@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
 
    private final RedisConnectionFactory connectionFactory;
    private final StringRedisTemplate redisTemplate;
 
    public WebSocketConfig(RedisConnectionFactory connectionFactory, StringRedisTemplate redisTemplate) {
        this.connectionFactory = connectionFactory;
        this.redisTemplate = redisTemplate;
    }
 
    @Override
    public void configureMessageBroker(MessageBrokerRegistry config) {
        config.enableSimpleBroker("/topic", "/queue"); // 客户端订阅地址前缀
        config.setApplicationDestinationPrefixes("/app"); // 客户端发送信息的地址前缀
    }
 
    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        registry.addEndpoint("/ws").withSockJS(); // 注册STOMP协议的节点,并映射指定的URL,并指定使用SockJS协议
    }
 
    @Override
    public void configureClientInboundChannel(ChannelRegistration registration) {
        registration.interceptors(new ChatMessageInterceptor()); // 注册客户端发往服务端的通道的拦截器
    }
 
    @Override
    public void configureClientOutboundChannel(ChannelRegistration registration) {
        // 注册服务端发往客户端的通道的拦截器
    }
 
    @Override
    public void configureMessageConverters(List<MessageConverter> converters) {
        // 配置消息转换器
    }
 
    @Override
    public void configureWebSocketTransport(WebSocketTransportRegistration registration) {
        // 配置WebSocket传输
    }
 
    // 使用RedisMessageListenerContainer来监听Redis订阅频道
    @Bean
    public RedisMessageListenerContainer redisMessageListenerContainer() {
        RedisMessageListenerContainer container = new RedisMessageListenerContainer();
        container.setConnectionFactory(connectionFactory);
        container.addMessageListener(new ChatMessageListener(), topic());
        return container;
    }
 
    // 定义订阅的Redis频道
    @Bean
    public Topic topic() {
        return new ChannelTopic("
2024-09-09



import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;
import io.netty.channel.Channel;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.http.DefaultFullHttpResponse;
import io.netty.handler.codec.http.FullHttpRequest;
import io.netty.handler.codec.http.HttpHeaderUtil;
import io.netty.handler.codec.http.HttpHeaders;
import io.netty.handler.codec.http.HttpResponseStatus;
import io.netty.handler.codec.http.HttpVersion;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.WebSocketSession;
 
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
 
@Component
public class WebSocketHandler {
 
    private static final Map<String, Channel> sessionMap = new ConcurrentHashMap<>();
 
    @Autowired
    private ApplicationContext applicationContext;
 
    public void handleHttpRequest(ChannelHandlerContext ctx, FullHttpRequest req) {
        if (!req.decoderResult().isSuccess()
                || (!"websocket".equals(req.headers().get("Upgrade")))) {
            sendHttpResponse(ctx, req, new DefaultFullHttpResponse(
                    HttpVersion.HTTP_1_1, HttpResponseStatus.BAD_REQUEST));
            return;
        }
 
        String sessionId = req.headers().get("Sec-WebSocket-Key");
        ctx.channel().attr(WebSocketSessionManager.SESSION_KEY).set(sessionId);
        WebSocketSessionManager.addSession(sessionId, ctx.channel());
 
        // 这里省略了WebSocket握手相关代码
    }
 
    public void sendMessageToClient(String sessionId, String message) {
        Channel channel = sessionMap.get(sessionId);
        if (channel == null) {
            return;
        }
 
        ByteBuf byteBuf = Unpooled.buffer();
        byteBuf.writeCharSequence(message, StandardCharsets.UTF_8);
        channel.writeAndFlush(byteBuf);
    }
 
    private void sendHttpResponse(ChannelHandlerContext ctx, FullHttpRequest req, DefaultFullHttpResponse res) {
        if (res.status().code() != 200) {
            ByteBuf byteBuf = Unpooled.copiedBuffer(res.status().toString(), StandardCharsets.UTF_8);
            res.content().writeBytes(byteBuf);
            byteBuf.release();
         
2024-09-06

在Spring Boot中,有两种常见的WebSocket实现方式:使用Spring的@EnableWebSocketMessageBroker注解和使用WebSocketServletFactory

  1. 使用@EnableWebSocketMessageBroker注解



import org.springframework.context.annotation.Configuration;
import org.springframework.messaging.simp.config.MessageBrokerRegistry;
import org.springframework.web.socket.config.annotation.*;
 
@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        registry.addEndpoint("/ws").withSockJS();
    }
 
    @Override
    public void configureMessageBroker(MessageBrokerRegistry registry) {
        registry.enableSimpleBroker("/topic");
        registry.setApplicationDestinationPrefixes("/app");
    }
}
  1. 使用WebSocketServletFactory



import org.eclipse.jetty.websocket.api.WebSocketServlet;
import org.eclipse.jetty.websocket.api.WebSocketServletFactory;
import org.eclipse.jetty.websocket.servlet.WebSocketServletFactory;
 
public class WebSocketEndpoint extends WebSocketServlet {
 
    @Override
    public void configure(WebSocketServletFactory factory) {
        factory.register(MyWebSocket.class);
    }
}

在这个例子中,MyWebSocket是一个继承自WebSocket的类,用于处理WebSocket连接的打开、关闭和消息接收。

注意:第二种方式使用的是org.eclipse.jetty.websocket.api包中的WebSocket API,这是Jetty项目提供的WebSocket实现。如果你使用的是Tomcat作为你的Servlet容器,那么你可能需要使用Spring的@EnableWebSocketMessageBroker注解方式。