76 lines
2.6 KiB
JavaScript
76 lines
2.6 KiB
JavaScript
const require_rolldown_runtime = require('../_virtual/rolldown_runtime.js');
|
|
let async_hooks = require("async_hooks");
|
|
async_hooks = require_rolldown_runtime.__toESM(async_hooks);
|
|
|
|
//#region src/common/trace.ts
|
|
const traceContext = new async_hooks.AsyncLocalStorage();
|
|
/**
|
|
* A regex to match the Cloud Trace header.
|
|
* - ([A-Fa-f0-9]{32}): The trace id, a 32 character hex value. (e.g. 4bf92f3577b34da6a3ce929d0e0e4736)
|
|
* - ([0-9]+): The parent span id, a 64 bit integer. (e.g. 00f067aa0ba902b7)
|
|
* - (?:;o=([0-3])): The trace mask, 1-3 denote it should be traced.
|
|
*/
|
|
const CLOUD_TRACE_REGEX = new RegExp("^(?<traceId>[A-Fa-f0-9]{32})/" + "(?<parentIdInt>[0-9]+)" + "(?:;o=(?<traceMask>[0-3]))?$");
|
|
const CLOUD_TRACE_HEADER = "X-Cloud-Trace-Context";
|
|
function matchCloudTraceHeader(carrier) {
|
|
let header = carrier?.[CLOUD_TRACE_HEADER];
|
|
if (!header) {
|
|
header = carrier?.[CLOUD_TRACE_HEADER.toLowerCase()];
|
|
}
|
|
if (header && typeof header === "string") {
|
|
const matches = CLOUD_TRACE_REGEX.exec(header);
|
|
if (matches && matches.groups) {
|
|
const { traceId, parentIdInt, traceMask } = matches.groups;
|
|
const parentId = parseInt(parentIdInt);
|
|
if (isNaN(parentId)) {
|
|
return;
|
|
}
|
|
const sample = !!traceMask && traceMask !== "0";
|
|
return {
|
|
traceId,
|
|
parentId: parentId.toString(16),
|
|
sample,
|
|
version: "00"
|
|
};
|
|
}
|
|
}
|
|
}
|
|
/**
|
|
* A regex to match the traceparent header.
|
|
* - ^([a-f0-9]{2}): The specification version (e.g. 00)
|
|
* - ([a-f0-9]{32}): The trace id, a 16-byte array. (e.g. 4bf92f3577b34da6a3ce929d0e0e4736)
|
|
* - ([a-f0-9]{16}): The parent span id, an 8-byte array. (e.g. 00f067aa0ba902b7)
|
|
* - ([a-f0-9]{2}: The sampled flag. (e.g. 00)
|
|
*/
|
|
const TRACEPARENT_REGEX = new RegExp("^(?<version>[a-f0-9]{2})-" + "(?<traceId>[a-f0-9]{32})-" + "(?<parentId>[a-f0-9]{16})-" + "(?<flag>[a-f0-9]{2})$");
|
|
const TRACEPARENT_HEADER = "traceparent";
|
|
function matchTraceparentHeader(carrier) {
|
|
const header = carrier?.[TRACEPARENT_HEADER];
|
|
if (header && typeof header === "string") {
|
|
const matches = TRACEPARENT_REGEX.exec(header);
|
|
if (matches && matches.groups) {
|
|
const { version, traceId, parentId, flag } = matches.groups;
|
|
const sample = flag === "01";
|
|
return {
|
|
traceId,
|
|
parentId,
|
|
sample,
|
|
version
|
|
};
|
|
}
|
|
}
|
|
}
|
|
/**
|
|
* Extracts trace context from given carrier object, if any.
|
|
*
|
|
* Supports Cloud Trace and traceparent format.
|
|
*
|
|
* @param carrier
|
|
*/
|
|
function extractTraceContext(carrier) {
|
|
return matchCloudTraceHeader(carrier) || matchTraceparentHeader(carrier);
|
|
}
|
|
|
|
//#endregion
|
|
exports.extractTraceContext = extractTraceContext;
|
|
exports.traceContext = traceContext; |