本文详解如何在 node.js 后端中正确使用 socket.io 实现定时更新通知,避免因错误地在每次请求中重复注册连接事件而导致消息无法发送的问题。
在构建基于 Socket.IO 的实时图像更新应用时,一个常见却容易被忽视的陷阱是:将 io.on(“connection”, …) 放入异步操作链(如 fetch 的 .then())中反复执行。这会导致每次生成新 SVG 文件时都尝试重新绑定连接监听器,而非向已连接的客户端广播消息——结果就是 socket.emit() 看似执行了,实则“发给了不存在的、尚未建立的 socket”,自然客户端收不到任何响应。
✅ 正确做法:服务端统一管理连接,按需广播
Socket.IO 服务器应在启动时一次性注册 connection 事件监听器,并维护一个活跃客户端 socket 的引用集合(如数组或 Map),后续所有业务逻辑(如定时获取 Gist 并生成 SVG)只需遍历该集合调用 socket.emit() 即可。
以下是优化后的服务端核心代码(关键改动已加注释):
const PORT = 8080;
const SOCKET_PORT = 3000;
const interval = 5000;
const io = new Server(SOCKET_PORT);
// ✅ 步骤1:全局维护已连接的客户端 socket 列表
const clients = [];
// ✅ 步骤2:仅在服务启动时注册一次 connection 监听器
io.on("connection", (socket) => {
console.log("✅ 新客户端已连接");
clients.push(socket);
// 可选:监听客户端断开,自动清理
socket.on("disconnect", () => {
const index = clients.indexOf(socket);
if (index > -1) {
clients.splice(index, 1);
console.log("⚠️ 客户端已断开,当前在线数:", clients.length);
}
});
// ✅ 处理客户端发来的自定义消息(保持原有逻辑)
socket.on("customMessage", (arg) => {
console.log("? 收到客户端消息:", arg);
});
});
// ✅ 步骤3:业务函数只负责生成文件 + 广播,不再触碰 io 连接逻辑
const getRandomGists = () => {
fetch(rndGist)
.then((res) => res.text())
.then((body) => {
const { window } = new JSDOM(body);
const { document } = window;
const codeBlock = document.querySelectorAll(".blob-code-inner");
console.log(`? 获取 Gist: ${rndGist}. 时间: ${new Date().toLocaleTimeString()}`);
const fauxCode = new FauxCode(codeBlock, options);
fs.writeFileSync(filename, fauxCode.render());
console.log(`? SVG 已写入: ${filename}`);
// ✅ 关键:向所有在线客户端广播更新事件
clients.forEach((socket) => {
try {
socket.emit("imageUpdate", { timestamp: Date.now(), message: "Image updated" });
} catch (err) {
console.warn("⚠️ 向某客户端 emit 失败,可能已断开:", err.message);
}
});
})
.catch((err) => {
console.error("❌ 获取或处理 Gist 失败:", err);
});
};
// ✅ 步骤4:启动定时任务(注意:无需再包裹在 getRandomGists 内部)
getRandomGists(); // 首次立即执行
setInterval(getRandomGists, interval);
// 静态资源与 HTTP 服务
app.use(express.static("public"));
app.listen(PORT, () => console.log(`? HTTP 服务运行于 http://localhost:${PORT}`));
? 客户端适配建议
你的客户端逻辑基本正确,但有两处可增强健壮性: