当前位置：首页 > news >正文

Node.js实现WebSocket教程

news 2025/12/25 15:54:50

Node.js实现WebSocket教程

1. WebSocket简介

WebSocket是一种在单个TCP连接上提供全双工通信的协议，允许服务器和客户端之间进行实时、双向通信。本教程将详细讲解如何在Node.js中实现WebSocket。

2. 技术选型

我们将使用ws库来实现WebSocket服务器，并结合express创建Web应用。

2.1 安装依赖

# 创建项目目录
mkdir nodejs-websocket-demo
cd nodejs-websocket-demo# 初始化项目
npm init -y# 安装依赖
npm install express ws uuid

依赖说明：

express: Web应用框架
ws: WebSocket服务器实现
uuid: 生成唯一标识符

3. 项目结构

nodejs-websocket-demo/
│
├── public/
│   ├── index.html
│   └── client.js
├── server.js
└── package.json

4. 详细实现

4.1 WebSocket服务器 (server.js)

const express = require('express');
const http = require('http');
const WebSocket = require('ws');
const { v4: uuidv4 } = require('uuid');const app = express();
const server = http.createServer(app);
const wss = new WebSocket.Server({ server });// 存储客户端连接
const clients = new Map();// 静态文件服务
app.use(express.static('public'));// WebSocket连接处理
wss.on('connection', (ws) => {// 为每个客户端分配唯一IDconst clientId = uuidv4();// 存储客户端连接clients.set(clientId, ws);// 发送客户端IDws.send(JSON.stringify({type: 'connection',clientId: clientId}));// 广播新用户连接broadcast({type: 'userJoined',clientId: clientId,message: `用户 ${clientId} 已加入聊天`}, clientId);// 消息处理ws.on('message', (message) => {try {const parsedMessage = JSON.parse(message);switch(parsedMessage.type) {case 'chat':handleChatMessage(clientId, parsedMessage);break;case 'typing':handleTypingNotification(clientId, parsedMessage);break;default:console.log('未知消息类型:', parsedMessage.type);}} catch (error) {console.error('消息解析错误:', error);}});// 连接关闭处理ws.on('close', () => {// 移除客户端clients.delete(clientId);// 广播用户离开broadcast({type: 'userLeft',clientId: clientId,message: `用户 ${clientId} 已离开聊天`}, clientId);});
});// 聊天消息处理
function handleChatMessage(senderId, message) {broadcast({type: 'chat',clientId: senderId,message: message.message}, senderId);
}// 输入状态通知
function handleTypingNotification(senderId, message) {broadcast({type: 'typing',clientId: senderId,isTyping: message.isTyping}, senderId);
}// 广播消息
function broadcast(message, excludeClientId = null) {clients.forEach((client, clientId) => {if (client.readyState === WebSocket.OPEN && clientId !== excludeClientId) {client.send(JSON.stringify(message));}});
}// 服务器启动
const PORT = process.env.PORT || 3000;
server.listen(PORT, () => {console.log(`WebSocket服务器运行在 ${PORT} 端口`);
});

4.2 客户端HTML (public/index.html)

<!DOCTYPE html>
<html lang="zh-CN">
<head><meta charset="UTF-8"><title>WebSocket聊天室</title><style>body { font-family: Arial, sans-serif; max-width: 600px; margin: 0 auto; padding: 20px; }#chat-messages {height: 300px;overflow-y: scroll;border: 1px solid #ccc;padding: 10px;margin-bottom: 10px;}</style>
</head>
<body><div id="chat-messages"></div><input type="text" id="message-input" placeholder="输入消息"><button id="send-btn">发送</button><script src="client.js"></script>
</body>
</html>

4.3 客户端JavaScript (public/client.js)

