添加websocket连接

index.js

@@ -1,14 +1,24 @@
+console.log("当前执行文件：", __filename);
 // .env 中的配置写入 Node.js 的全局环境变量对象，下面的文件都可以直接使用
 require('dotenv').config();
 
 // 搭建http服务器
 const express = require("express");
+const http = require("http");
 
 const app = express();
 
+// 创建HTTP服务器
+const server = http.createServer(app);
+
 // 服务器解析 JSON 请求体
 app.use(express.json());
 
+// 防止控制台打印404
+app.get("/", (req, res) => {
+  res.send("chat-demo-server运行正常");
+});
+
 // 接入路由
 const loginRouter = require("./routes/login");
 const userRouter = require("./routes/user");
@@ -19,9 +29,14 @@ app.use("/api", loginRouter);
 app.use("/api", userRouter);
 app.use("/api", messageRouter);
 
+// 初始化WebSocket
+const initWebSocket = require("./websocket/index.js");
+initWebSocket(server);
+
 // 启动服务器，监听端口
-app.listen(3000, () => {
+server.listen(3000, () => {
     console.log("服务器启动成功：http://localhost:3000");
+    console.log("WebSocket服务器启动成功：ws://localhost:3000");
 });
 
 // index.js 就是 Node 后端的“总入口”：加载配置 → 创建 Express → 解析请求 → 加载路由 → 注册接口 → 启动 3000 端口服务器