@vtvlive/interactive-apm 0.0.2 → 0.0.4

package/dist/providers/elastic-apm.tracing-provider.js

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ElasticApmTracingProvider = void 0;
+const debug_logger_1 = require("../utils/debug-logger");
 /**
  * Elastic APM Tracing Provider Implementation
  * Sử dụng elastic-apm-node agent để gửi traces về Elastic APM server
@@ -8,45 +9,65 @@ exports.ElasticApmTracingProvider = void 0;
 class ElasticApmTracingProvider {
     constructor(config = {}) {
         this.apm = null;
-        this.serviceName = config.serviceName || process.env.ELASTIC_APM_SERVICE_NAME || 'interactive-backend';
-        this.environment = config.environment || process.env.ELASTIC_APM_ENVIRONMENT || 'development';
+        this.serviceName =
+            config.serviceName || process.env.ELASTIC_APM_SERVICE_NAME || "interactive-backend";
+        this.environment = config.environment || process.env.ELASTIC_APM_ENVIRONMENT || "development";
     }
     /**
      * Initialize Elastic APM agent
      * Should be called before using the provider
      */
-    initialize(config) {
+    async initialize(config) {
         try {
             // @ts-ignore - Optional peer dependency
-            const apm = require('elastic-apm-node');
+            const apm = require("elastic-apm-node");
             // If agent is already started, just get reference
             if (apm.isStarted()) {
                 this.apm = apm;
+                if ((0, debug_logger_1.isDebugEnabled)()) {
+                    console.log(`[APM-DEBUG] [ElasticAPM] Agent already started, using existing instance`);
+                }
                 return;
             }
+            const serverUrl = config?.serverUrl || process.env.ELASTIC_APM_SERVER_URL || "http://localhost:8200";
+            const secretToken = config?.secretToken || process.env.ELASTIC_APM_SECRET_TOKEN;
+            const serviceVersion = config?.serviceVersion || process.env.npm_package_version || "1.0.0";
+            const environment = config?.environment || this.environment;
+            // Log initialization details when debug mode is on
+            (0, debug_logger_1.logInitialization)("ElasticAPM", {
+                serviceName: config?.serviceName || this.serviceName,
+                serverUrl,
+                hasSecretToken: !!secretToken,
+                serviceVersion,
+                environment,
+            });
             // Start the agent with provided config
             this.apm = apm.start({
                 serviceName: config?.serviceName || this.serviceName,
-                serverUrl: config?.serverUrl || process.env.ELASTIC_APM_SERVER_URL || 'http://localhost:8200',
-                secretToken: config?.secretToken || process.env.ELASTIC_APM_SECRET_TOKEN,
-                serviceVersion: config?.serviceVersion || process.env.npm_package_version || '1.0.0',
-                environment: config?.environment || this.environment,
-                logLevel: 'error',
+                serverUrl,
+                secretToken,
+                serviceVersion,
+                environment,
+                logLevel: (0, debug_logger_1.isDebugEnabled)() ? "info" : "error",
             });
-            console.log(`[ElasticAPM] Service: ${this.serviceName}, Environment: ${this.environment}`);
-            console.log('[ElasticAPM] Initialized');
+            if ((0, debug_logger_1.isDebugEnabled)()) {
+                (0, debug_logger_1.infoLog)(`[ElasticAPM] Service: ${this.serviceName}, Environment: ${this.environment}`);
+                (0, debug_logger_1.infoLog)("[ElasticAPM] Initialized");
+            }
         }
         catch (error) {
-            console.error('[ElasticAPM] Failed to initialize:', error);
+            (0, debug_logger_1.errorLog)("[ElasticAPM] Failed to initialize:", error);
             throw error;
         }
     }
     /**
      * Bắt đầu một span mới
      */
-    startSpan(name, attributes, spanKind = 'INTERNAL') {
+    startSpan(name, attributes, spanKind = "INTERNAL") {
         if (!this.apm) {
-            console.warn('[ElasticAPM] Not initialized');
+            if ((0, debug_logger_1.isDebugEnabled)()) {
+                console.warn("[ElasticAPM] Not initialized");
+            }
             return null;
         }
         // Elastic APM cần transaction để tạo span
@@ -60,7 +81,7 @@ class ElasticApmTracingProvider {
             }
             createdTransaction = true;
         }
-        const span = this.apm.startSpan(name, 'custom');
+        const span = this.apm.startSpan(name, "custom");
         if (!span) {
             if (createdTransaction && transaction) {
                 transaction.end();
@@ -75,10 +96,56 @@ class ElasticApmTracingProvider {
         }
         // Set span type dựa trên kind
         span.setType(this.mapSpanKindToType(spanKind));
+        // Capture startTime for logging (will log when span ends)
+        const startTime = Date.now();
+        // Add end protection to prevent operations on ended span
+        let isEnded = false;
+        // Override setLabel to check if span has ended
+        const originalSetLabel = span.setLabel.bind(span);
+        span.setLabel = (key, value) => {
+            if (isEnded) {
+                if ((0, debug_logger_1.isDebugEnabled)()) {
+                    console.warn(`[ElasticAPM] Cannot set label "${key}" on ended span "${name}". ` +
+                        `This attribute will not be sent to APM server.`);
+                }
+                return;
+            }
+            originalSetLabel(key, value);
+        };
+        // Override setAttribute (if it exists, maps to setLabel)
+        if (typeof span.setAttribute === "function") {
+            const originalSetAttribute = span.setAttribute.bind(span);
+            span.setAttribute = (key, value) => {
+                if (isEnded) {
+                    if ((0, debug_logger_1.isDebugEnabled)()) {
+                        console.warn(`[ElasticAPM] Cannot set attribute "${key}" on ended span "${name}". ` +
+                            `This attribute will not be sent to APM server.`);
+                    }
+                    return;
+                }
+                originalSetAttribute(key, value);
+            };
+        }
         // Override end() để end transaction sau
         const originalEnd = span.end.bind(span);
         span.end = () => {
+            if (isEnded) {
+                if ((0, debug_logger_1.isDebugEnabled)()) {
+                    console.warn(`[ElasticAPM] Span "${name}" has already been ended. Ignoring duplicate end().`);
+                }
+                return;
+            }
+            isEnded = true;
+            // Log span details with actual duration
+            const endTime = Date.now();
+            (0, debug_logger_1.logSpan)("ElasticAPM", {
+                name: span.name,
+                kind: this.mapSpanKindToNumber(spanKind),
+                startTime,
+                endTime,
+            });
             originalEnd();
+            // End transaction if we created it
             if (createdTransaction && transaction) {
                 transaction.end();
             }
@@ -97,7 +164,10 @@ class ElasticApmTracingProvider {
         catch (error) {
             this.apm?.captureError(error);
             if (span) {
-                span.outcome = 'failure';
+                span.outcome = "failure";
+            }
+            if ((0, debug_logger_1.isDebugEnabled)()) {
+                (0, debug_logger_1.infoLog)(`[ElasticAPM] Span failed: ${name} ${error.message || error}`);
             }
             throw error;
         }
@@ -128,7 +198,7 @@ class ElasticApmTracingProvider {
      * End span manually
      */
     endSpan(span) {
-        const activeSpan = span || this.apm?.currentSpan;
+        const activeSpan = (span || this.apm?.currentSpan);
         if (activeSpan) {
             activeSpan.end();
         }
@@ -138,7 +208,18 @@ class ElasticApmTracingProvider {
      */
     async shutdown() {
         if (this.apm) {
-            await this.apm.flush();
+            try {
+                if (this.apm.flush) {
+                    await this.apm.flush();
+                }
+                if ((0, debug_logger_1.isDebugEnabled)()) {
+                    (0, debug_logger_1.infoLog)("[ElasticAPM] Shutdown completed successfully");
+                }
+            }
+            catch (error) {
+                (0, debug_logger_1.errorLog)("[ElasticAPM] Shutdown failed:", error);
+                throw error;
+            }
         }
     }
     /**
@@ -146,17 +227,35 @@ class ElasticApmTracingProvider {
      */
     mapSpanKindToType(kind) {
         switch (kind.toUpperCase()) {
-            case 'SERVER':
-                return 'request';
-            case 'CLIENT':
-                return 'db';
-            case 'PRODUCER':
-                return 'messaging';
-            case 'CONSUMER':
-                return 'messaging';
-            case 'INTERNAL':
+            case "SERVER":
+                return "request";
+            case "CLIENT":
+                return "db";
+            case "PRODUCER":
+                return "messaging";
+            case "CONSUMER":
+                return "messaging";
+            case "INTERNAL":
+            default:
+                return "code";
+        }
+    }
+    /**
+     * Map string span kind to numeric value (for logging)
+     */
+    mapSpanKindToNumber(kind) {
+        switch (kind.toUpperCase()) {
+            case "SERVER":
+                return 1;
+            case "CLIENT":
+                return 2;
+            case "PRODUCER":
+                return 3;
+            case "CONSUMER":
+                return 4;
+            case "INTERNAL":
             default:
-                return 'code';
+                return 0;
         }
     }
 }

package/dist/providers/opentelemetry.tracing-provider.d.ts

@@ -1,4 +1,6 @@
-import { ITracingProvider } from '../interfaces/tracing-provider.interface';
+import { ITracingProvider } from "../interfaces/tracing-provider.interface";
+import { OtlpTransport } from "../types/otlp-transport.type";
+import { ISpan } from "../types/apm.types";
 /**
  * OpenTelemetry Tracing Provider Implementation
  * Sử dụng OpenTelemetry SDK với OTLP exporter để gửi traces về Elastic APM
@@ -9,6 +11,7 @@ export declare class OpenTelemetryTracingProvider implements ITracingProvider {
     private readonly serviceName;
     private readonly environment;
     private readonly otlpEndpoint;
+    private readonly otlpTransport;
     private readonly secretToken;
     private readonly otlpAuthToken;
     private readonly otlpHeaders;
@@ -18,6 +21,7 @@ export declare class OpenTelemetryTracingProvider implements ITracingProvider {
         serviceName?: string;
         environment?: string;
         otlpEndpoint?: string;
+        otlpTransport?: OtlpTransport | string;
         secretToken?: string;
         otlpAuthToken?: string;
         otlpHeaders?: Record<string, string>;
@@ -39,11 +43,11 @@ export declare class OpenTelemetryTracingProvider implements ITracingProvider {
     /**
      * Bắt đầu một span mới
      */
-    startSpan(name: string, attributes?: Record<string, string | number>, spanKind?: string): any;
+    startSpan(name: string, attributes?: Record<string, string | number>, spanKind?: string): ISpan | null;
     /**
      * Thực thi function với tracing context
      */
-    startSpanWithParent<T>(name: string, fn: (span: any) => Promise<T>, attributes?: Record<string, string | number>): Promise<T>;
+    startSpanWithParent<T>(name: string, fn: (span: ISpan | null) => Promise<T>, attributes?: Record<string, string | number>): Promise<T>;
     /**
      * Capture error vào active span
      */
@@ -52,10 +56,14 @@ export declare class OpenTelemetryTracingProvider implements ITracingProvider {
      * Set attribute cho active span
      */
     setAttribute(key: string, value: string | number): void;
+    /**
+     * Force flush all pending spans
+     */
+    forceFlush(): Promise<void>;
     /**
      * End span manually
      */
-    endSpan(span?: any): void;
+    endSpan(span?: ISpan | null): void;
     /**
      * Shutdown gracefully
      */

package/dist/providers/opentelemetry.tracing-provider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"opentelemetry.tracing-provider.d.ts","sourceRoot":"","sources":["../../src/providers/opentelemetry.tracing-provider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,0CAA0C,CAAC;AAE5E;;;GAGG;AACH,qBAAa,4BAA6B,YAAW,gBAAgB;IACnE,OAAO,CAAC,GAAG,CAAa;IACxB,OAAO,CAAC,MAAM,CAAa;IAC3B,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IACrC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IACrC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAS;IACtC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IACrC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAS;IACvC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAyB;IACrD,OAAO,CAAC,QAAQ,CAAC,qBAAqB,CAAU;IAChD,OAAO,CAAC,WAAW,CAAS;gBAEhB,MAAM,GAAE;QAClB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACrC,qBAAqB,CAAC,EAAE,OAAO,CAAC;KAC5B;IAUN;;;OAGG;IACG,UAAU,CAAC,MAAM,CAAC,EAAE;QACxB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACrC,qBAAqB,CAAC,EAAE,OAAO,CAAC;QAChC,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,GAAG,OAAO,CAAC,IAAI,CAAC;IAwIjB;;OAEG;IACH,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC,EAAE,QAAQ,GAAE,MAAmB,GAAG,GAAG;IAqBzG;;OAEG;IACG,mBAAmB,CAAC,CAAC,EACzB,IAAI,EAAE,MAAM,EACZ,EAAE,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,CAAC,CAAC,EAC7B,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC,GAC3C,OAAO,CAAC,CAAC,CAAC;IAoBb;;OAEG;IACH,YAAY,CAAC,KAAK,EAAE,KAAK,GAAG,IAAI;IAchC;;OAEG;IACH,YAAY,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAavD;;OAEG;IACH,OAAO,CAAC,IAAI,CAAC,EAAE,GAAG,GAAG,IAAI;IAazB;;OAEG;IACG,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAM/B;;OAEG;IACH,OAAO,CAAC,WAAW;CAiBpB"}
+{"version":3,"file":"opentelemetry.tracing-provider.d.ts","sourceRoot":"","sources":["../../src/providers/opentelemetry.tracing-provider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,0CAA0C,CAAC;AAC5E,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAyG3C;;;GAGG;AACH,qBAAa,4BAA6B,YAAW,gBAAgB;IACnE,OAAO,CAAC,GAAG,CAAiB;IAC5B,OAAO,CAAC,MAAM,CAAiB;IAC/B,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IACrC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IACrC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAS;IACtC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAgB;IAC9C,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IACrC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAS;IACvC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAyB;IACrD,OAAO,CAAC,QAAQ,CAAC,qBAAqB,CAAU;IAChD,OAAO,CAAC,WAAW,CAAS;gBAG1B,MAAM,GAAE;QACN,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,aAAa,CAAC,EAAE,aAAa,GAAG,MAAM,CAAC;QACvC,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACrC,qBAAqB,CAAC,EAAE,OAAO,CAAC;KAC5B;IA6BR;;;OAGG;IACG,UAAU,CAAC,MAAM,CAAC,EAAE;QACxB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACrC,qBAAqB,CAAC,EAAE,OAAO,CAAC;QAChC,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,GAAG,OAAO,CAAC,IAAI,CAAC;IA+NjB;;OAEG;IACH,SAAS,CACP,IAAI,EAAE,MAAM,EACZ,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC,EAC5C,QAAQ,GAAE,MAAmB,GAC5B,KAAK,GAAG,IAAI;IA+Gf;;OAEG;IACG,mBAAmB,CAAC,CAAC,EACzB,IAAI,EAAE,MAAM,EACZ,EAAE,EAAE,CAAC,IAAI,EAAE,KAAK,GAAG,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,EACtC,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC,GAC3C,OAAO,CAAC,CAAC,CAAC;IAmCb;;OAEG;IACH,YAAY,CAAC,KAAK,EAAE,KAAK,GAAG,IAAI;IAiBhC;;OAEG;IACH,YAAY,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAavD;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAajC;;OAEG;IACH,OAAO,CAAC,IAAI,CAAC,EAAE,KAAK,GAAG,IAAI,GAAG,IAAI;IAalC;;OAEG;IACG,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAM/B;;OAEG;IACH,OAAO,CAAC,WAAW;CAiBpB"}