const socket = new WebSocket('ws://localhost:3000');
let clientId = null;const chatMessages = document.getElementById('chat-messages');
const messageInput = document.getElementById('message-input');
const sendBtn = document.getElementById('send-btn');// WebSocket事件处理
socket.addEventListener('open', (event) => {appendMessage('系统', '连接成功');
});socket.addEventListener('message', (event) => {const message = JSON.parse(event.data);switch(message.type) {case 'connection':clientId = message.clientId;appendMessage('系统', `您的ID是: ${clientId}`);break;case 'chat':appendMessage(message.clientId, message.message);break;case 'userJoined':case 'userLeft':appendMessage('系统', message.message);break;case 'typing':handleTypingNotification(message);break;}
});socket.addEventListener('close', (event) => {appendMessage('系统', '连接已关闭');
});// 发送消息
sendBtn.addEventListener('click', sendMessage);
messageInput.addEventListener('keypress', (e) => {if (e.key === 'Enter') {sendMessage();}
});function sendMessage() {const message = messageInput.value.trim();if (message) {socket.send(JSON.stringify({type: 'chat',message: message}));appendMessage(clientId, message);messageInput.value = '';}
}// 消息追加到聊天窗口
function appendMessage(sender, text) {const messageEl = document.createElement('div');messageEl.innerHTML = `<strong>${sender}:</strong> ${text}`;chatMessages.appendChild(messageEl);chatMessages.scrollTop = chatMessages.scrollHeight;
}// 处理打字状态通知
function handleTypingNotification(message) {// 可以在这里实现打字状态提示
}

5. 运行项目

# 启动服务器
node server.js# 访问 http://localhost:3000

6. 功能特点

实时双向通信
唯一客户端标识
广播消息机制
连接/断开事件处理
聊天消息和系统通知

7. 性能与扩展建议

生产环境考虑使用Redis等外部存储管理WebSocket连接
增加身份验证机制
实现消息持久化
使用负载均衡

8. 安全注意事项

使用WSS（WebSocket Secure）
实现连接速率限制
验证和过滤消息内容
防止跨站WebSocket劫持

Node.js实现WebSocket教程

Node.js实现WebSocket教程 1. WebSocket简介 WebSocket是一种在单个TCP连接上提供全双工通信的协议，允许服务器和客户端之间进行实时、双向通信。本教程将详细讲解如何在Node.js中实现WebSocket。 2. 技术选型我们将使用ws库来实现WebSocket服务器，…...

编程日记 2024/12/10 2:33:21

Docker Compose实战一（轻松部署 Nginx）

通过过前面的文章（Docker Compose基础语法）你已经掌握基本语法和常用指令认识到Docker Compose作为一款强大工具的重要性，它极大地简化了多容器Docker应用程序的部署与管理流程。本文将详细介绍如何使用 Docker Compose 部署 Nginx&#xff0…...

编程日记 2024/12/10 2:31:19

hive分区分桶、数据倾斜总结

一、hive的基本概念 hive是一个构建在hadoop上的数据仓库工具，可以将结构化的数据文件映射为一张数据库表并提供数据查询功能二、hive的特点 （1）数据是存储在hdfs上 （2）底层是将sql转换为MapReduce任务进行计算 …...

编程日记 2024/12/10 2:26:12

unity打包到安卓帧率降低

这个问题遇到过很多次了我的做法就是直接设置Application.targetFrameRate60 参考...

编程日记 2024/12/10 2:25:11

【Python3】装饰器自动更新缓存

自动更新缓存的需求场景在某些应用中，我们可能需要定期从外部数据源（如 Redis 或者远程接口）拉取数据，并将其缓存在内存中。当有其他代码需要访问这些数据时，可以立刻从内存获取最新数据，而无需每次都进行…...

编程日记 2024/12/10 2:23:07

通过EPEL 仓库，在 CentOS 7 上安装 OpenResty

通过EPEL 仓库，在 CentOS 7 上安装 OpenResty 通过EPEL 仓库，在 CentOS 7 上安装 OpenResty步骤 1: 安装 EPEL 仓库步骤 2: 安装 OpenResty步骤 3: 启动 OpenResty步骤 4: 设置开机自启步骤 5: 验证安装说明通过EPEL 仓库，在 CentOS 7 上安装…...

编程日记 2024/12/10 2:21:04

[RabbitMQ] RabbitMQ常见应用问题

🌸个人主页:https://blog.csdn.net/2301_80050796?spm1000.2115.3001.5343 🏵️热门专栏: 🧊 Java基本语法(97平均质量分)https://blog.csdn.net/2301_80050796/category_12615970.html?spm1001.2014.3001.5482 🍕 Collection与…...

