UNPKG

@namchee/decora

Version:

Useful ECMAScript-compliant decorators to enhance JS (specifically, TypeScript) development experience

51 lines (50 loc) 2.33 kB
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; }); }; }; }