pomelo-upgrade/lib/connectors/hybrid/wsprocessor.js

网易Pomelo升级版本

const HttpServer = require('http').Server;
const EventEmitter = require('events').EventEmitter;
const util = require('util');
const WebSocketServer = require('ws').Server;

const ST_STARTED = 1;
const ST_CLOSED = 2;

/**
 * webSocket protocol processor
 */
const Processor = function()
{
    EventEmitter.call(this);
    this.httpServer = new HttpServer();

    this.wsServer = new WebSocketServer({server: this.httpServer});
    // todo 該方法要測試
    this.wsServer.on('connection', this.emit.bind(this, 'connection'));
    // this.wsServer.on('connection', function(socket)
    // {
    //     // emit socket to outside
    //     self.emit('connection', socket);
    // });
    this.state = ST_STARTED;
};
util.inherits(Processor, EventEmitter);

module.exports = Processor;

Processor.prototype.add = function(socket, data)
{
    if (this.state !== ST_STARTED)
    {
        return;
    }
    this.httpServer.emit('connection', socket);
    if (typeof socket.ondata === 'function')
    {
        // compatible with stream2
        socket.ondata(data, 0, data.length);
    }
    else
    {
        // compatible with old stream
        socket.emit('data', data);
    }
};

Processor.prototype.close = function()
{
    if (this.state !== ST_STARTED)
    {
        return;
    }
    this.state = ST_CLOSED;
    this.wsServer.close();
    this.wsServer = null;
    this.httpServer = null;
};