编程日记 2024/12/10 2:19:02

每日速记10道java面试题13-MySQL篇

其他资料每日速记10道java面试题01-CSDN博客每日速记10道java面试题02-CSDN博客每日速记10道java面试题03-CSDN博客每日速记10道java面试题04-CSDN博客每日速记10道java面试题05-CSDN博客每日速记10道java面试题06-CSDN博客每日速记10道java面试题07-CSDN博客每…...

编程日记 2024/12/10 2:11:53

乐鑫科技嵌入式面试题及参考答案（3万字长文）

嵌入式开发为什么用 C 语言，而不用 C++ 语言？在嵌入式开发中，C 语言被广泛使用而 C++ 相对少用有以下一些原因。首先，C 语言具有更高的效率。嵌入式系统通常资源受限，包括处理器速度、内存容量等。C 语言的代码生成效率高，能够生成紧凑的机器码，占用较少的内存空间和处…...

编程日记 2024/12/10 2:07:48

Leetcode 每日一题 56.合并区间

目录问题描述示例示例 1 示例 2 问题分析算法设计步骤 1：排序步骤 2：合并区间步骤 3：返回结果过题图片代码实现复杂度分析题目链接结语问题描述给定一个区间数组 intervals，其中每个区间由两个整数 s…...

编程日记 2024/12/10 2:05:44

【Vue】v-model、ref获取DOM

目录 v-moel v-model的原理 v-model用在组件标签上方式 defineModel()简写 ref属性获取原生DOM 获取组件实例 nextTick() v-moel v-model：双向数据绑定指令数据变了，视图跟着变（数据驱动视图）视图变了，数…...

编程日记 2024/12/10 2:04:41

Python 类的设计（以植物大战僵尸为例）

关于类的设计——以植物大战僵尸为例一、设计类需满足的三要素1. 类名2. 属性和方法二、以植物大战僵尸的为例的类的设计1. 尝试分类2. 创建对象调用类的属性和方法*【代码二】*3. 僵尸的继承三、代码实现一、设计类需满足的三要素 1. 类名类名：某类事物的名…...

编程日记 2024/12/10 1:59:34

python中权重剪枝，低秩分解，量化技术代码

目录 python中权重剪枝，低秩分解，量化技术代码权重剪枝低秩分解 scipy 量化技术 python中权重剪枝，低秩分解，量化技术代码权重剪枝权重剪枝可以通过PyTorch的torch.nn.utils.prune模块实现。以下是一个简单的例子： import torch import torch.nn as nn impor…...

编程日记 2024/12/10 1:58:33

调用matlab用户自定义的function函数时，有多个输出变量只输出第一个变量

很多朋友在使用matlab时，会使用或自己编辑多个function函数，来满足自己对任务处理的要求，但是在调用function函数时，会出现这个问题：调用matlab用户自定义的function函数时，有多个输出变量只输出第一个变量…...

编程日记 2024/12/10 1:56:31

RabbitMQ七种工作模式之简单模式, 工作队列模式, 发布订阅模式, 路由模式, 通配符模式

文章目录一. Simple(简单模式)公共代码:生产者:消费者: 二. Work Queue(工作队列模式)公共代码:生产者:消费者1, 消费者2(代码相同): 三. Publish/Subscribe(发布/订阅模式)公共代码:生产者:消费者: 四. Routing(路由模式)公共代码:消费者: 五. Topics(通配符模式)公共代码:生…...

编程日记 2024/12/10 1:54:29

Win10安装kafka并用C#调用

kafka安装 jdk、kafka版本如下，zookeeper使用kafka自带版本安装包下载位置:https://download.csdn.net/download/henreash/90087368 (赚点csdn下载资源分) 安装jdk后，解压kafka压缩包，修改配置文件： kafka_2.13-3.9.0\config\…...

编程日记 2024/12/10 1:49:18

高级架构二 Git基础到高级

一 Git仓库的基本概念和流程什么是版本库？版本库又名仓库，英文名repository,你可以简单的理解一个目录，这个目录里面的所有文件都可以被Git管理起来，每个文件的修改，删除，Git都能跟踪，以便任何…...

