@opentelemetry/instrumentation-aws-sdk
Version:
OpenTelemetry instrumentation for `aws-sdk` and `@aws-sdk/client-*` clients for various AWS services
68 lines • 2.78 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.addPropagationFieldsToAttributeNames = exports.extractPropagationContext = exports.injectPropagationContext = exports.contextGetter = exports.contextSetter = exports.MAX_MESSAGE_ATTRIBUTES = void 0;
/*
* Copyright The OpenTelemetry Authors
* SPDX-License-Identifier: Apache-2.0
*/
const api_1 = require("@opentelemetry/api");
// https://docs.aws.amazon.com/AWSSimpleQueueService/latest/SQSDeveloperGuide/sqs-quotas.html
exports.MAX_MESSAGE_ATTRIBUTES = 10;
class ContextSetter {
set(carrier, key, value) {
carrier[key] = {
DataType: 'String',
StringValue: value,
};
}
}
exports.contextSetter = new ContextSetter();
class ContextGetter {
keys(carrier) {
if (carrier == null) {
return [];
}
return Object.keys(carrier);
}
get(carrier, key) {
return carrier?.[key]?.StringValue || carrier?.[key]?.Value;
}
}
exports.contextGetter = new ContextGetter();
const injectPropagationContext = (attributesMap) => {
const attributes = attributesMap ?? {};
if (Object.keys(attributes).length + api_1.propagation.fields().length <=
exports.MAX_MESSAGE_ATTRIBUTES) {
api_1.propagation.inject(api_1.context.active(), attributes, exports.contextSetter);
}
else {
api_1.diag.warn('aws-sdk instrumentation: cannot set context propagation on SQS/SNS message due to maximum amount of MessageAttributes');
}
return attributes;
};
exports.injectPropagationContext = injectPropagationContext;
const extractPropagationContext = (message, sqsExtractContextPropagationFromPayload) => {
const propagationFields = api_1.propagation.fields();
const hasPropagationFields = Object.keys(message.MessageAttributes || []).some(attr => propagationFields.includes(attr));
if (hasPropagationFields) {
return message.MessageAttributes;
}
else if (sqsExtractContextPropagationFromPayload && message.Body) {
try {
const payload = JSON.parse(message.Body);
return payload.MessageAttributes;
}
catch {
api_1.diag.debug('failed to parse SQS payload to extract context propagation, trace might be incomplete.');
}
}
return undefined;
};
exports.extractPropagationContext = extractPropagationContext;
const addPropagationFieldsToAttributeNames = (messageAttributeNames = [], propagationFields) => {
return messageAttributeNames.length
? Array.from(new Set([...messageAttributeNames, ...propagationFields]))
: propagationFields;
};
exports.addPropagationFieldsToAttributeNames = addPropagationFieldsToAttributeNames;
//# sourceMappingURL=MessageAttributes.js.map