occaecatidicta/packages/omelox/lib/filters/handler/serial.ts

/**
 * Filter to keep request sequence.
 */
import {getLogger} from 'omelox-logger';
import * as taskManager from '../../common/manager/taskManager';
import {RouteRecord} from '../../util/constants';
import {IHandlerFilter} from '../../interfaces/IHandlerFilter';
import {HandlerCallback} from '../../common/service/handlerService';
import {FrontendOrBackendSession} from '../../server/server';
import * as path from 'path';

let logger = getLogger('omelox', path.basename(__filename));


export class SerialFilter implements IHandlerFilter {
    constructor(private timeout: number, private timeOutResponse?: any) {
    }

    /**
     * request serialization after filter
     */
    before(routeRecord: RouteRecord, msg: any, session: FrontendOrBackendSession, next: HandlerCallback) {
        taskManager.addTask(session.id, function (task) {
            (session as any).__serialTask__ = task;
            next(null);
        }, () => {
            logger.error('[serial filter] msg timeout, msg:' + JSON.stringify(msg) + ' routeRecord:' + JSON.stringify(routeRecord));
            if (this.timeOutResponse) {
                next(new Error('msg timeout:' + session.id + ' uid:' + (session.uid ? session.uid : '')), this.timeOutResponse);
            }
        }, this.timeout);
    }

    /**
     * request serialization after filter
     */
    after(err: Error, routeRecord: RouteRecord, msg: any, session: FrontendOrBackendSession, resp: any, next: HandlerCallback) {
        let task = (session as any).__serialTask__;
        if (task) {
            if (!task.done() && !err) {
                err = new Error('task time out. msg:' + JSON.stringify(msg));
            }
        }
        next(err);
    }
}