编程日记 2024/12/10 1:47:15

深入解析二叉树算法

引言二叉树（Binary Tree）作为数据结构中的一种重要形式，在计算机科学的诸多领域中得到了广泛应用。从文件系统到表达式解析，再到搜索和排序，二叉树都扮演着关键角色。本文将从二叉树的基础概念出发，详细探讨其各种算法及其应用，并提供相关代码示例，旨在为读者建立扎实…...

编程日记 2024/12/10 1:46:14

如何解决maven项目使用Ctrl + /添加注释时的顶格问题

一、问题描述相信后端开发的程序员一定很熟悉IDEA编译器和Maven脚手架，使用IDEA新建一个Maven工程，通过SpringBoot快速构建Spring项目。在Spring项目pom.xml文件中想添加注释，快捷键Ctrl /，但是总是顶格书写。想保证缩进统一…...

编程日记 2024/12/10 1:44:11

总结的一些MySql面试题

目录一：基础篇二：索引原理和SQL优化三：事务原理四：缓存策略一：基础篇 1：定义：按照数据结构来组织、存储和管理数据的仓库；是一个长期存储在计算机内的、有组织的、可共享的…...

编程日记 2024/12/10 1:40:06

XML Group端口详解

在XML数据映射过程中，经常需要对数据进行分组聚合操作。例如，当处理包含多个物料明细的XML文件时，可能需要将相同物料号的明细归为一组，或对相同物料号的数量进行求和计算。传统实现方式通常需要编写脚本代码，增加了开…...

编程新知 2025/12/24 13:52:56

阿里云ACP云计算备考笔记 (5)——弹性伸缩

目录第一章概述第二章弹性伸缩简介 1、弹性伸缩 2、垂直伸缩 3、优势 4、应用场景 ① 无规律的业务量波动 ② 有规律的业务量波动 ③ 无明显业务量波动 ④ 混合型业务 ⑤ 消息通知 ⑥ 生命周期挂钩 ⑦ 自定义方式 ⑧ 滚的升级 5、使用限制第三章主要定义 …...

编程新知 2025/12/15 15:15:10

3.3.1_1 检错编码（奇偶校验码）

从这节课开始，我们会探讨数据链路层的差错控制功能，差错控制功能的主要目标是要发现并且解决一个帧内部的位错误，我们需要使用特殊的编码技术去发现帧内部的位错误，当我们发现位错误之后，通常来说有两种解决方案。第一…...

编程新知 2025/12/18 15:23:39

Mybatis逆向工程，动态创建实体类、条件扩展类、Mapper接口、Mapper.xml映射文件

今天呢，博主的学习进度也是步入了Java Mybatis 框架，目前正在逐步杨帆旗航。那么接下来就给大家出一期有关 Mybatis 逆向工程的教学，希望能对大家有所帮助，也特别欢迎大家指点不足之处，小生很乐意接受正确的建议&…...

编程新知 2025/12/19 13:41:56

【JVM】- 内存结构

引言 JVM：Java Virtual Machine 定义：Java虚拟机，Java二进制字节码的运行环境好处： 一次编写，到处运行自动内存管理，垃圾回收的功能数组下标越界检查（会抛异常，不会覆盖到其他代码…...

编程新知 2025/12/18 20:09:12

CMake 从 GitHub 下载第三方库并使用

有时我们希望直接使用 GitHub 上的开源库，而不想手动下载、编译和安装。可以利用 CMake 提供的 FetchContent 模块来实现自动下载、构建和链接第三方库。 FetchContent 命令官方文档✅ 示例代码我们将以 fmt 这个流行的格式化库为例，演示如何：使用 FetchContent 从 GitH…...

编程新知 2025/12/11 5:12:21

OpenPrompt 和直接对提示词的嵌入向量进行训练有什么区别

OpenPrompt 和直接对提示词的嵌入向量进行训练有什么区别直接训练提示词嵌入向量的核心区别您提到的代码： prompt_embedding = initial_embedding.clone().requires_grad_(True) optimizer = torch.optim.Adam([prompt_embedding...

编程新知 2025/7/4 0:50:01