IOTConnect-Web/node_modules/websocket-stream/server.js
2024-05-09 09:49:52 +08:00

30 lines
611 B
JavaScript

'use strict'
var WebSocketServer = require('ws').Server
var stream = require('./stream')
class Server extends WebSocketServer{
constructor(opts, cb) {
super(opts)
var proxied = false
this.on('newListener', function(event) {
if (!proxied && event === 'stream') {
proxied = true
this.on('connection', function(conn, req) {
this.emit('stream', stream(conn, opts), req)
})
}
})
if (cb) {
this.on('stream', cb)
}
}
}
module.exports.Server = Server
module.exports.createServer = function(opts, cb) {
return new Server(opts, cb)
}