web-dev-qa-db-fra.com

Écoute Express et WebSocket sur le même port

J'ai un app.js qui est utilisé pour déclencher deux événements lorsque des données POST sont reçues:

  1. Insérer des données POST dans une base de données
  2. Envoyer un message à un client à l'aide d'un WebSocket

Voici le app.js (seulement les lignes importantes)

var express = require('express');
var bodyParser = require('body-parser');
var server = require('./server');

var app = express();
var port = process.env.PORT || 3000;

app.post('/server', server);

app.listen(port, function(){
  console.log('Slack bot listening');
});

Et voici le server.js (seulement les lignes importantes)

var db = require('./DB');
var WebSocketServer = require('ws').Server;

var insertData = function(req, res){

    var wss = new WebSocketServer({server: server});
    console.log('WebSocketServer created');
    wss.on('connection', function(wss){
        wss.send(JSON.stringify('Socket open'));
    });
    wss.on('close', function(){
        console.log('WebServerSocket has been closed');
    });
};

module.exports = insertData;

Ce que je voudrais réaliser, c'est de configurer le WebSocketServer de manière à ce qu'il écoute le même port de l'application. J'ai pensé à passer le serveur var de app.js à server.js mais

  1. Je pense que ce n'est pas une manière élégante de le faire
  2. Je ne sais pas comment faire

Qu'en pensez-vous?

javascriptnode.jswebsocket
21
Mornor

Sur la base de votre code et de vos commentaires, voici un exemple très simple de la façon dont cela fonctionnerait ensemble.

Tout d'abord, le http-server.js - une application express typique, sauf que nous ne démarrons pas le serveur avec app.listen():

'use strict';

let fs = require('fs');
let express = require('express');
let app = express();
let bodyParser = require('body-parser');

app.use(bodyParser.json());

// Let's create the regular HTTP request and response
app.get('/', function(req, res) {

  console.log('Get index');
  fs.createReadStream('./index.html')
  .pipe(res);
});

app.post('/', function(req, res) {

  let message = req.body.message;
  console.log('Regular POST message: ', message);
  return res.json({

    answer: 42
  });
});

module.exports = app;

Maintenant, l'exemple ws-server.js, où nous créons le serveur WSS à partir d'un nœud natif http.createServer(). Maintenant, notez que c'est là que nous importons l'application et donnons à ce http.createServer natif l'instance d'application à utiliser.

Démarrez l'application avec PORT=8080 node ws-server.js:

'use strict';

let WSServer = require('ws').Server;
let server = require('http').createServer();
let app = require('./http-server');

// Create web socket server on top of a regular http server
let wss = new WSServer({

  server: server
});

// Also mount the app here
server.on('request', app);

wss.on('connection', function connection(ws) {

  ws.on('message', function incoming(message) {

    console.log(`received: ${message}`);

    ws.send(JSON.stringify({

      answer: 42
    }));
  });
});


server.listen(process.env.PORT, function() {

  console.log(`http/ws server listening on ${process.env.PORT}`);
});

Enfin, cet exemple index.html fonctionnera en créant à la fois une POST et une "demande" de socket et affichera la réponse:

<html>
<head>
  <title>WS example</title>
</head>

<body>
  <h2>Socket message response: </h2>
  <pre id="response"></pre>
  <hr/>
  <h2>POST message response: </h2>
  <pre id="post-response"></pre>
  <script>

  // Extremely simplified here, no error handling or anything
document.body.onload = function() {

    'use strict';

  // First the socket requesta
  function socketExample() {
    console.log('Creating socket');
    let socket = new WebSocket('ws://localhost:8080/');
    socket.onopen = function() {

      console.log('Socket open.');
      socket.send(JSON.stringify({message: 'What is the meaning of life, the universe and everything?'}));
      console.log('Message sent.')
    };
    socket.onmessage = function(message) {

      console.log('Socket server message', message);
      let data = JSON.parse(message.data);
      document.getElementById('response').innerHTML = JSON.stringify(data, null, 2);
    };
  }

  // Now the simple POST demo
  function postExample() {

    console.log('Creating regular POST message');

    fetch('/', {  
      method: 'post',  
      headers: {  
        "Content-type": "application/json"  
      },  
      body: JSON.stringify({message: 'What is the meaning of post-life, the universe and everything?'})  
    })
    .then(response => response.json())  
    .then(function (data) {  

      console.log('POST response:', data);
      document.getElementById('post-response').innerHTML = JSON.stringify(data, null, 2);   
    })  
    .catch(function (error) {  
      console.log('Request failed', error);  
    });   
  }

  // Call them both;

  socketExample();
  postExample();
}
  </script>
</body>
</html>

Notez que vous aurez besoin d'un navigateur assez récent, qui a à la fois des API WebSocket et fetch pour cette partie côté client, mais ce n'est pas pertinent de toute façon, il vous donne juste l'essentiel.

41
Zlatko