logixia 1.2.1 → 1.3.1

package/dist/{logitron-logger.module-iO8DPE7_.mjs → logitron-logger.module-2AzkadqZ.mjs}

@@ -1,10 +1,79 @@
 import { a as __toESM, t as __commonJS } from "./chunk-uEZWKkIX.mjs";
-import { d as internalLog, f as internalWarn, t as TransportManager, u as internalError } from "./transport.manager-Vi__pagn.mjs";
+import { d as internalLog, f as internalWarn, t as TransportManager, u as internalError } from "./transport.manager-5VVdqS3o.mjs";
 import buildFastStringify from "fast-json-stringify";
 import { AsyncLocalStorage } from "node:async_hooks";
-import crypto from "crypto";
+import crypto$1 from "crypto";
 import { Injectable, Module, RequestMethod, Scope } from "@nestjs/common";
 
+//#region src/context/async-context.ts
+const _storage = new AsyncLocalStorage();
+const LogixiaContext = {
+	run(store, callback) {
+		const parent = _storage.getStore() ?? {};
+		return _storage.run({
+			...parent,
+			...store
+		}, callback);
+	},
+	get() {
+		return _storage.getStore();
+	},
+	set(fields) {
+		const store = _storage.getStore();
+		if (!store) return;
+		Object.assign(store, fields);
+	},
+	getStorage() {
+		return _storage;
+	}
+};
+/**
+* Create an Express/Connect-compatible middleware that wraps each request in
+* a `LogixiaContext.run()` scope populated with common request fields.
+*
+* @example
+* ```ts
+* import { createExpressContextMiddleware } from 'logixia';
+* app.use(createExpressContextMiddleware());
+* ```
+*/
+function createExpressContextMiddleware(options = {}) {
+	const { enrich, requestIdHeader = "x-request-id", traceIdHeader = "x-trace-id" } = options;
+	return function logixiaContextMiddleware(req, _res, next) {
+		const headers = req["headers"] ?? {};
+		const traceId = headers[traceIdHeader];
+		const base = {
+			requestId: headers[requestIdHeader] ?? crypto.randomUUID().slice(0, 8),
+			...traceId !== void 0 ? { traceId } : {}
+		};
+		LogixiaContext.run({
+			...base,
+			...enrich ? enrich(req) : {}
+		}, next);
+	};
+}
+/**
+* Create a Fastify lifecycle hook that wraps each request in a context scope.
+*
+* Register with `fastify.addHook('onRequest', createFastifyContextHook())`.
+*/
+function createFastifyContextHook(options = {}) {
+	const { enrich, requestIdHeader = "x-request-id", traceIdHeader = "x-trace-id" } = options;
+	return function logixiaFastifyHook(request, _reply, done) {
+		const headers = request["headers"] ?? {};
+		const traceId = headers[traceIdHeader];
+		const base = {
+			requestId: headers[requestIdHeader] ?? request["id"] ?? crypto.randomUUID().slice(0, 8),
+			...traceId !== void 0 ? { traceId } : {}
+		};
+		LogixiaContext.run({
+			...base,
+			...enrich ? enrich(request) : {}
+		}, done);
+	};
+}
+
+//#endregion
 //#region src/types/index.ts
 const LogLevel = {
 	ERROR: "error",
@@ -223,6 +292,137 @@ function isPlainObject(value) {
 	return value !== null && typeof value === "object" && !Array.isArray(value) && !(value instanceof Date) && !(value instanceof Error) && !(value instanceof RegExp);
 }
 
+//#endregion
+//#region src/utils/sampling.utils.ts
+const ALWAYS_EMIT_LEVELS = new Set(["error", "fatal"]);
+var Sampler = class {
+	constructor(config, onStats) {
+		this.sampledTraces = /* @__PURE__ */ new Set();
+		this.droppedTraces = /* @__PURE__ */ new Set();
+		this._tokenBucket = 0;
+		this._lastRefillMs = Date.now();
+		this._stats = {
+			evaluated: 0,
+			emitted: 0,
+			dropped: 0,
+			byLevel: {},
+			windowStart: Date.now()
+		};
+		this.config = config;
+		if (config.maxLogsPerSecond && config.maxLogsPerSecond > 0) this._tokenBucket = config.maxLogsPerSecond;
+		const interval$1 = config.statsIntervalMs ?? 6e4;
+		if (interval$1 > 0 && onStats) {
+			this._statsTimer = setInterval(() => {
+				onStats(this.getStats());
+				this.resetStats();
+			}, interval$1);
+			if (this._statsTimer.unref) this._statsTimer.unref();
+		}
+	}
+	/**
+	* Decide whether a given log entry should be emitted.
+	*
+	* @param level    Log level string (lowercase)
+	* @param traceId  Active trace ID, if any
+	* @returns true → emit, false → drop
+	*/
+	shouldEmit(level, traceId) {
+		var _this$config$perLevel;
+		const lvl = level.toLowerCase();
+		this._trackEvaluated(lvl);
+		if (ALWAYS_EMIT_LEVELS.has(lvl) && ((_this$config$perLevel = this.config.perLevel) === null || _this$config$perLevel === void 0 ? void 0 : _this$config$perLevel[lvl]) === void 0) {
+			this._trackEmitted(lvl);
+			return true;
+		}
+		if (this.config.traceConsistent && traceId) {
+			if (this.sampledTraces.has(traceId)) {
+				this._trackEmitted(lvl);
+				return true;
+			}
+			if (this.droppedTraces.has(traceId)) {
+				this._trackDropped(lvl);
+				return false;
+			}
+			const emit = this._sampleByRate(lvl);
+			if (emit) this.sampledTraces.add(traceId);
+			else this.droppedTraces.add(traceId);
+			if (emit) this._trackEmitted(lvl);
+			else this._trackDropped(lvl);
+			return emit;
+		}
+		if (!this._sampleByRate(lvl)) {
+			this._trackDropped(lvl);
+			return false;
+		}
+		if (this.config.maxLogsPerSecond && this.config.maxLogsPerSecond > 0) {
+			if (!this._consumeToken()) {
+				this._trackDropped(lvl);
+				return false;
+			}
+		}
+		this._trackEmitted(lvl);
+		return true;
+	}
+	getStats() {
+		return {
+			...this._stats,
+			byLevel: { ...this._stats.byLevel }
+		};
+	}
+	resetStats() {
+		this._stats = {
+			evaluated: 0,
+			emitted: 0,
+			dropped: 0,
+			byLevel: {},
+			windowStart: Date.now()
+		};
+		this.sampledTraces.clear();
+		this.droppedTraces.clear();
+	}
+	destroy() {
+		if (this._statsTimer) clearInterval(this._statsTimer);
+	}
+	_sampleByRate(level) {
+		var _this$config$perLevel2, _this$config$perLevel3;
+		const rate = ((_this$config$perLevel2 = this.config.perLevel) === null || _this$config$perLevel2 === void 0 ? void 0 : _this$config$perLevel2[level]) ?? ((_this$config$perLevel3 = this.config.perLevel) === null || _this$config$perLevel3 === void 0 ? void 0 : _this$config$perLevel3["*"]) ?? this.config.rate ?? 1;
+		if (rate >= 1) return true;
+		if (rate <= 0) return false;
+		return Math.random() < rate;
+	}
+	_consumeToken() {
+		const now = Date.now();
+		const elapsed = (now - this._lastRefillMs) / 1e3;
+		const max$1 = this.config.maxLogsPerSecond;
+		this._tokenBucket = Math.min(max$1, this._tokenBucket + elapsed * max$1);
+		this._lastRefillMs = now;
+		if (this._tokenBucket >= 1) {
+			this._tokenBucket -= 1;
+			return true;
+		}
+		return false;
+	}
+	_ensure(level) {
+		if (!this._stats.byLevel[level]) this._stats.byLevel[level] = {
+			evaluated: 0,
+			emitted: 0
+		};
+	}
+	_trackEvaluated(level) {
+		this._stats.evaluated++;
+		this._ensure(level);
+		this._stats.byLevel[level].evaluated++;
+	}
+	_trackEmitted(level) {
+		this._stats.emitted++;
+		this._ensure(level);
+		this._stats.byLevel[level].emitted++;
+	}
+	_trackDropped(_level) {
+		this._stats.dropped++;
+	}
+};
+
 //#endregion
 //#region src/utils/shutdown.utils.ts
 /** Module-level registry of all logger instances that have opted into graceful shutdown */
@@ -286,7 +486,7 @@ const rnds8Pool = new Uint8Array(256);
 let poolPtr = rnds8Pool.length;
 function rng() {
 	if (poolPtr > rnds8Pool.length - 16) {
-		crypto.randomFillSync(rnds8Pool);
+		crypto$1.randomFillSync(rnds8Pool);
 		poolPtr = 0;
 	}
 	return rnds8Pool.slice(poolPtr, poolPtr += 16);
@@ -306,7 +506,7 @@ function unsafeStringify(arr, offset = 0) {
 
 //#endregion
 //#region node_modules/uuid/dist/esm-node/native.js
-var native_default = { randomUUID: crypto.randomUUID };
+var native_default = { randomUUID: crypto$1.randomUUID };
 
 //#endregion
 //#region node_modules/uuid/dist/esm-node/v4.js
@@ -544,6 +744,13 @@ var LogixiaLogger = class LogixiaLogger {
 		};
 		this.context = context$1 ?? "";
 		if (this.config.transports) this.transportManager = new TransportManager(this.config.transports);
+		if (this.config.sampling) this._sampler = new Sampler(this.config.sampling, (stats) => {
+			process.stdout.write(JSON.stringify({
+				level: "info",
+				message: "[logixia/sampling] stats",
+				...stats
+			}) + "\n");
+		});
 		this.setupGracefulShutdown();
 		this.createCustomLevelMethods();
 		this._buildPerfCaches();
@@ -783,6 +990,7 @@ var LogixiaLogger = class LogixiaLogger {
 		return this.transportManager.healthCheck();
 	}
 	async close() {
+		var _this$_sampler;
 		for (const [label, timer$1] of this.timers) await this.warn(`Timer '${label}' was not ended properly`, {
 			startTime: new Date(timer$1.startTime).toISOString(),
 			duration: `${Date.now() - timer$1.startTime}ms (incomplete)`
@@ -792,15 +1000,25 @@ var LogixiaLogger = class LogixiaLogger {
 			await this.transportManager.flush();
 			await this.transportManager.close();
 		}
+		(_this$_sampler = this._sampler) === null || _this$_sampler === void 0 || _this$_sampler.destroy();
 		deregisterFromShutdown(this);
 	}
 	async log(level, message, data) {
 		if (this.config.silent) return;
 		if (!this.shouldLog(level)) return;
-		const rawPayload = this._hasContextData ? {
-			...this.contextData,
+		if (this._sampler) {
+			const traceId$1 = this.config.traceId ? getCurrentTraceId() ?? this.fallbackTraceId : void 0;
+			if (!this._sampler.shouldEmit(level, traceId$1)) return;
+		}
+		const alsContext = LogixiaContext.get();
+		const mergedData = alsContext && Object.keys(alsContext).length > 0 ? {
+			...alsContext,
 			...data
 		} : data;
+		const rawPayload = this._hasContextData ? {
+			...this.contextData,
+			...mergedData
+		} : mergedData;
 		let payload;
 		if (rawPayload !== void 0 && rawPayload !== null) payload = this._hasRedact ? applyRedaction(rawPayload, this.config.redact) ?? rawPayload : rawPayload;
 		const traceId = this.config.traceId ? getCurrentTraceId() ?? this.fallbackTraceId : void 0;
@@ -10303,5 +10521,5 @@ let LogixiaLoggerModule = class LogixiaLoggerModule$1 {
 LogixiaLoggerModule = _LogixiaLoggerModule = __decorate([Module({})], LogixiaLoggerModule);
 
 //#endregion
-export { redactObject as C, DEFAULT_LOG_COLORS as D, serializeError as E, DEFAULT_LOG_LEVELS as O, applyRedaction as S, normalizeError as T, traceStorage as _, TraceMiddleware as a, registerForShutdown as b, LogixiaLogger as c, createTraceMiddleware as d, extractTraceId as f, setTraceId as g, runWithTraceId as h, WebSocketTraceInterceptor as i, LogLevel as k, createLogger as l, getCurrentTraceId as m, LOGIXIA_LOGGER_PREFIX as n, KafkaTraceInterceptor as o, generateTraceId as p, LogixiaLoggerModule as r, LogixiaLoggerService as s, LOGIXIA_LOGGER_CONFIG as t, DEFAULT_TRACE_HEADERS$1 as u, deregisterFromShutdown as v, isError as w, resetShutdownHandlers as x, flushOnExit as y };
-//# sourceMappingURL=logitron-logger.module-iO8DPE7_.mjs.map
+export { LogixiaContext as A, redactObject as C, DEFAULT_LOG_COLORS as D, serializeError as E, createFastifyContextHook as M, DEFAULT_LOG_LEVELS as O, applyRedaction as S, normalizeError as T, traceStorage as _, TraceMiddleware as a, registerForShutdown as b, LogixiaLogger as c, createTraceMiddleware as d, extractTraceId as f, setTraceId as g, runWithTraceId as h, WebSocketTraceInterceptor as i, createExpressContextMiddleware as j, LogLevel as k, createLogger as l, getCurrentTraceId as m, LOGIXIA_LOGGER_PREFIX as n, KafkaTraceInterceptor as o, generateTraceId as p, LogixiaLoggerModule as r, LogixiaLoggerService as s, LOGIXIA_LOGGER_CONFIG as t, DEFAULT_TRACE_HEADERS$1 as u, deregisterFromShutdown as v, isError as w, resetShutdownHandlers as x, flushOnExit as y };
+//# sourceMappingURL=logitron-logger.module-2AzkadqZ.mjs.map