一个简单的聊天服务器

以下代码设置了一个WebSocket服务器和一个Express服务器，并连接到MongoDB数据库。它定义了一个Barrage模式，并处理WebSocket连接和POST请求以发送弹幕。

当建立WebSocket连接时，服务器侦听传入的消息。如果消息内容长度在1到50个字符之间，它将弹幕保存到数据库并向所有客户端广播。

当向’/barrage’端点发出POST请求时，服务器使用请求正文内容创建WebSocket消息，并将其发送到WebSocket服务器，后者将其广播到所有客户端。

总的来说，这段代码设置了一个简单的聊天服务器，允许客户端实时发送和接收消息。

// Import necessary modules
const WebSocket = require('ws');
const express = require('express');
const bodyParser = require('body-parser');
const mongoose = require('mongoose');

// Set up WebSocket server
const wss = new WebSocket.Server({ port: 8080 });

// Set up Express server
const app = express();
app.use(bodyParser.json());

// Connect to MongoDB database
mongoose.connect('mongodb://localhost/barrage', { useNewUrlParser: true });
const db = mongoose.connection;
db.on('error', console.error.bind(console, 'MongoDB connection error:'));

// Define Barrage schema
const barrageSchema = new mongoose.Schema({
  content: String,
  timestamp: Date
});
const Barrage = mongoose.model('Barrage', barrageSchema);

// Handle WebSocket connection
wss.on('connection', function connection(ws) {
  ws.on('message', function incoming(message) {
    // Validate message content
    const content = JSON.parse(message).content;
    if (content.length > 0 && content.length <= 50) {
      // Save barrage to database
      const barrage = new Barrage({
        content: content,
        timestamp: new Date()
      });
      barrage.save(function(err) {
        if (err) return console.error(err);
      });
      // Broadcast barrage to all clients
      wss.clients.forEach(function each(client) {
        if (client.readyState === WebSocket.OPEN) {
          client.send(message);
        }
      });
    }
  });
});

// Handle POST request to send barrage
app.post('/barrage', function(req, res) {
  // Create WebSocket message
  const message = JSON.stringify({
    content: req.body.content
  });
  // Send message to WebSocket server
  wss.clients.forEach(function each(client) {
    if (client.readyState === WebSocket.OPEN) {
      client.send(message);
    }
  });
  res.sendStatus(200);
});

// Start Express server
app.listen(3000, function() {
  console.log('Barrage server listening on port 3000');
});