nodejs实现websocket

1.安装nodejs-websocket

npm install nodejs-websocket

2.websocket.js

// websocket 连接 导入nodejs-websocket模块
const ws = require("nodejs-websocket");

// 执行websocket处理连接方法
const createServer = () => {
  let server = ws.createServer((socket) => {
    //  链接上socket之后进行的处理逻
    //console.log("new socket...", socket.key, 111);
    let susccessObj = { type: "system", socket_key: socket.key };
    // 将拿到的唯一连接key返回给客户端 相当于http的端口的token  send只能发送字符串或缓冲区实例Buffer instance
    socket.send(JSON.stringify(susccessObj));
  

    //处理客户端发送过来的消息
    socket.on("text", function (data) {
      console.log("接收到的客户端消息", data);
    });
    socket.on("connect", function (code) {
      console.log("开启连接", code);
    });
    //监听关闭
    socket.on("close", function (code) {
      console.log("关闭连接", code);
    });
    //监听异常
    socket.on("error", function (code) {
      console.log("服务异常异常关闭", code);
    });
  });
  return server;
};


// 方式1：在本js文件中直接开启websocket的服务端口
let io = createServer().listen(3000);

// 方式2：
// module.exports = createServer();//导出在app.js中导入并调用---》开启websocket的服务端口

3.app.js

当使用方式1进行开启ws服务端口时  app.js不需要导入

当使用方式2时，app.js中需要导入并启动

// websocket相关
const wsSocket = require("./websocket.js");
let io = wsSocket;
io.listen(3000);