websocket路径参数

ws://example.com/path/param1/param2

const WebSocket = require('ws');
const wss = new WebSocket.Server({ port: 8080 });

wss.on('connection', (ws, request) => {
  const path = new URL(request.url, `ws://${request.headers.host}`).pathname;
  
  if (path.startsWith('/chat/')) {
    const roomId = path.split('/')[2];
    ws.on('message', (message) => {
      console.log(`Message in room ${roomId}: ${message}`);
      // Broadcast message to all clients in the same room
      wss.clients.forEach(client => {
        if (client.readyState === WebSocket.OPEN && client.roomId === roomId) {
          client.send(message);
        }
      });
    });
    ws.roomId = roomId;
  } else {
    ws.close(1008, 'Invalid path');
  }
});

const express = require('express');
const http = require('http');
const WebSocket = require('ws');

const app = express();
const server = http.createServer(app);
const wss = new WebSocket.Server({ server });

app.use(express.static('public'));

wss.on('connection', (ws, request) => {
  const path = new URL(request.url, `ws://${request.headers.host}`).pathname;
  
  if (path.startsWith('/chat/')) {
    const roomId = path.split('/')[2];
    ws.on('message', (message) => {
      console.log(`Message in room ${roomId}: ${message}`);
      wss.clients.forEach(client => {
        if (client.readyState === WebSocket.OPEN && client.roomId === roomId) {
          client.send(message);
        }
      });
    });
    ws.roomId = roomId;
  } else {
    ws.close(1008, 'Invalid path');
  }
});

server.listen(8080, () => {
  console.log('Server started on http://localhost:8080');
});