@namchee/decora
Version:
Useful ECMAScript-compliant decorators to enhance JS (specifically, TypeScript) development experience
51 lines (50 loc) • 2.33 kB
JavaScript
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
import { isNode, isBrowser } from './utils';
export function benchmark(metric = 'ms', precision, stream = process.stdout) {
return function (target, keyName, descriptor) {
if (!descriptor || !(descriptor.value instanceof Function)) {
throw new Error('@benchmark decorator can only be applied to functions');
}
let logger;
let performance;
if (isNode) {
const Console = require('console').Console;
logger = new Console({ stdout: stream });
performance = require('perf_hooks').performance;
}
else if (isBrowser) {
logger = window.console;
performance = window.performance;
if (stream) {
logger.warn('The `stream` option only available on Node.JS environment. This option will be ignored.');
}
}
else {
return;
}
const fn = descriptor.value;
descriptor.value = function (...args) {
return __awaiter(this, void 0, void 0, function* () {
const start = performance.now();
const result = yield fn.apply(this, args);
const end = performance.now();
let diff = end - start;
if (metric !== 'ms') {
diff = metric === 's' ?
diff / 1000 :
diff * 1000;
}
logger.info(`Function ${keyName} from class ${target.constructor.name} was executed in ${diff.toFixed(precision)} ${metric}`);
return result;
});
};
};
}