sm-utility 2.4.26 → 2.4.27

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sm-utility",
-  "version": "2.4.26",
+  "version": "2.4.27",
   "description": "reusable utility codes for sm projects",
   "main": "index.js",
   "types": "./index.d.ts",
@@ -28,6 +28,7 @@
   },
   "dependencies": {
     "@aws-sdk/client-secrets-manager": "3.758.0",
+    "@aws-sdk/client-sqs": "3.758.0",
     "@elastic/ecs-winston-format": "^1.3.1",
     "@opentelemetry/auto-instrumentations-node": "0.55.3",
     "@opentelemetry/exporter-trace-otlp-http": "0.57.1",

package/tracing/sqsOtelContextHandler.d.ts

@@ -0,0 +1,5 @@
+import type { SendMessageCommandInput, Message } from '@aws-sdk/client-sqs';
+export declare class SQSOtelContextHandler {
+    static injectContext(params: SendMessageCommandInput): SendMessageCommandInput;
+    static extractContext(message: Message): import("@opentelemetry/api").Context;
+}

package/tracing/sqsOtelContextHandler.js

@@ -0,0 +1,45 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SQSOtelContextHandler = void 0;
+const api_1 = require("@opentelemetry/api");
+class SQSOtelContextHandler {
+    static injectContext(params) {
+        var _a;
+        const carrier = {};
+        // Inject current active trace context into carrier
+        api_1.propagation.inject(api_1.context.active(), carrier, {
+            set: (c, k, v) => {
+                c[k] = String(v);
+            },
+        });
+        // Convert carrier to SQS MessageAttributes
+        const otelAttributes = {};
+        for (const [key, value] of Object.entries(carrier)) {
+            otelAttributes[key] = {
+                DataType: 'String',
+                StringValue: value,
+            };
+        }
+        return {
+            ...params,
+            MessageAttributes: {
+                ...((_a = params.MessageAttributes) !== null && _a !== void 0 ? _a : {}),
+                ...otelAttributes,
+            },
+        };
+    }
+    static extractContext(message) {
+        var _a;
+        const attrs = (_a = message.MessageAttributes) !== null && _a !== void 0 ? _a : {};
+        const carrier = {};
+        for (const [key, val] of Object.entries(attrs)) {
+            if (val === null || val === void 0 ? void 0 : val.StringValue)
+                carrier[key] = val.StringValue;
+        }
+        return api_1.propagation.extract(api_1.context.active(), carrier, {
+            get: (c, k) => c[k],
+            keys: (c) => Object.keys(c),
+        });
+    }
+}
+exports.SQSOtelContextHandler = SQSOtelContextHandler;