@loadstrike/loadstrike-sdk 1.0.21401 → 1.0.22301

package/dist/cjs/runtime.js

@@ -1724,9 +1724,10 @@ class LoadStrikeScenario {
             throw new Error("Tracking source endpoint must be provided.");
         }
         const destinationSpec = asTrackingRecord(pickTrackingValue(copied, "Destination", "destination"));
-        const sourceEndpoint = mapRuntimeTrackingEndpointSpec(sourceSpec);
+        const useLoadStrikeTraceIdHeader = readUseLoadStrikeTraceIdHeader(copied);
+        const sourceEndpoint = mapRuntimeTrackingEndpointSpec(sourceSpec, useLoadStrikeTraceIdHeader);
         const destinationEndpoint = Object.keys(destinationSpec).length
-            ? mapRuntimeTrackingEndpointSpec(destinationSpec)
+            ? mapRuntimeTrackingEndpointSpec(destinationSpec, useLoadStrikeTraceIdHeader)
             : null;
         validateRuntimeTrackingConfiguration(copied, sourceEndpoint, destinationEndpoint);
         return new LoadStrikeScenario(this.name, this.runHandler, this.initHandler, this.cleanHandler, this.loadSimulations, this.thresholds, copied, this.maxFailCount, this.withoutWarmUpValue, this.warmUpDurationSeconds, this.weight, this.restartIterationOnFail, this.internalLicenseFeatures);
@@ -5730,9 +5731,10 @@ class ManagedScenarioTrackingRuntime {
         this.timeoutLoop = null;
         const sourceSpec = asTrackingRecord(pickTrackingValue(tracking, "Source", "source"));
         const destinationSpec = asTrackingRecord(pickTrackingValue(tracking, "Destination", "destination"));
-        this.sourceEndpoint = mapRuntimeTrackingEndpointSpec(sourceSpec);
+        const useLoadStrikeTraceIdHeader = readUseLoadStrikeTraceIdHeader(tracking);
+        this.sourceEndpoint = mapRuntimeTrackingEndpointSpec(sourceSpec, useLoadStrikeTraceIdHeader);
         this.destinationEndpoint = Object.keys(destinationSpec).length
-            ? mapRuntimeTrackingEndpointSpec(destinationSpec)
+            ? mapRuntimeTrackingEndpointSpec(destinationSpec, useLoadStrikeTraceIdHeader)
             : null;
         validateRuntimeTrackingConfiguration(tracking, this.sourceEndpoint, this.destinationEndpoint);
         this.runMode = normalizeTrackingRunMode(pickTrackingString(tracking, "RunMode", "runMode", "GenerateAndCorrelate"));
@@ -6227,9 +6229,12 @@ function getScenarioTrackingRuntime(context, tracking) {
     return lease.runtime;
 }
 function buildTrackingLeaseKey(sessionId, scenarioName, tracking) {
-    const source = mapRuntimeTrackingEndpointSpec(asTrackingRecord(pickTrackingValue(tracking, "Source", "source")));
+    const useLoadStrikeTraceIdHeader = readUseLoadStrikeTraceIdHeader(tracking);
+    const source = mapRuntimeTrackingEndpointSpec(asTrackingRecord(pickTrackingValue(tracking, "Source", "source")), useLoadStrikeTraceIdHeader);
     const destinationSpec = asTrackingRecord(pickTrackingValue(tracking, "Destination", "destination"));
-    const destination = Object.keys(destinationSpec).length ? mapRuntimeTrackingEndpointSpec(destinationSpec) : null;
+    const destination = Object.keys(destinationSpec).length
+        ? mapRuntimeTrackingEndpointSpec(destinationSpec, useLoadStrikeTraceIdHeader)
+        : null;
     const runMode = normalizeTrackingRunMode(pickTrackingString(tracking, "RunMode", "runMode", "GenerateAndCorrelate"));
     return `${sessionId}|${scenarioName}|${source.name}|${destination?.name ?? "<source-only>"}|${runMode}`;
 }
@@ -6309,6 +6314,9 @@ function validateRuntimeTrackingConfiguration(tracking, sourceEndpoint, destinat
     }
     throw new RangeError(`Unsupported tracking run mode: ${runMode}.`);
 }
