ai-perf-sdk 1.0.2 → 1.0.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-perf-sdk",
-  "version": "1.0.2",
+  "version": "1.0.3",
   "description": "Plug-and-play performance monitoring SDK using OpenTelemetry",
   "license": "MIT",
   "author": "Akash",

package/src/sdk.js

@@ -1,9 +1,3 @@
-// 1. Force JSON Protocol (Critical for backend compatibility)
-// MUST happen before requiring OpenTelemetry modules
-if (!process.env.OTEL_EXPORTER_OTLP_PROTOCOL) {
-    process.env.OTEL_EXPORTER_OTLP_PROTOCOL = 'http/json';
-}
-
 const { NodeSDK } = require("@opentelemetry/sdk-node");
 const { getNodeAutoInstrumentations } = require("@opentelemetry/auto-instrumentations-node");
 const { OTLPTraceExporter } = require("@opentelemetry/exporter-trace-otlp-http");
@@ -27,27 +21,21 @@ function startSDK(options = {}) {
         return;
     }
 
-    // 2. Merge Configuration
+    // 1. Merge Configuration
     const config = {
         ...defaults,
         ...options
     };
 
-    // 3. Enable Debug Logging if requested
+    // 2. Enable Debug Logging if requested
     if (config.debug) {
         console.log("[AI-PERF] Debug Mode Enabled");
         diag.setLogger(new DiagConsoleLogger(), DiagLogLevel.DEBUG);
     }
 
-    // 4. Robust Header Handling
-    // Merge headers from:
-    // a. Environment Variable OTEL_EXPORTER_OTLP_HEADERS (Standard)
-    // b. options.headers (User provided)
-    // c. defaults.headers (If any)
-
+    // 3. Robust Header Handling
     let combinedHeaders = { ...config.headers };
 
-    // Parse OTEL_EXPORTER_OTLP_HEADERS env var manually to ensure they aren't lost
     if (process.env.OTEL_EXPORTER_OTLP_HEADERS) {
         try {
             const envHeaders = process.env.OTEL_EXPORTER_OTLP_HEADERS.split(',').reduce((acc, pair) => {
@@ -65,7 +53,7 @@ function startSDK(options = {}) {
         console.log("[AI-PERF] Exporter Config:", {
             url: config.collectorEndpoint,
             headers: combinedHeaders,
-            protocol: process.env.OTEL_EXPORTER_OTLP_PROTOCOL
+            protocol: process.env.OTEL_EXPORTER_OTLP_PROTOCOL || 'default (protobuf)'
         });
     }
 
@@ -74,8 +62,7 @@ function startSDK(options = {}) {
         headers: combinedHeaders
     });
 
-    // 5. Configure Auto-Instrumentation
-    // Allow disabling DB instrumentation to prevent crashes (e.g. Mongoose 8.x conflicts)
+    // 4. Configure Auto-Instrumentation
     const instrumentationConfig = {};
 
     if (config.disableDbInstrumentation) {
@@ -98,6 +85,17 @@ function startSDK(options = {}) {
     try {
         sdkInstance.start();
         console.log(`[AI-PERF] SDK started for ${config.serviceName} ` + (config.disableDbInstrumentation ? "(Safe Mode)" : ""));
+
+        // 5. Automatic Graceful Shutdown
+        const handleShutdown = async () => {
+            console.log("[AI-PERF] Graceful shutdown initiated...");
+            await shutdownSDK();
+            process.exit(0);
+        };
+
+        process.on('SIGTERM', handleShutdown);
+        process.on('SIGINT', handleShutdown);
+
     } catch (e) {
         console.error("[AI-PERF] Failed to start SDK:", e);
     }