node-qps/dist/collection/shimmer.js

Test Node Application QPS

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
var Module = require("module");
var report_1 = require("./report");
var wrapMethod;
exports.wrapMethod = wrapMethod;
exports.wrapMethod = wrapMethod = function (nodule, method, wrapper) {
    var orginal = nodule[method];
    if (!orginal)
        return;
    if (orginal["__unwrap"])
        return;
    var wraped = wrapper(orginal);
    // nodule[method] = 
    var __unwrap;
    __unwrap = function () {
        nodule[method] = orginal;
    };
    wraped["orginal"] = orginal;
    wraped["__unwrap"] = __unwrap;
    nodule[method] = wraped;
};
var wrapModuleLoad;
exports.wrapModuleLoad = wrapModuleLoad;
exports.wrapModuleLoad = wrapModuleLoad = function (counter) {
    wrapMethod(Module, "_load", function _wrapModuleLoad(orginal) {
        return function load(file) {
            if (file === "http" || file === "https") {
                return wrapHttp(counter, orginal.call(this, file));
            }
            else {
                return orginal.apply(this, arguments);
            }
        };
    });
};
var wrapHttp;
wrapHttp = function (counter, http) {
    wrapMethod(http.Server.prototype, "on", function (addListener) {
        var addListenerWraper;
        addListenerWraper = function (type, listener) {
            if (type === "request" && typeof listener === "function") {
                return addListener.call(this, type, warpListener(counter, listener));
            }
            return addListener.apply(this, arguments);
        };
        return addListenerWraper;
    });
    wrapMethod(http.Server.prototype, "addListener", function (addListener) {
        var addListenerWraper;
        addListenerWraper = function (type, listener) {
            if (type === "request" && typeof listener === "function") {
                return addListener.call(this, type, warpListener(counter, listener));
            }
            return addListener.apply(this, arguments);
        };
        return addListenerWraper;
    });
    return http;
};
var warpListener;
warpListener = function (counter, listener) {
    var newListener;
    newListener = function (request, response) {
        counter.addRequestCount(1);
        var url = request.originalUrl || request.url;
        var method = request.method;
        var resWrite = response.write;
        var resEnd = response.end;
        response.write = function (chunk, encoding, callback) {
            if (url === "/QPS" && method === "GET") {
                if (typeof chunk === 'function') {
                    callback = chunk;
                    chunk = null;
                }
                else if (typeof encoding === 'function') {
                    callback = encoding;
                    encoding = null;
                }
                return resWrite.call(this, null, encoding, callback);
            }
            return resWrite.apply(this, arguments);
        };
        response.end = function (chunk, encoding, callback) {
            if (url === "/QPS" && method === "GET") {
                if (typeof chunk === 'function') {
                    callback = chunk;
                    chunk = null;
                }
                else if (typeof encoding === 'function') {
                    callback = encoding;
                    encoding = null;
                }
                return resEnd.call(this, report_1.reportQPS(counter), encoding, callback);
            }
            return resEnd.apply(this, arguments);
        };
        response.once('finish', function onResponseFinish() {
            counter.addResponseCount(1);
        });
        return listener.apply(this, arguments);
    };
    return newListener;
};