lux-framework/src/packages/router/route/action/enhancers/track-perf.js

Build scalable, Node.js-powered REST APIs with almost no code.

// @flow
import { FINAL_HANDLER } from '../constants';
import getActionName from '../utils/get-action-name';
import getControllerName from '../utils/get-controller-name';
import type { Action } from '../interfaces';

/**
 * @private
 */
export default function trackPerf<T, U: Action<T>>(action: U): Action<T> {
  // eslint-disable-next-line func-names
  const trackedAction = async function (...args: Array<any>) {
    const [req, res] = args;
    const start = Date.now();
    const result = await action(...args);
    let { name } = action;
    let type = 'middleware';

    if (name === FINAL_HANDLER) {
      type = 'action';
      name = getActionName(req);
    } else if (!name) {
      name = 'anonymous';
    }

    res.stats.push({
      type,
      name,
      duration: Date.now() - start,
      controller: getControllerName(req)
    });

    return result;
  };

  Reflect.defineProperty(trackedAction, 'name', {
    value: action.name
  });

  return trackedAction;
}