将socket.io传递给节点js中的路由器模块

const express = require('express');
const http = require('http');
const socketIo = require('socket.io');

const app = express();
const server = http.createServer(app);
const io = socketIo(server);

// 将 io 实例传递给路由器模块
const router = require('./routes')(io);

app.use('/', router);

server.listen(3000, () => {
  console.log('Server is running on port 3000');
});

module.exports = (io) => {
  const express = require('express');
  const router = express.Router();

  router.get('/', (req, res) => {
    res.sendFile(__dirname + '/index.html');
  });

  io.on('connection', (socket) => {
    console.log('A user connected');

    socket.on('disconnect', () => {
      console.log('User disconnected');
    });

    // 处理自定义事件
    socket.on('chat message', (msg) => {
      io.emit('chat message', msg); // 广播消息
    });
  });

  return router;
};

const io = socketIo(server, {
  cors: {
    origin: '*',
    methods: ['GET', 'POST']
  }
});