pomelo-upgrade/lib/connectors/common/handler.js

网易Pomelo升级版本

'use strict';
const protocol = require('pomelo-protocol');
const Package = protocol.Package;
const logger = require('pomelo-logger-upgrade').getLogger('pomelo', __filename);

const handlers = {};

const ST_INITED = 0;
const ST_WAIT_ACK = 1;
const ST_WORKING = 2;
const ST_CLOSED = 3;

const handleHandshake = function(socket, pkg)
{
    if (socket.state !== ST_INITED)
    {
        return;
    }
    try
    {
        socket.emit('handshake', JSON.parse(protocol.strdecode(pkg.body)));
    }
    catch (ex)
    {
        socket.emit('handshake', {});
    }
};

const handleHandshakeAck = function(socket, pkg)
{
    if (socket.state !== ST_WAIT_ACK)
    {
        return;
    }
    socket.state = ST_WORKING;
    socket.emit('heartbeat');
};

const handleHeartbeat = function(socket, pkg)
{
    if (socket.state !== ST_WORKING)
    {
        return;
    }
    socket.emit('heartbeat');
};

const handleData = function(socket, pkg)
{
    if (socket.state !== ST_WORKING)
    {
        return;
    }
    socket.emit('message', pkg);
};

handlers[Package.TYPE_HANDSHAKE] = handleHandshake;
handlers[Package.TYPE_HANDSHAKE_ACK] = handleHandshakeAck;
handlers[Package.TYPE_HEARTBEAT] = handleHeartbeat;
handlers[Package.TYPE_DATA] = handleData;

const handle = function(socket, pkg)
{
    const handler = handlers[pkg.type];
    if (handler)
    {
        handler(socket, pkg);
    }
    else
    {
        logger.error('could not find handle invalid data package.');
        socket.disconnect();
    }
};

module.exports = handle;