@trifrost/core
Version:
Blazingly fast, runtime-agnostic server framework for modern edge and node environments
76 lines (75 loc) • 2.75 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.Sym_TriFrostSpan = void 0;
exports.span = span;
exports.spanFn = spanFn;
const Als_1 = require("../../utils/Als");
/**
* Symbol attached to methods to prevent duplicate span wraps
*/
exports.Sym_TriFrostSpan = Symbol('trifrost.logger.is_span');
function hasLogger(val) {
return typeof val?.logger?.span === 'function';
}
/**
* Resolve a TriFrost logger instance from any combination of:
* - ALS-bound context (if available)
* - First argument (if it's a TriFrostContext)
* - `this.logger` or `this.ctx.logger` (for instance methods)
*/
function resolveLogger(self, args) {
// ALS-bound context
const ctxAls = (0, Als_1.ctx)();
if (hasLogger(ctxAls))
return ctxAls?.logger;
// First argument
const ctxArg = Array.isArray(args) && args.length ? args[0] : undefined;
if (hasLogger(ctxArg))
return ctxArg?.logger;
// Fallback to self.logger
if (hasLogger(self))
return self.logger;
// Fallback to self.ctx.logger
if (hasLogger(self?.ctx))
return self.ctx.logger;
return null;
}
function span(name) {
return function (method, context) {
/* Prevent re-decoration */
if (Reflect.get(method, exports.Sym_TriFrostSpan))
return method;
/* Define span name */
const span_name = typeof name === 'string' && name.length ? name : String(context.name);
const wrapped = function (...fn_args) {
const logger = resolveLogger(this, fn_args);
if (!logger)
return method.call(this, ...fn_args);
return logger.span(span_name, () => method.call(this, ...fn_args));
};
/* Set to prevent re-decoration */
Reflect.set(wrapped, exports.Sym_TriFrostSpan, true);
return wrapped;
};
}
function spanFn(...args) {
const [name_or_fn, maybe_fn] = args;
/* Define function */
const fn = typeof name_or_fn === 'function' ? name_or_fn : maybe_fn;
/* Prevent re-decoration */
if (Reflect.get(fn, exports.Sym_TriFrostSpan))
return fn;
/* Define name */
// eslint-disable-next-line prettier/prettier
const span_name = typeof name_or_fn === 'string' && name_or_fn.length ? name_or_fn : typeof fn.name === 'string' && fn.name.length ? fn.name : 'anonymous';
/* Span function */
const fn_span = function (...fn_args) {
const logger = resolveLogger(this, fn_args);
if (!logger)
return fn.apply(this, fn_args);
return logger.span(span_name, () => fn.apply(this, fn_args));
};
/* Set to prevent re-decoration */
Reflect.set(fn_span, exports.Sym_TriFrostSpan, true);
return fn_span;
}