WebSocket 实战案例：从设计到部署

在前六篇文章中,我们深入探讨了 WebSocket 的基础原理、服务端开发、客户端实现、安全实践、性能优化和测试调试。今天,让我们通过一个实战案例,看看如何将这些知识应用到实际项目中。我曾在一个大型在线教育平台中,通过 WebSocket 实现了实时互动课堂,支持了数万名师生的同时在线。

项目背景

我们要实现一个实时互动课堂系统,主要功能包括：

实时音视频
课堂互动
共享白板
实时聊天
课堂管理

让我们从系统设计开始。

系统架构

实现系统架构：

// app.js
const express = require('express')
const https = require('https')
const fs = require('fs')
const path = require('path')
const WebSocket = require('ws')
const Redis = require('ioredis')
const { ClusterManager } = require('./cluster-manager')
const { ConnectionPool } = require('./connection-pool')
const { MessageQueue } = require('./message-queue')
const { RoomManager } = require('./room-manager')
const { UserManager } = require('./user-manager')
const { MediaServer } = require('./media-server')class ClassroomServer {constructor(options = {}) {this.options = {port: 8080,sslPort: 8443,redisUrl: 'redis://localhost:6379',mediaServer: 'localhost:8000',...options}// 初始化组件this.cluster = new ClusterManager()this.pool = new ConnectionPool()this.queue = new MessageQueue()this.rooms = new RoomManager()this.users = new UserManager()this.media = new MediaServer(this.options.mediaServer)this.initialize()}// 初始化服务器async initialize() {// 创建 Express 应用this.app = express()this.setupExpress()// 创建 HTTPS 服务器this.server = https.createServer({key: fs.readFileSync('server.key'),cert: fs.readFileSync('server.cert')}, this.app)// 创建 WebSocket 服务器this.wss = new WebSocket.Server({server: this.server,path: '/ws'})// 连接 Redisthis.redis = new Redis(this.options.redisUrl)// 设置事件处理器this.setupEventHandlers()// 启动服务器await this.start()}// 设置 ExpresssetupExpress() {// 静态文件this.app.use(express.static('public'))// API 路���this.app.use('/api', require('./routes/api'))// 错误处理this.app.use((err, req, res, next) => {console.error('Express error:', err)res.status(500).json({ error: 'Internal server error' })})}// 设置事件处理器setupEventHandlers() {// WebSocket 连接this.wss.on('connection', (ws, req) => {this.handleConnection(ws, req)})// Redis 订阅this.redis.on('message', (channel, message) => {this.handleRedisMessage(channel, message)})// 进程消息process.on('message', (message) => {this.handleProcessMessage(message)})}// 处理 WebSocket 连接async handleConnection(ws, req) {try {// 验证用户const user = await this.users.authenticate(req)// 创建连接const connection = this.pool.createConnection(ws, user)// 加入房间const roomId = req.query.roomIdif (roomId) {await this.rooms.joinRoom(roomId, connection)}// 设置消息处理器ws.on('message', (message) => {this.handleMessage(connection, message)})// 设置关闭处理器ws.on('close', () => {this.handleClose(connection)})// 发送欢迎消息connection.send({type: 'welcome',data: {user: user.toJSON(),room: roomId ? await this.rooms.getRoomInfo(roomId) : null}})} catch (error) {console.error('Connection error:', error)ws.close()}}// 处理消息async handleMessage(connection, message) {try {const data = JSON.parse(message)// 验证消息if (!this.validateMessage(data)) {throw new Error('Invalid message format')}// 处理不同类型的消息switch (data.type) {case 'chat':await this.handleChatMessage(connection, data)breakcase 'whiteboard':await this.handleWhiteboardMessage(connection, data)breakcase 'media':await this.handleMediaMessage(connection, data)breakcase 'control':await this.handleControlMessage(connection, data)breakdefault:throw new Error('Unknown message type')}} catch (error) {console.error('Message error:', error)connection.send({type: 'error',error: error.message})}}// 处理聊天消息async handleChatMessage(connection, message) {const { roomId, content } = message.data// 验证权限if (!await this.rooms.canChat(connection.user, roomId)) {throw new Error('No permission to chat')}// 创建聊天消息const chat = {id: generateId(),roomId,userId: connection.user.id,content,timestamp: Date.now()}// 保存到数据库await this.rooms.saveChatMessage(chat)// 广播到房间await this.rooms.broadcast(roomId, {type: 'chat',data: chat})}// 处理白板消息async handleWhiteboardMessage(connection, message) {const { roomId, action } = message.data// 验证权限if (!await this.rooms.canDraw(connection.user, roomId)) {throw new Error('No permission to draw')}// 处理白板动作const result = await this.rooms.handleWhiteboardAction(roomId, action)// 广播到房间await this.rooms.broadcast(roomId, {type: 'whiteboard',data: {action,result}})}// 处理媒体消息async handleMediaMessage(connection, message) {const { roomId, stream } = message.data// 验证权限if (!await this.rooms.canPublish(connection.user, roomId)) {throw new Error('No permission to publish')}// 处理媒体流const result = await this.media.handleStream(roomId, stream)// 广播到房间await this.rooms.broadcast(roomId, {type: 'media',data: {stream,result}})}// 处理控制消息async handleControlMessage(connection, message) {const { roomId, action } = message.data// 验证权限if (!await this.rooms.canControl(connection.user, roomId)) {throw new Error('No permission to control')}// 处理控制命令const result = await this.rooms.handleControlAction(roomId, action)// 广播到房间await this.rooms.broadcast(roomId, {type: 'control',data: {action,result}})}// 处理连接关闭async handleClose(connection) {try {// 离开房间const roomId = connection.roomIdif (roomId) {await this.rooms.leaveRoom(roomId, connection)}// 清理连接this.pool.removeConnection(connection)// 广播离开消息if (roomId) {await this.rooms.broadcast(roomId, {type: 'user_left',data: {userId: connection.user.id}})}} catch (error) {console.error('Close error:', error)}}// 处理 Redis 消息handleRedisMessage(channel, message) {try {const data = JSON.parse(message)// 处理不同类型的消息switch (channel) {case 'room_update':this.handleRoomUpdate(data)breakcase 'user_update':this.handleUserUpdate(data)breakcase 'system_update':this.handleSystemUpdate(data)break}} catch (error) {console.error('Redis message error:', error)}}// 处理进程消息handleProcessMessage(message) {try {// 处理不同类型的消息switch (message.type) {case 'status':this.handleStatusUpdate(message.data)breakcase 'reload':this.handleReload(message.data)breakcase 'shutdown':this.handleShutdown(message.data)break}} catch (error) {console.error('Process message error:', error)}}// 启动服务器async start() {// 启动 HTTP 服务器this.server.listen(this.options.sslPort, () => {console.log(`HTTPS server running on port ${this.options.sslPort}`)})// 启动 HTTP 重定向const redirectServer = express().use((req, res) => {res.redirect(`https://${req.headers.host}${req.url}`)}).listen(this.options.port, () => {console.log(`HTTP redirect server running on port ${this.options.port}`)})}// 关闭服务器async shutdown() {console.log('Shutting down classroom server...')// 关闭 WebSocket 服务器this.wss.close()// 关闭 HTTP 服务器this.server.close()// 关闭 Redis 连接await this.redis.quit()// 清理资源await this.pool.shutdown()await this.queue.shutdown()await this.rooms.shutdown()await this.media.shutdown()console.log('Classroom server shutdown complete')}
}

房间管理

实现房间管理：

// room-manager.js
class RoomManager {constructor(options = {}) {this.options = {maxRooms: 1000,maxUsersPerRoom: 100,...options}this.rooms = new Map()this.stats = new Stats()this.initialize()}// 初始化房间管理器initialize() {// 监控房间数this.stats.gauge('rooms.total', () => this.rooms.size)this.stats.gauge('rooms.active', () => this.getActiveRooms().size)}// 创建房间async createRoom(options) {// 检查房间数限制if (this.rooms.size >= this.options.maxRooms) {throw new Error('Room limit reached')}// 创建房间const room = {id: generateId(),name: options.name,type: options.type,createdAt: Date.now(),users: new Map(),state: {whiteboard: [],chat: [],media: []},...options}this.rooms.set(room.id, room)this.stats.increment('rooms.created')return room}// 加入房间async joinRoom(roomId, connection) {const room = this.rooms.get(roomId)if (!room) {throw new Error('Room not found')}// 检查人数限制if (room.users.size >= this.options.maxUsersPerRoom) {throw new Error('Room is full')}// 添加用户room.users.set(connection.user.id, {connection,joinedAt: Date.now(),state: {}})// 更新连接connection.roomId = roomIdthis.stats.increment('room.users.joined')// 广播加入消息await this.broadcast(roomId, {type: 'user_joined',data: {user: connection.user.toJSON()}})return room}// 离开房间async leaveRoom(roomId, connection) {const room = this.rooms.get(roomId)if (!room) return// 移除用户room.users.delete(connection.user.id)// 更新连接delete connection.roomIdthis.stats.increment('room.users.left')// 如果房间为空,清理房间if (room.users.size === 0) {await this.cleanupRoom(roomId)}}// 广播消息async broadcast(roomId, message, excludeId = null) {const room = this.rooms.get(roomId)if (!room) return 0let count = 0room.users.forEach((user, userId) => {if (userId !== excludeId) {try {user.connection.send(message)count++} catch (error) {console.error('Broadcast error:', error)}}})this.stats.increment('room.messages.broadcast', count)return count}// 获取房间信息async getRoomInfo(roomId) {const room = this.rooms.get(roomId)if (!room) {throw new Error('Room not found')}return {id: room.id,name: room.name,type: room.type,users: Array.from(room.users.values()).map(user => ({id: user.connection.user.id,name: user.connection.user.name,role: user.connection.user.role,joinedAt: user.joinedAt})),state: room.state}}// 更新房间状态async updateRoomState(roomId, update) {const room = this.rooms.get(roomId)if (!room) {throw new Error('Room not found')}// 更新状态room.state = {...room.state,...update}// 广播更新await this.broadcast(roomId, {type: 'room_state_updated',data: {state: room.state}})return room.state}// 清理房间async cleanupRoom(roomId) {const room = this.rooms.get(roomId)if (!room) return// 保存房间数据await this.saveRoomData(room)// 删除房间this.rooms.delete(roomId)this.stats.increment('rooms.cleaned')}// 保存房间数据async saveRoomData(room) {// 实现数据持久化逻辑}// 获取活跃房间getActiveRooms() {const activeRooms = new Map()this.rooms.forEach((room, id) => {if (room.users.size > 0) {activeRooms.set(id, room)}})return activeRooms}// 获取统计信息getStats() {return {rooms: {total: this.rooms.size,active: this.getActiveRooms().size},...this.stats.getAll()}}// 关闭管理器async shutdown() {// 保存所有房间数据for (const room of this.rooms.values()) {await this.saveRoomData(room)}// 清理资源this.rooms.clear()}
}

用户管理

实现用户管理：

// user-manager.js
class UserManager {constructor(options = {}) {this.options = {sessionTimeout: 3600000, // 1 小时...options}this.users = new Map()this.sessions = new Map()this.stats = new Stats()this.initialize()}// 初始化用户管理器initialize() {// 启动会话清理setInterval(() => {this.cleanupSessions()}, 300000) // 5 分钟// 监控用户数this.stats.gauge('users.total', () => this.users.size)this.stats.gauge('users.online', () => this.getOnlineUsers().size)}// 认证用户async authenticate(req) {const token = this.extractToken(req)if (!token) {throw new Error('No token provided')}// 验证会话const session = this.sessions.get(token)if (!session) {throw new Error('Invalid session')}// 更新会话session.lastActivity = Date.now()return session.user}// 创建会话async createSession(user) {const token = generateToken()this.sessions.set(token, {user,createdAt: Date.now(),lastActivity: Date.now()})this.stats.increment('sessions.created')return token}// 清理会话cleanupSessions() {const now = Date.now()let cleaned = 0this.sessions.forEach((session, token) => {if (now - session.lastActivity > this.options.sessionTimeout) {this.sessions.delete(token)cleaned++}})if (cleaned > 0) {this.stats.increment('sessions.cleaned', cleaned)}}// 获取在线用户getOnlineUsers() {const onlineUsers = new Map()this.sessions.forEach(session => {onlineUsers.set(session.user.id, session.user)})return onlineUsers}// 获取用户信息async getUserInfo(userId) {const user = this.users.get(userId)if (!user) {throw new Error('User not found')}return {id: user.id,name: user.name,role: user.role,online: this.isUserOnline(userId)}}// 检查用户是否在线isUserOnline(userId) {return Array.from(this.sessions.values()).some(session => session.user.id === userId)}// 获取统计信息getStats() {return {users: {total: this.users.size,online: this.getOnlineUsers().size},sessions: {total: this.sessions.size},...this.stats.getAll()}}
}

媒体服务器

实现媒体服务器：

// media-server.js
class MediaServer {constructor(url, options = {}) {this.url = urlthis.options = {maxStreams: 1000,...options}this.streams = new Map()this.stats = new Stats()this.initialize()}// 初始化媒体服务器initialize() {// 监控���数量this.stats.gauge('streams.total', () => this.streams.size)this.stats.gauge('streams.active', () => this.getActiveStreams().size)}// 处理媒体流async handleStream(roomId, stream) {// 检查流数量限制if (this.streams.size >= this.options.maxStreams) {throw new Error('Stream limit reached')}// 创建流const mediaStream = {id: generateId(),roomId,type: stream.type,createdAt: Date.now(),state: 'new'}// 处理不同类型的流switch (stream.type) {case 'video':await this.handleVideoStream(mediaStream, stream)breakcase 'audio':await this.handleAudioStream(mediaStream, stream)breakcase 'screen':await this.handleScreenStream(mediaStream, stream)break}this.streams.set(mediaStream.id, mediaStream)this.stats.increment('streams.created')return mediaStream}// 处理视频流async handleVideoStream(mediaStream, stream) {// 实现视频流处理逻辑}// 处理音频流async handleAudioStream(mediaStream, stream) {// 实现音频流处理逻辑}// 处理屏幕共享流async handleScreenStream(mediaStream, stream) {// 实现屏幕共享处理逻辑}// 停止流async stopStream(streamId) {const stream = this.streams.get(streamId)if (!stream) return// 停止流stream.state = 'stopped'// 清理资源this.streams.delete(streamId)this.stats.increment('streams.stopped')}// 获取活跃流getActiveStreams() {const activeStreams = new Map()this.streams.forEach((stream, id) => {if (stream.state === 'active') {activeStreams.set(id, stream)}})return activeStreams}// 获取统计信息getStats() {return {streams: {total: this.streams.size,active: this.getActiveStreams().size},...this.stats.getAll()}}// 关闭服务器async shutdown() {// 停止所有流for (const stream of this.streams.values()) {await this.stopStream(stream.id)}}
}

部署配置

实现部署配置：

// config.js
module.exports = {// 服务器配置server: {port: process.env.PORT || 8080,sslPort: process.env.SSL_PORT || 8443,host: process.env.HOST || 'localhost'},// Redis 配置redis: {url: process.env.REDIS_URL || 'redis://localhost:6379',options: {retryStrategy: (times) => {return Math.min(times * 50, 2000)}}},// 媒体服务器配置media: {url: process.env.MEDIA_SERVER || 'localhost:8000',options: {maxStreams: 1000}},// 集群配置cluster: {workers: process.env.WORKERS || require('os').cpus().length,restartDelay: 1000},// 安全配置security: {ssl: {key: process.env.SSL_KEY || 'server.key',cert: process.env.SSL_CERT || 'server.cert'},cors: {origin: process.env.CORS_ORIGIN || '*'}},// 房间配置room: {maxRooms: 1000,maxUsersPerRoom: 100},// 用户配置user: {sessionTimeout: 3600000},// 监控配置monitor: {enabled: true,interval: 1000,historySize: 3600},// 日志配置log: {level: process.env.LOG_LEVEL || 'info',file: process.env.LOG_FILE || 'classroom.log'}
}