@yhiot/utils/lib/koamw/xResponseTime.js

基本工具集

"use strict";

var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.default = xResponseTime;

var _regenerator = _interopRequireDefault(require("@babel/runtime/regenerator"));

var _asyncToGenerator2 = _interopRequireDefault(require("@babel/runtime/helpers/asyncToGenerator"));

/**
 * 将api执行时间返回.
 * @param {*} ctx
 * @param {*} next
 */
function xResponseTime(_x, _x2) {
  return _xResponseTime.apply(this, arguments);
}

function _xResponseTime() {
  _xResponseTime = (0, _asyncToGenerator2.default)( /*#__PURE__*/_regenerator.default.mark(function _callee(ctx, next) {
    var start, ms;
    return _regenerator.default.wrap(function _callee$(_context) {
      while (1) {
        switch (_context.prev = _context.next) {
          case 0:
            start = Date.now();
            _context.next = 3;
            return next();

          case 3:
            ms = Date.now() - start;
            ctx.set('X-Response-Time', "".concat(ms));

          case 5:
          case "end":
            return _context.stop();
        }
      }
    }, _callee);
  }));
  return _xResponseTime.apply(this, arguments);
}