+function readUseLoadStrikeTraceIdHeader(tracking) {
+    return pickTrackingBoolean(tracking, "UseLoadStrikeTraceIdHeader", "useLoadStrikeTraceIdHeader", false);
+}
 function validateRuntimeRedisCorrelationStoreConfiguration(tracking) {
     const storeSpec = asTrackingRecord(pickTrackingValue(tracking, "CorrelationStore", "correlationStore"));
     const kind = pickTrackingString(storeSpec, "Kind", "kind", "").trim().toLowerCase();
@@ -6392,12 +6400,12 @@ function sanitizeTrackingNamespacePart(value) {
         .replace(/\s+/g, "_");
 }
 async function produceOrConsumeTrackingPayload(adapter, endpoint, runMode, index) {
-    const payload = runMode === "correlateexistingtraffic"
-        ? await adapter.consume()
-        : await adapter.produce();
-    return normalizeRuntimeTrackingPayload(payload, endpoint, index);
+    if (runMode === "correlateexistingtraffic") {
+        return normalizeRuntimeObservedTrackingPayload(await adapter.consume(), endpoint);
+    }
+    return normalizeRuntimeTrackingPayload(await adapter.produce(), endpoint, index);
 }
-function mapRuntimeTrackingEndpointSpec(spec) {
+function mapRuntimeTrackingEndpointSpec(spec, useLoadStrikeTraceIdHeader = false) {
     const pollIntervalMs = Math.trunc(pickTrackingNumber(spec, ["PollIntervalMs", "pollIntervalMs"], pickTrackingNumber(spec, ["PollIntervalSeconds", "pollIntervalSeconds", "PollInterval"], 0.25) * 1000));
     const httpSpec = asTrackingRecord(pickTrackingValue(spec, "Http", "http"));
     const delegateSpec = asTrackingRecord(pickTrackingValue(spec, "delegate", "Delegate", "DelegateStream"));
@@ -6424,7 +6432,7 @@ function mapRuntimeTrackingEndpointSpec(spec) {
         kind: pickTrackingString(spec, "Kind", "kind", "DelegateStream"),
         mode: pickTrackingString(spec, "Mode", "mode", "Produce"),
         name: pickTrackingString(spec, "Name", "name", "endpoint"),
-        trackingField: pickTrackingSelectorString(spec, "TrackingField", "trackingField", "header:x-correlation-id"),
+        trackingField: pickTrackingSelectorString(spec, "TrackingField", "trackingField", useLoadStrikeTraceIdHeader ? transports_js_1.LOADSTRIKE_TRACE_ID_TRACKING_FIELD : "header:x-correlation-id"),
         gatherByField: pickOptionalTrackingSelectorString(spec, "GatherByField", "gatherByField"),
         autoGenerateTrackingIdWhenMissing: pickTrackingBoolean(spec, "AutoGenerateTrackingIdWhenMissing", "autoGenerateTrackingIdWhenMissing", true),
         pollIntervalMs,
@@ -6530,7 +6538,9 @@ function normalizeRuntimeTrackingPayload(payload, endpoint, index) {
     if (existing || !endpoint.autoGenerateTrackingIdWhenMissing) {
         return normalized;
     }
-    const generated = `${endpoint.name}-auto-${index + 1}`;
+    const generated = endpoint.trackingField.trim().toLowerCase() === transports_js_1.LOADSTRIKE_TRACE_ID_TRACKING_FIELD
+        ? (0, node_crypto_1.randomUUID)()
+        : `${endpoint.name}-auto-${index + 1}`;
     const selector = endpoint.trackingField.trim();
     if (selector.toLowerCase().startsWith("header:")) {
         const headerName = selector.slice("header:".length).trim();
@@ -8034,7 +8044,7 @@ function buildRichHtmlReport(result) {
     <section id="scenarios" class="panel active">
       ${bars || "<div>No scenario data.</div>"}
       <table>
-        <thead><tr><th>Scenario</th><th>Requests</th><th>OK</th><th>Fail</th><th>TotalBytes</th><th>AvgLatencyMs</th></tr></thead>
+        <thead><tr><th>Scenario</th><th>Requests</th><th>OK</th><th>Fail</th><th>TotalBytes</th><th>AvgLatency (ms)</th></tr></thead>
         <tbody>${scenarioRows || "<tr><td colspan='6'>No rows</td></tr>"}</tbody>
       </table>
     </section>

package/dist/cjs/transports.js

@@ -33,9 +33,11 @@ var __importStar = (this && this.__importStar) || (function () {
     };
 })();
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.__loadstrikeTestExports = exports.EndpointAdapterFactory = exports.KafkaSaslOptions = exports.HttpAuthOptions = exports.HttpOAuth2ClientCredentialsOptions = exports.PushDiffusionEndpointDefinition = exports.DelegateStreamEndpointDefinition = exports.AzureEventHubsEndpointDefinition = exports.RedisStreamsEndpointDefinition = exports.NatsEndpointDefinition = exports.RabbitMqEndpointDefinition = exports.KafkaEndpointDefinition = exports.HttpEndpointDefinition = exports.TrafficEndpointDefinition = void 0;
+exports.__loadstrikeTestExports = exports.EndpointAdapterFactory = exports.KafkaSaslOptions = exports.HttpAuthOptions = exports.HttpOAuth2ClientCredentialsOptions = exports.PushDiffusionEndpointDefinition = exports.DelegateStreamEndpointDefinition = exports.AzureEventHubsEndpointDefinition = exports.RedisStreamsEndpointDefinition = exports.NatsEndpointDefinition = exports.RabbitMqEndpointDefinition = exports.KafkaEndpointDefinition = exports.HttpEndpointDefinition = exports.TrafficEndpointDefinition = exports.LOADSTRIKE_TRACE_ID_TRACKING_FIELD = exports.LOADSTRIKE_TRACE_ID_HEADER = void 0;
 const node_crypto_1 = require("node:crypto");
 const correlation_js_1 = require("./correlation.js");
+exports.LOADSTRIKE_TRACE_ID_HEADER = "loadstrike-trace-id";
+exports.LOADSTRIKE_TRACE_ID_TRACKING_FIELD = `header:${exports.LOADSTRIKE_TRACE_ID_HEADER}`;
 class TrafficEndpointDefinitionModel {
     get JsonSerializerSettings() {
         return this.JsonSettings;
@@ -1084,8 +1086,9 @@ class HttpEndpointAdapter extends CallbackAdapter {
         if (this.endpoint.mode !== "Produce") {
             return null;
         }
-        const resolved = payload ?? this.defaultPayload();
         const httpOptions = this.endpoint.http;
+        const responseTracking = canonicalizeHttpTrackingPayloadSource(String(httpOptions?.trackingPayloadSource ?? "Request")) === "Response";
+        const resolved = prepareProducedPayload(this.endpoint, payload, !responseTracking);
         if (!httpOptions?.url) {
             return super.produce(resolved);
         }
@@ -2988,7 +2991,7 @@ function createDefaultPayload(endpoint) {
         jsonConvertSettings: endpoint.jsonConvertSettings ? { ...endpoint.jsonConvertSettings } : undefined
     });
 }
-function prepareProducedPayload(endpoint, payload) {
+function prepareProducedPayload(endpoint, payload, allowAutoGenerateTrackingId = true) {
     const base = payload
         ? attachPayloadHelpers({
             headers: {
@@ -3007,15 +3010,24 @@ function prepareProducedPayload(endpoint, payload) {
     if (trackingId) {
         return base;
     }
+    if (!allowAutoGenerateTrackingId) {
+        return base;
+    }
     if (endpoint.autoGenerateTrackingIdWhenMissing === false) {
         throw new Error(`Tracking id could not be extracted using selector \`${endpoint.trackingField}\` and AutoGenerateTrackingIdWhenMissing is disabled for endpoint \`${endpoint.name}\`.`);
     }
-    const generated = (0, node_crypto_1.randomUUID)().replace(/-/g, "");
+    const generated = isLoadStrikeTraceSelector(endpoint.trackingField)
+        ? (0, node_crypto_1.randomUUID)()
+        : (0, node_crypto_1.randomUUID)().replace(/-/g, "");
     if (!injectTrackingValue(base, endpoint.trackingField, generated)) {
         throw new Error(`Tracking id could not be injected using selector \`${endpoint.trackingField}\` for endpoint \`${endpoint.name}\`. Verify the selector and payload shape.`);
     }
     return base;
 }
+function isLoadStrikeTraceSelector(selector) {
+    const trimmed = selector.trim();
+    return trimmed.toLowerCase() === exports.LOADSTRIKE_TRACE_ID_TRACKING_FIELD;
+}
 function toWirePayload(payload, endpoint) {
     return {
         headers: { ...(payload.headers ?? {}) },

package/dist/esm/index.js

@@ -2,5 +2,5 @@ export { LoadStrikeAutopilot, LoadStrikeAutopilotResult } from "./autopilot.js";
 export { LoadStrikeAutopilotReadiness } from "./autopilot-contracts.js";
 export { CrossPlatformScenarioConfigurator, ScenarioTrackingExtensions, LoadStrikeContext, LoadStrikePluginData, LoadStrikePluginDataTable, LoadStrikeNodeType, LoadStrikeReportFormat, LoadStrikeResponse, LoadStrikeLogLevel, LoadStrikeScenarioOperation, LoadStrikeOperationType, LoadStrikeRunner, LoadStrikeScenario, LoadStrikeSimulation, LoadStrikeMetric, LoadStrikeCounter, LoadStrikeGauge, LoadStrikeStep, LoadStrikeThreshold } from "./runtime.js";
 export { CorrelationStoreConfiguration, CrossPlatformTrackingRuntime, InMemoryCorrelationStore, RedisCorrelationStoreOptions, RedisCorrelationStore, TrackingPayloadBuilder, TrackingFieldSelector } from "./correlation.js";
-export { EndpointAdapterFactory, TrafficEndpointDefinition, HttpEndpointDefinition, KafkaEndpointDefinition, KafkaSaslOptions, RabbitMqEndpointDefinition, NatsEndpointDefinition, RedisStreamsEndpointDefinition, AzureEventHubsEndpointDefinition, DelegateStreamEndpointDefinition, PushDiffusionEndpointDefinition, HttpOAuth2ClientCredentialsOptions, HttpAuthOptions } from "./transports.js";
+export { EndpointAdapterFactory, LOADSTRIKE_TRACE_ID_HEADER, LOADSTRIKE_TRACE_ID_TRACKING_FIELD, TrafficEndpointDefinition, HttpEndpointDefinition, KafkaEndpointDefinition, KafkaSaslOptions, RabbitMqEndpointDefinition, NatsEndpointDefinition, RedisStreamsEndpointDefinition, AzureEventHubsEndpointDefinition, DelegateStreamEndpointDefinition, PushDiffusionEndpointDefinition, HttpOAuth2ClientCredentialsOptions, HttpAuthOptions } from "./transports.js";
 export { DatadogReportingSink, DatadogReportingSinkOptions, GrafanaLokiReportingSink, GrafanaLokiReportingSinkOptions, InfluxDbReportingSink, InfluxDbReportingSinkOptions, OtelCollectorReportingSink, OtelCollectorReportingSinkOptions, SplunkReportingSink, SplunkReportingSinkOptions, TimescaleDbReportingSink, TimescaleDbReportingSinkOptions } from "./sinks.js";

package/dist/esm/local.js

@@ -3,7 +3,7 @@ import * as fs from "node:fs";
 import * as childProcess from "node:child_process";
 import { createHash, createVerify, randomUUID } from "node:crypto";
 import { CorrelationStoreConfiguration, CrossPlatformTrackingRuntime, RedisCorrelationStoreOptions, RedisCorrelationStore, TrackingFieldSelector } from "./correlation.js";
-import { EndpointAdapterFactory } from "./transports.js";
+import { EndpointAdapterFactory, LOADSTRIKE_TRACE_ID_TRACKING_FIELD } from "./transports.js";
 const DEFAULT_LICENSING_API_BASE_URL = "https://licensing.loadstrike.com";
 const INTERNAL_BLACKBOX_LICENSING_API_BASE_URL_ENVIRONMENT_VARIABLE = "LOADSTRIKE_INTERNAL_BLACKBOX_API_BASE_URL";
 let developmentLicensingApiBaseUrlOverride;
@@ -810,10 +810,11 @@ async function evaluateScenarioOutcome(scenario, requestCount, context) {
     if (!Object.keys(sourceSpec).length) {
         return { requestCount, okCount: requestCount, failCount: 0 };
     }
-    const sourceEndpoint = mapEndpointSpec(sourceSpec);
+    const useLoadStrikeTraceIdHeader = toBoolean(pickValue(tracking, "UseLoadStrikeTraceIdHeader", "useLoadStrikeTraceIdHeader"), false);
+    const sourceEndpoint = mapEndpointSpec(sourceSpec, useLoadStrikeTraceIdHeader);
     const destinationSpec = asRecord(tracking.Destination);
     const hasDestination = Object.keys(destinationSpec).length > 0;
-    const destinationEndpoint = hasDestination ? mapEndpointSpec(destinationSpec) : null;
+    const destinationEndpoint = hasDestination ? mapEndpointSpec(destinationSpec, useLoadStrikeTraceIdHeader) : null;
     validateTrackingConfiguration(tracking, sourceEndpoint, destinationEndpoint);
     const sourceAdapter = EndpointAdapterFactory.create(sourceEndpoint);
     const destinationAdapter = destinationEndpoint ? EndpointAdapterFactory.create(destinationEndpoint) : null;
@@ -1069,7 +1070,7 @@ function inferLegacyHttpResponseSourceValue(value) {
             return undefined;
     }
 }
-function mapEndpointSpec(spec) {
+function mapEndpointSpec(spec, useLoadStrikeTraceIdHeader = false) {
     const pollIntervalOverride = pickValue(spec, "PollIntervalMs", "pollIntervalMs");
     const pollIntervalSeconds = pickValue(spec, "PollIntervalSeconds", "pollIntervalSeconds", "PollInterval");
     const pollIntervalMs = pollIntervalOverride != null && String(pollIntervalOverride).trim() !== ""
@@ -1102,7 +1103,7 @@ function mapEndpointSpec(spec) {
         kind: stringOrDefault(pickValue(spec, "Kind", "kind"), "DelegateStream"),
         mode: stringOrDefault(pickValue(spec, "Mode", "mode"), "Produce"),
         name: stringOrDefault(pickValue(spec, "Name", "name"), "endpoint"),
-        trackingField: readTrackingSelectorValue(pickValue(spec, "TrackingField", "trackingField"), "header:x-correlation-id"),
+        trackingField: readTrackingSelectorValue(pickValue(spec, "TrackingField", "trackingField"), useLoadStrikeTraceIdHeader ? LOADSTRIKE_TRACE_ID_TRACKING_FIELD : "header:x-correlation-id"),
         gatherByField: readOptionalTrackingSelectorValue(pickValue(spec, "GatherByField", "gatherByField")),
         autoGenerateTrackingIdWhenMissing: toBoolean(pickValue(spec, "AutoGenerateTrackingIdWhenMissing", "autoGenerateTrackingIdWhenMissing"), true),
         pollIntervalMs,
@@ -1202,10 +1203,13 @@ function normalizePayload(payload, endpoint, index) {
         return normalized;
     }
     const existing = readTrackingId(normalized, selector);
-    if (existing || !endpoint.autoGenerateTrackingIdWhenMissing) {
+    const endpointMode = String(endpoint.mode ?? "Produce").trim().toLowerCase();
+    if (existing || endpointMode === "consume" || !endpoint.autoGenerateTrackingIdWhenMissing) {
         return normalized;
     }
-    const generated = `${endpoint.name}-auto-${index + 1}`;
+    const generated = selector.toLowerCase() === LOADSTRIKE_TRACE_ID_TRACKING_FIELD
+        ? randomUUID()
+        : `${endpoint.name}-auto-${index + 1}`;
     if (selector.toLowerCase().startsWith("header:")) {
         const headerName = selector.slice("header:".length).trim();
         if (headerName) {