pomelo-upgrade/lib/filters/handler/serial.js

网易Pomelo升级版本

'use strict';
/**
 * Filter to keep request sequence.
 */
const logger = require('pomelo-logger-upgrade').getLogger('pomelo', __filename);
const taskManager = require('../../common/manager/taskManager');

module.exports = function(timeout)
{
    return new Filter(timeout);
};

const Filter = function(timeout)
{
    this.timeout = timeout;
};

/**
 * request serialization after filter
 */
Filter.prototype.before = function(msg, session, next)
{
    taskManager.addTask(session.id, function(task)
    {
        session.__serialTask__ = task;
        next();
    }, function()
    {
        logger.error(`[serial filter] msg timeout, msg:${JSON.stringify(msg)}`);
    }, this.timeout);
};

/**
 * request serialization after filter
 */
Filter.prototype.after = function(err, msg, session, resp, next)
{
    const task = session.__serialTask__;
    if (task)
    {
        if (!task.done() && !err)
        {
            err = new Error(`task time out. msg:${JSON.stringify(msg)}`);
        }
    }
    next(err);
};