Igor Lins e Silva
31 lines
649 B
JavaScript
31 lines
649 B
JavaScript
const fp = require('fastify-plugin');
|
|
const WebSocket = require('ws');
|
|
|
|
module.exports = fp((fastify, opts, next) => {
|
|
|
|
const wss = new WebSocket.Server({
|
|
server: fastify.server
|
|
});
|
|
|
|
const ws = new WebSocket('ws://127.0.0.1:7001', [], {
|
|
perMessageDeflate: false
|
|
});
|
|
|
|
ws.on('open', function open() {
|
|
console.log('router connected');
|
|
});
|
|
|
|
fastify.decorate('wss', wss);
|
|
fastify.decorate('ws', ws);
|
|
|
|
fastify.addHook('onClose', (fastify, done) => {
|
|
fastify.ws.close(done);
|
|
fastify.wss.close(done);
|
|
});
|
|
|
|
next();
|
|
}, {
|
|
fastify: '2.x',
|
|
name: 'fastify-websocket'
|
|
});
|