@superblocksteam/telemetry 2.0.83-next.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE.txt +87 -0
- package/README.md +155 -0
- package/dist/browser/index.d.ts +8 -0
- package/dist/browser/index.d.ts.map +1 -0
- package/dist/browser/index.js +19 -0
- package/dist/browser/index.js.map +1 -0
- package/dist/browser/init.d.ts +75 -0
- package/dist/browser/init.d.ts.map +1 -0
- package/dist/browser/init.js +169 -0
- package/dist/browser/init.js.map +1 -0
- package/dist/browser/resilient-exporter.d.ts +43 -0
- package/dist/browser/resilient-exporter.d.ts.map +1 -0
- package/dist/browser/resilient-exporter.js +88 -0
- package/dist/browser/resilient-exporter.js.map +1 -0
- package/dist/common/contracts/tier2-traces.d.ts +75 -0
- package/dist/common/contracts/tier2-traces.d.ts.map +1 -0
- package/dist/common/contracts/tier2-traces.js +186 -0
- package/dist/common/contracts/tier2-traces.js.map +1 -0
- package/dist/common/deployment-type.d.ts +18 -0
- package/dist/common/deployment-type.d.ts.map +1 -0
- package/dist/common/deployment-type.js +30 -0
- package/dist/common/deployment-type.js.map +1 -0
- package/dist/common/guardrails.d.ts +116 -0
- package/dist/common/guardrails.d.ts.map +1 -0
- package/dist/common/guardrails.js +189 -0
- package/dist/common/guardrails.js.map +1 -0
- package/dist/common/index.d.ts +16 -0
- package/dist/common/index.d.ts.map +1 -0
- package/dist/common/index.js +32 -0
- package/dist/common/index.js.map +1 -0
- package/dist/common/log-sanitizer.d.ts +78 -0
- package/dist/common/log-sanitizer.d.ts.map +1 -0
- package/dist/common/log-sanitizer.js +340 -0
- package/dist/common/log-sanitizer.js.map +1 -0
- package/dist/common/policy-evaluator.d.ts +103 -0
- package/dist/common/policy-evaluator.d.ts.map +1 -0
- package/dist/common/policy-evaluator.js +200 -0
- package/dist/common/policy-evaluator.js.map +1 -0
- package/dist/common/resource.d.ts +62 -0
- package/dist/common/resource.d.ts.map +1 -0
- package/dist/common/resource.js +106 -0
- package/dist/common/resource.js.map +1 -0
- package/dist/common/tier-hints.d.ts +182 -0
- package/dist/common/tier-hints.d.ts.map +1 -0
- package/dist/common/tier-hints.js +209 -0
- package/dist/common/tier-hints.js.map +1 -0
- package/dist/index.d.ts +43 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +76 -0
- package/dist/index.js.map +1 -0
- package/dist/lint/forbidden-attributes.d.ts +149 -0
- package/dist/lint/forbidden-attributes.d.ts.map +1 -0
- package/dist/lint/forbidden-attributes.js +214 -0
- package/dist/lint/forbidden-attributes.js.map +1 -0
- package/dist/lint/index.d.ts +9 -0
- package/dist/lint/index.d.ts.map +1 -0
- package/dist/lint/index.js +16 -0
- package/dist/lint/index.js.map +1 -0
- package/dist/llmobs/index.d.ts +22 -0
- package/dist/llmobs/index.d.ts.map +1 -0
- package/dist/llmobs/index.js +29 -0
- package/dist/llmobs/index.js.map +1 -0
- package/dist/llmobs/tier1-exporter.d.ts +146 -0
- package/dist/llmobs/tier1-exporter.d.ts.map +1 -0
- package/dist/llmobs/tier1-exporter.js +196 -0
- package/dist/llmobs/tier1-exporter.js.map +1 -0
- package/dist/llmobs/tier2-summarizer.d.ts +268 -0
- package/dist/llmobs/tier2-summarizer.d.ts.map +1 -0
- package/dist/llmobs/tier2-summarizer.js +650 -0
- package/dist/llmobs/tier2-summarizer.js.map +1 -0
- package/dist/node/exporters/resilient-exporter.d.ts +77 -0
- package/dist/node/exporters/resilient-exporter.d.ts.map +1 -0
- package/dist/node/exporters/resilient-exporter.js +129 -0
- package/dist/node/exporters/resilient-exporter.js.map +1 -0
- package/dist/node/index.d.ts +11 -0
- package/dist/node/index.d.ts.map +1 -0
- package/dist/node/index.js +24 -0
- package/dist/node/index.js.map +1 -0
- package/dist/node/init.d.ts +75 -0
- package/dist/node/init.d.ts.map +1 -0
- package/dist/node/init.js +245 -0
- package/dist/node/init.js.map +1 -0
- package/dist/node/log-processor.d.ts +83 -0
- package/dist/node/log-processor.d.ts.map +1 -0
- package/dist/node/log-processor.js +266 -0
- package/dist/node/log-processor.js.map +1 -0
- package/dist/node/metrics-client.d.ts +66 -0
- package/dist/node/metrics-client.d.ts.map +1 -0
- package/dist/node/metrics-client.js +193 -0
- package/dist/node/metrics-client.js.map +1 -0
- package/dist/node/traced-socket.d.ts +76 -0
- package/dist/node/traced-socket.d.ts.map +1 -0
- package/dist/node/traced-socket.js +261 -0
- package/dist/node/traced-socket.js.map +1 -0
- package/dist/testing/in-memory-exporter.d.ts +179 -0
- package/dist/testing/in-memory-exporter.d.ts.map +1 -0
- package/dist/testing/in-memory-exporter.js +254 -0
- package/dist/testing/in-memory-exporter.js.map +1 -0
- package/dist/testing/index.d.ts +8 -0
- package/dist/testing/index.d.ts.map +1 -0
- package/dist/testing/index.js +19 -0
- package/dist/testing/index.js.map +1 -0
- package/dist/testing/test-init.d.ts +80 -0
- package/dist/testing/test-init.d.ts.map +1 -0
- package/dist/testing/test-init.js +144 -0
- package/dist/testing/test-init.js.map +1 -0
- package/dist/types/index.d.ts +40 -0
- package/dist/types/index.d.ts.map +1 -0
- package/dist/types/index.js +23 -0
- package/dist/types/index.js.map +1 -0
- package/dist/types/policy.d.ts +92 -0
- package/dist/types/policy.d.ts.map +1 -0
- package/dist/types/policy.js +125 -0
- package/dist/types/policy.js.map +1 -0
- package/dist-esm/browser/index.d.ts +8 -0
- package/dist-esm/browser/index.d.ts.map +1 -0
- package/dist-esm/browser/index.js +9 -0
- package/dist-esm/browser/index.js.map +1 -0
- package/dist-esm/browser/init.d.ts +75 -0
- package/dist-esm/browser/init.d.ts.map +1 -0
- package/dist-esm/browser/init.js +162 -0
- package/dist-esm/browser/init.js.map +1 -0
- package/dist-esm/browser/resilient-exporter.d.ts +43 -0
- package/dist-esm/browser/resilient-exporter.d.ts.map +1 -0
- package/dist-esm/browser/resilient-exporter.js +84 -0
- package/dist-esm/browser/resilient-exporter.js.map +1 -0
- package/dist-esm/common/contracts/tier2-traces.d.ts +75 -0
- package/dist-esm/common/contracts/tier2-traces.d.ts.map +1 -0
- package/dist-esm/common/contracts/tier2-traces.js +178 -0
- package/dist-esm/common/contracts/tier2-traces.js.map +1 -0
- package/dist-esm/common/deployment-type.d.ts +18 -0
- package/dist-esm/common/deployment-type.d.ts.map +1 -0
- package/dist-esm/common/deployment-type.js +27 -0
- package/dist-esm/common/deployment-type.js.map +1 -0
- package/dist-esm/common/guardrails.d.ts +116 -0
- package/dist-esm/common/guardrails.d.ts.map +1 -0
- package/dist-esm/common/guardrails.js +179 -0
- package/dist-esm/common/guardrails.js.map +1 -0
- package/dist-esm/common/index.d.ts +16 -0
- package/dist-esm/common/index.d.ts.map +1 -0
- package/dist-esm/common/index.js +16 -0
- package/dist-esm/common/index.js.map +1 -0
- package/dist-esm/common/log-sanitizer.d.ts +78 -0
- package/dist-esm/common/log-sanitizer.d.ts.map +1 -0
- package/dist-esm/common/log-sanitizer.js +331 -0
- package/dist-esm/common/log-sanitizer.js.map +1 -0
- package/dist-esm/common/policy-evaluator.d.ts +103 -0
- package/dist-esm/common/policy-evaluator.d.ts.map +1 -0
- package/dist-esm/common/policy-evaluator.js +196 -0
- package/dist-esm/common/policy-evaluator.js.map +1 -0
- package/dist-esm/common/resource.d.ts +62 -0
- package/dist-esm/common/resource.d.ts.map +1 -0
- package/dist-esm/common/resource.js +100 -0
- package/dist-esm/common/resource.js.map +1 -0
- package/dist-esm/common/tier-hints.d.ts +182 -0
- package/dist-esm/common/tier-hints.d.ts.map +1 -0
- package/dist-esm/common/tier-hints.js +199 -0
- package/dist-esm/common/tier-hints.js.map +1 -0
- package/dist-esm/index.d.ts +43 -0
- package/dist-esm/index.d.ts.map +1 -0
- package/dist-esm/index.js +53 -0
- package/dist-esm/index.js.map +1 -0
- package/dist-esm/lint/forbidden-attributes.d.ts +149 -0
- package/dist-esm/lint/forbidden-attributes.d.ts.map +1 -0
- package/dist-esm/lint/forbidden-attributes.js +209 -0
- package/dist-esm/lint/forbidden-attributes.js.map +1 -0
- package/dist-esm/lint/index.d.ts +9 -0
- package/dist-esm/lint/index.d.ts.map +1 -0
- package/dist-esm/lint/index.js +9 -0
- package/dist-esm/lint/index.js.map +1 -0
- package/dist-esm/llmobs/index.d.ts +22 -0
- package/dist-esm/llmobs/index.d.ts.map +1 -0
- package/dist-esm/llmobs/index.js +22 -0
- package/dist-esm/llmobs/index.js.map +1 -0
- package/dist-esm/llmobs/tier1-exporter.d.ts +146 -0
- package/dist-esm/llmobs/tier1-exporter.d.ts.map +1 -0
- package/dist-esm/llmobs/tier1-exporter.js +190 -0
- package/dist-esm/llmobs/tier1-exporter.js.map +1 -0
- package/dist-esm/llmobs/tier2-summarizer.d.ts +268 -0
- package/dist-esm/llmobs/tier2-summarizer.d.ts.map +1 -0
- package/dist-esm/llmobs/tier2-summarizer.js +646 -0
- package/dist-esm/llmobs/tier2-summarizer.js.map +1 -0
- package/dist-esm/node/exporters/resilient-exporter.d.ts +77 -0
- package/dist-esm/node/exporters/resilient-exporter.d.ts.map +1 -0
- package/dist-esm/node/exporters/resilient-exporter.js +125 -0
- package/dist-esm/node/exporters/resilient-exporter.js.map +1 -0
- package/dist-esm/node/index.d.ts +11 -0
- package/dist-esm/node/index.d.ts.map +1 -0
- package/dist-esm/node/index.js +11 -0
- package/dist-esm/node/index.js.map +1 -0
- package/dist-esm/node/init.d.ts +75 -0
- package/dist-esm/node/init.d.ts.map +1 -0
- package/dist-esm/node/init.js +239 -0
- package/dist-esm/node/init.js.map +1 -0
- package/dist-esm/node/log-processor.d.ts +83 -0
- package/dist-esm/node/log-processor.d.ts.map +1 -0
- package/dist-esm/node/log-processor.js +261 -0
- package/dist-esm/node/log-processor.js.map +1 -0
- package/dist-esm/node/metrics-client.d.ts +66 -0
- package/dist-esm/node/metrics-client.d.ts.map +1 -0
- package/dist-esm/node/metrics-client.js +189 -0
- package/dist-esm/node/metrics-client.js.map +1 -0
- package/dist-esm/node/traced-socket.d.ts +76 -0
- package/dist-esm/node/traced-socket.d.ts.map +1 -0
- package/dist-esm/node/traced-socket.js +257 -0
- package/dist-esm/node/traced-socket.js.map +1 -0
- package/dist-esm/testing/in-memory-exporter.d.ts +179 -0
- package/dist-esm/testing/in-memory-exporter.d.ts.map +1 -0
- package/dist-esm/testing/in-memory-exporter.js +248 -0
- package/dist-esm/testing/in-memory-exporter.js.map +1 -0
- package/dist-esm/testing/index.d.ts +8 -0
- package/dist-esm/testing/index.d.ts.map +1 -0
- package/dist-esm/testing/index.js +8 -0
- package/dist-esm/testing/index.js.map +1 -0
- package/dist-esm/testing/test-init.d.ts +80 -0
- package/dist-esm/testing/test-init.d.ts.map +1 -0
- package/dist-esm/testing/test-init.js +137 -0
- package/dist-esm/testing/test-init.js.map +1 -0
- package/dist-esm/types/index.d.ts +40 -0
- package/dist-esm/types/index.d.ts.map +1 -0
- package/dist-esm/types/index.js +7 -0
- package/dist-esm/types/index.js.map +1 -0
- package/dist-esm/types/policy.d.ts +92 -0
- package/dist-esm/types/policy.d.ts.map +1 -0
- package/dist-esm/types/policy.js +122 -0
- package/dist-esm/types/policy.js.map +1 -0
- package/package.json +101 -0
|
@@ -0,0 +1,257 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* TracedSocket for WebSocket connections with OpenTelemetry tracing.
|
|
3
|
+
*
|
|
4
|
+
* This is a generic implementation that can be used across services.
|
|
5
|
+
* Server-specific behavior (like error handling) is injected via callbacks.
|
|
6
|
+
*/
|
|
7
|
+
import { ROOT_CONTEXT, SpanKind, SpanStatusCode, context as otelCtx, propagation, trace, } from "@opentelemetry/api";
|
|
8
|
+
import { SEMATTRS_MESSAGING_DESTINATION_KIND, SEMATTRS_MESSAGING_SYSTEM, } from "@opentelemetry/semantic-conventions";
|
|
9
|
+
import { ISocket, OBS_TAG_HTTP_ROUTE, OBS_TAG_APPLICATION_ID, OBS_TAG_BRANCH, OBS_TAG_API_ID, OBS_TAG_COMMIT_ID, OBS_TAG_HTTP_STATUS_CODE, OBS_SOCKET_STATUS_CODE, OBS_TAG_USER_EMAIL, OBS_TAG_ORG_ID, OBS_TAG_USER_TYPE, sanitizeError, } from "@superblocksteam/shared";
|
|
10
|
+
// Helper to cast ISocket's logger to pino-compatible logger
|
|
11
|
+
const asPinoLogger = (logger) => logger;
|
|
12
|
+
// Helper to check if a value is a Promise
|
|
13
|
+
const isPromise = (value) => {
|
|
14
|
+
return (value !== null &&
|
|
15
|
+
typeof value === "object" &&
|
|
16
|
+
"then" in value &&
|
|
17
|
+
typeof value.then === "function");
|
|
18
|
+
};
|
|
19
|
+
// End span with proper error handling
|
|
20
|
+
const endSpan = (traced, span, logger, setHttpStatusFromError) => {
|
|
21
|
+
try {
|
|
22
|
+
const result = traced();
|
|
23
|
+
if (isPromise(result)) {
|
|
24
|
+
return Promise.resolve(result)
|
|
25
|
+
.catch((err) => {
|
|
26
|
+
setHttpStatusFromError(span, typeof err === "string" ? new Error(err) : err, logger);
|
|
27
|
+
throw err;
|
|
28
|
+
})
|
|
29
|
+
.finally(() => span.end());
|
|
30
|
+
}
|
|
31
|
+
else {
|
|
32
|
+
span.end();
|
|
33
|
+
return result;
|
|
34
|
+
}
|
|
35
|
+
}
|
|
36
|
+
catch (error) {
|
|
37
|
+
setHttpStatusFromError(span, error, logger);
|
|
38
|
+
span.end();
|
|
39
|
+
throw error;
|
|
40
|
+
}
|
|
41
|
+
};
|
|
42
|
+
/**
|
|
43
|
+
* TracedSocket extends ISocket to add OpenTelemetry tracing capabilities.
|
|
44
|
+
*
|
|
45
|
+
* It manages spans for WebSocket handler calls to provide detailed tracing information.
|
|
46
|
+
*/
|
|
47
|
+
export class TracedSocket extends ISocket {
|
|
48
|
+
activeSpanByRequestId = new Map();
|
|
49
|
+
middlewareSpanByReqId = new Map();
|
|
50
|
+
lastActiveSpan;
|
|
51
|
+
requestHeaders;
|
|
52
|
+
tracer;
|
|
53
|
+
setHttpStatusFromError;
|
|
54
|
+
constructor(ws, requestHandlers, globalMiddlewares, config, options) {
|
|
55
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
56
|
+
super(ws, requestHandlers, globalMiddlewares, options);
|
|
57
|
+
this.requestHeaders = config.requestHeaders;
|
|
58
|
+
this.tracer = trace.getTracer("superblocks-server");
|
|
59
|
+
// Create error handler function
|
|
60
|
+
this.setHttpStatusFromError = (span, error, logger) => {
|
|
61
|
+
let status = 500;
|
|
62
|
+
if (config.getResponseMetaByError) {
|
|
63
|
+
const respMeta = config.getResponseMetaByError(error, logger);
|
|
64
|
+
status = respMeta.status;
|
|
65
|
+
}
|
|
66
|
+
span.setAttribute(OBS_TAG_HTTP_STATUS_CODE, status);
|
|
67
|
+
span.recordException(sanitizeError(error));
|
|
68
|
+
span.setStatus({ code: SpanStatusCode.ERROR, message: error.message });
|
|
69
|
+
};
|
|
70
|
+
// Log WebSocket connection establishment
|
|
71
|
+
asPinoLogger(this.logger).info({
|
|
72
|
+
"ws.event": "connection_established",
|
|
73
|
+
"ws.host": config.requestHeaders["host"] || "unknown",
|
|
74
|
+
}, "WebSocket connection established");
|
|
75
|
+
}
|
|
76
|
+
async callHandler(handler, params, ctx, client, next) {
|
|
77
|
+
let currentContext = otelCtx.active();
|
|
78
|
+
const middleWareSpan = this.middlewareSpanByReqId.get(ctx.requestId);
|
|
79
|
+
const activeSpan = this.activeSpanByRequestId.get(ctx.requestId);
|
|
80
|
+
// End middleware span since callHandler is called recursively
|
|
81
|
+
if (middleWareSpan) {
|
|
82
|
+
middleWareSpan.end();
|
|
83
|
+
if (activeSpan) {
|
|
84
|
+
currentContext = trace.setSpan(otelCtx.active(), activeSpan);
|
|
85
|
+
}
|
|
86
|
+
}
|
|
87
|
+
// After first (authHandler) these values will be set
|
|
88
|
+
const src = ctx;
|
|
89
|
+
const user = src.user;
|
|
90
|
+
const organization = src.organization;
|
|
91
|
+
if (this.lastActiveSpan && (user || organization)) {
|
|
92
|
+
this.lastActiveSpan.setAttribute(OBS_TAG_USER_EMAIL, user?.email ?? "");
|
|
93
|
+
this.lastActiveSpan.setAttribute(OBS_TAG_USER_TYPE, user?.getUserType?.() ?? "");
|
|
94
|
+
this.lastActiveSpan.setAttribute(OBS_TAG_ORG_ID, organization?.id ?? "");
|
|
95
|
+
}
|
|
96
|
+
let result;
|
|
97
|
+
const paramsObj = params;
|
|
98
|
+
await otelCtx.with(currentContext, async () => {
|
|
99
|
+
await this.tracer.startActiveSpan(`WS HANDLER ${handler.name === "" ? ctx.method : handler.name}`, {
|
|
100
|
+
attributes: {
|
|
101
|
+
[SEMATTRS_MESSAGING_SYSTEM]: "ws",
|
|
102
|
+
[SEMATTRS_MESSAGING_DESTINATION_KIND]: "websocket",
|
|
103
|
+
[OBS_TAG_USER_EMAIL]: src.user?.email ?? "",
|
|
104
|
+
[OBS_TAG_ORG_ID]: src.organization?.id ?? "",
|
|
105
|
+
[OBS_TAG_APPLICATION_ID]: String(paramsObj?.["applicationId"] ?? src.jwtClaims?.app_id ?? ""),
|
|
106
|
+
[OBS_TAG_API_ID]: String(paramsObj?.["apiId"] ?? ""),
|
|
107
|
+
[OBS_TAG_BRANCH]: String(paramsObj?.["branch"] ?? paramsObj?.["branchName"] ?? ""),
|
|
108
|
+
[OBS_TAG_COMMIT_ID]: String(paramsObj?.["commitId"] ?? ""),
|
|
109
|
+
},
|
|
110
|
+
kind: SpanKind.SERVER,
|
|
111
|
+
}, async (span) => {
|
|
112
|
+
this.middlewareSpanByReqId.set(ctx.requestId, span);
|
|
113
|
+
result = (await endSpan(() => super.callHandler(handler, params, ctx, client, next), span, asPinoLogger(this.logger), this.setHttpStatusFromError));
|
|
114
|
+
this.middlewareSpanByReqId.delete(ctx.requestId);
|
|
115
|
+
});
|
|
116
|
+
});
|
|
117
|
+
return result;
|
|
118
|
+
}
|
|
119
|
+
async handleMessage(message) {
|
|
120
|
+
if (message.request) {
|
|
121
|
+
const spanName = message.request.method;
|
|
122
|
+
const requestId = message.request.id;
|
|
123
|
+
const payload = message.request.payload;
|
|
124
|
+
const logData = {
|
|
125
|
+
"ws.event": "request_received",
|
|
126
|
+
"ws.method": spanName,
|
|
127
|
+
"ws.request_id": requestId,
|
|
128
|
+
"ws.host": this.requestHeaders["host"] || "unknown",
|
|
129
|
+
};
|
|
130
|
+
if (payload?.["applicationId"]) {
|
|
131
|
+
logData[OBS_TAG_APPLICATION_ID] = String(payload["applicationId"]);
|
|
132
|
+
}
|
|
133
|
+
if (payload?.["apiId"]) {
|
|
134
|
+
logData[OBS_TAG_API_ID] = String(payload["apiId"]);
|
|
135
|
+
}
|
|
136
|
+
if (payload?.["branch"] || payload?.["branchName"]) {
|
|
137
|
+
logData[OBS_TAG_BRANCH] = String(payload["branch"] ?? payload["branchName"]);
|
|
138
|
+
}
|
|
139
|
+
if (payload?.["commitId"]) {
|
|
140
|
+
logData[OBS_TAG_COMMIT_ID] = String(payload["commitId"]);
|
|
141
|
+
}
|
|
142
|
+
asPinoLogger(this.logger).info(logData, `WebSocket request: ${spanName}`);
|
|
143
|
+
await otelCtx.with(propagation.extract(ROOT_CONTEXT, message.request), async () => await this.tracer.startActiveSpan(`WS SERVER ${spanName}`, {
|
|
144
|
+
attributes: (() => {
|
|
145
|
+
const attrs = {
|
|
146
|
+
[SEMATTRS_MESSAGING_SYSTEM]: "ws",
|
|
147
|
+
[SEMATTRS_MESSAGING_DESTINATION_KIND]: "websocket",
|
|
148
|
+
[OBS_TAG_HTTP_ROUTE]: spanName,
|
|
149
|
+
};
|
|
150
|
+
if (payload?.["applicationId"]) {
|
|
151
|
+
attrs[OBS_TAG_APPLICATION_ID] = String(payload["applicationId"]);
|
|
152
|
+
}
|
|
153
|
+
if (payload?.["apiId"]) {
|
|
154
|
+
attrs[OBS_TAG_API_ID] = String(payload["apiId"]);
|
|
155
|
+
}
|
|
156
|
+
if (payload?.["branch"] || payload?.["branchName"]) {
|
|
157
|
+
attrs[OBS_TAG_BRANCH] = String(payload["branch"] ?? payload["branchName"]);
|
|
158
|
+
}
|
|
159
|
+
if (payload?.["commitId"]) {
|
|
160
|
+
attrs[OBS_TAG_COMMIT_ID] = String(payload["commitId"]);
|
|
161
|
+
}
|
|
162
|
+
return attrs;
|
|
163
|
+
})(),
|
|
164
|
+
kind: SpanKind.SERVER,
|
|
165
|
+
}, async (span) => {
|
|
166
|
+
this.lastActiveSpan = span;
|
|
167
|
+
this.activeSpanByRequestId.set(requestId, span);
|
|
168
|
+
const result = await endSpan(() => super.handleMessage(message), span, asPinoLogger(this.logger), this.setHttpStatusFromError);
|
|
169
|
+
this.activeSpanByRequestId.delete(requestId);
|
|
170
|
+
return result;
|
|
171
|
+
}));
|
|
172
|
+
}
|
|
173
|
+
else if (message.response) {
|
|
174
|
+
this.addEvent("ws.received-response", {
|
|
175
|
+
["ws.response.id"]: message.response.id,
|
|
176
|
+
});
|
|
177
|
+
return await super.handleMessage(message);
|
|
178
|
+
}
|
|
179
|
+
}
|
|
180
|
+
request(method, params, authorization) {
|
|
181
|
+
return this.tracer.startActiveSpan(`WS CLIENT ${method}`, (span) => {
|
|
182
|
+
this.addEvent("ws.send-request", {
|
|
183
|
+
["ws.request.method"]: method,
|
|
184
|
+
["ws.request.id"]: this.nxtRequestId,
|
|
185
|
+
});
|
|
186
|
+
const result = super.request(method, params, authorization);
|
|
187
|
+
span.end();
|
|
188
|
+
return result;
|
|
189
|
+
});
|
|
190
|
+
}
|
|
191
|
+
respond(requestId, result) {
|
|
192
|
+
this.addEvent("ws.send-response", { ["ws.response.id"]: requestId });
|
|
193
|
+
// Using http status codes so that our existing monitors can work with minimum change
|
|
194
|
+
const resultObj = result;
|
|
195
|
+
if (this.lastActiveSpan &&
|
|
196
|
+
resultObj["responseMeta"]?.status !== undefined) {
|
|
197
|
+
this.lastActiveSpan.setAttribute(OBS_TAG_HTTP_STATUS_CODE, resultObj["responseMeta"].status);
|
|
198
|
+
}
|
|
199
|
+
// Log successful WebSocket response
|
|
200
|
+
asPinoLogger(this.logger).info({
|
|
201
|
+
"ws.event": "response_sent",
|
|
202
|
+
"ws.request_id": requestId,
|
|
203
|
+
"ws.status": resultObj["responseMeta"]?.status || 200,
|
|
204
|
+
"ws.response_size": (() => {
|
|
205
|
+
try {
|
|
206
|
+
return JSON.stringify(result).length;
|
|
207
|
+
}
|
|
208
|
+
catch {
|
|
209
|
+
return -1;
|
|
210
|
+
}
|
|
211
|
+
})(),
|
|
212
|
+
"ws.host": this.requestHeaders["host"] || "unknown",
|
|
213
|
+
}, `WebSocket response sent for request ${requestId}`);
|
|
214
|
+
return super.respond(requestId, result);
|
|
215
|
+
}
|
|
216
|
+
respondError(requestId, error, exception) {
|
|
217
|
+
this.addEvent("ws.send-error", { ["ws.response.id"]: requestId });
|
|
218
|
+
if (this.lastActiveSpan) {
|
|
219
|
+
this.setHttpStatusFromError(this.lastActiveSpan, exception ?? new Error(error.message), asPinoLogger(this.logger));
|
|
220
|
+
this.lastActiveSpan.setAttribute(OBS_SOCKET_STATUS_CODE, error.code);
|
|
221
|
+
}
|
|
222
|
+
asPinoLogger(this.logger).error({
|
|
223
|
+
"ws.event": "error_response",
|
|
224
|
+
"ws.request_id": requestId,
|
|
225
|
+
"ws.error_code": error.code,
|
|
226
|
+
"ws.error_message": error.message,
|
|
227
|
+
"ws.exception_type": exception?.constructor.name,
|
|
228
|
+
"ws.exception_message": exception?.message,
|
|
229
|
+
"ws.host": this.requestHeaders["host"] || "unknown",
|
|
230
|
+
error: exception || error,
|
|
231
|
+
}, `WebSocket error for request ${requestId}: ${error.message}`);
|
|
232
|
+
return super.respondError(requestId, error, exception);
|
|
233
|
+
}
|
|
234
|
+
addEvent(eventName, attributes) {
|
|
235
|
+
if (this.lastActiveSpan) {
|
|
236
|
+
this.lastActiveSpan.addEvent(eventName, attributes, new Date());
|
|
237
|
+
}
|
|
238
|
+
}
|
|
239
|
+
close(reason) {
|
|
240
|
+
asPinoLogger(this.logger).info({
|
|
241
|
+
"ws.event": "connection_closed",
|
|
242
|
+
"ws.close_reason": reason || "normal",
|
|
243
|
+
"ws.host": this.requestHeaders["host"] || "unknown",
|
|
244
|
+
}, `WebSocket connection closed: ${reason || "normal"}`);
|
|
245
|
+
for (const span of this.middlewareSpanByReqId.values()) {
|
|
246
|
+
span.end();
|
|
247
|
+
}
|
|
248
|
+
this.middlewareSpanByReqId = new Map();
|
|
249
|
+
for (const span of this.activeSpanByRequestId.values()) {
|
|
250
|
+
span.end();
|
|
251
|
+
}
|
|
252
|
+
this.activeSpanByRequestId = new Map();
|
|
253
|
+
this.lastActiveSpan = undefined;
|
|
254
|
+
super.close(reason);
|
|
255
|
+
}
|
|
256
|
+
}
|
|
257
|
+
//# sourceMappingURL=traced-socket.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"traced-socket.js","sourceRoot":"","sources":["../../src/node/traced-socket.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAGL,YAAY,EAEZ,QAAQ,EACR,cAAc,EAEd,OAAO,IAAI,OAAO,EAClB,WAAW,EACX,KAAK,GAEN,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,mCAAmC,EACnC,yBAAyB,GAC1B,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EACL,OAAO,EAGP,kBAAkB,EAIlB,sBAAsB,EACtB,cAAc,EACd,cAAc,EACd,iBAAiB,EACjB,wBAAwB,EACxB,sBAAsB,EACtB,kBAAkB,EAClB,cAAc,EACd,iBAAiB,EAIjB,aAAa,GACd,MAAM,yBAAyB,CAAC;AAWjC,4DAA4D;AAC5D,MAAM,YAAY,GAAG,CAAC,MAAe,EAAc,EAAE,CAAC,MAAoB,CAAC;AA4C3E,0CAA0C;AAC1C,MAAM,SAAS,GAAG,CAAI,KAAc,EAAuB,EAAE;IAC3D,OAAO,CACL,KAAK,KAAK,IAAI;QACd,OAAO,KAAK,KAAK,QAAQ;QACzB,MAAM,IAAI,KAAK;QACf,OAAQ,KAAoB,CAAC,IAAI,KAAK,UAAU,CACjD,CAAC;AACJ,CAAC,CAAC;AAEF,sCAAsC;AACtC,MAAM,OAAO,GAAG,CACd,MAAwC,EACxC,IAAU,EACV,MAAkB,EAClB,sBAIS,EACmB,EAAE;IAC9B,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC;QACxB,IAAI,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;YACtB,OAAO,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;iBAC3B,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBACb,sBAAsB,CACpB,IAAI,EACJ,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,EAC9C,MAAM,CACP,CAAC;gBACF,MAAM,GAAG,CAAC;YACZ,CAAC,CAAC;iBACD,OAAO,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;QAC/B,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,MAAM,CAAC;QAChB,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,sBAAsB,CAAC,IAAI,EAAE,KAAc,EAAE,MAAM,CAAC,CAAC;QACrD,IAAI,CAAC,GAAG,EAAE,CAAC;QACX,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC,CAAC;AAmBF;;;;GAIG;AACH,MAAM,OAAO,YAIX,SAAQ,OAA4D;IAC5D,qBAAqB,GAAG,IAAI,GAAG,EAAgB,CAAC;IAChD,qBAAqB,GAAG,IAAI,GAAG,EAAgB,CAAC;IAChD,cAAc,CAAmB;IACxB,cAAc,CAAsB;IACpC,MAAM,CAAS;IACf,sBAAsB,CAI7B;IAEV,YACE,EAAiB,EACjB,eAIC,EACD,iBAAuE,EACvE,MAA0B,EAC1B,OAA4B;QAE5B,8DAA8D;QAC9D,KAAK,CAAC,EAAS,EAAE,eAAe,EAAE,iBAAiB,EAAE,OAAO,CAAC,CAAC;QAC9D,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,cAAc,CAAC;QAC5C,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;QAEpD,gCAAgC;QAChC,IAAI,CAAC,sBAAsB,GAAG,CAC5B,IAAU,EACV,KAAY,EACZ,MAAkB,EAClB,EAAE;YACF,IAAI,MAAM,GAAG,GAAG,CAAC;YACjB,IAAI,MAAM,CAAC,sBAAsB,EAAE,CAAC;gBAClC,MAAM,QAAQ,GAAG,MAAM,CAAC,sBAAsB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;gBAC9D,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC;YAC3B,CAAC;YACD,IAAI,CAAC,YAAY,CAAC,wBAAwB,EAAE,MAAM,CAAC,CAAC;YACpD,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,KAAK,CAAU,CAAC,CAAC;YACpD,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QACzE,CAAC,CAAC;QAEF,yCAAyC;QACzC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAC5B;YACE,UAAU,EAAE,wBAAwB;YACpC,SAAS,EAAE,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,SAAS;SACtD,EACD,kCAAkC,CACnC,CAAC;IACJ,CAAC;IAES,KAAK,CAAC,WAAW,CAMzB,OAKC,EACD,MAAc,EACd,GAAwB,EACxB,MAA2C,EAC3C,IAA2B;QAE3B,IAAI,cAAc,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;QACtC,MAAM,cAAc,GAAG,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACrE,MAAM,UAAU,GAAG,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAEjE,8DAA8D;QAC9D,IAAI,cAAc,EAAE,CAAC;YACnB,cAAc,CAAC,GAAG,EAAE,CAAC;YACrB,IAAI,UAAU,EAAE,CAAC;gBACf,cAAc,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,UAAU,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC;QAED,qDAAqD;QACrD,MAAM,GAAG,GAAG,GAAsC,CAAC;QACnD,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC;QACtB,MAAM,YAAY,GAAG,GAAG,CAAC,YAAY,CAAC;QACtC,IAAI,IAAI,CAAC,cAAc,IAAI,CAAC,IAAI,IAAI,YAAY,CAAC,EAAE,CAAC;YAClD,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,kBAAkB,EAAE,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;YACxE,IAAI,CAAC,cAAc,CAAC,YAAY,CAC9B,iBAAiB,EACjB,IAAI,EAAE,WAAW,EAAE,EAAE,IAAI,EAAE,CAC5B,CAAC;YACF,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,cAAc,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;QAC3E,CAAC;QAED,IAAI,MAA0B,CAAC;QAC/B,MAAM,SAAS,GAAG,MAA6C,CAAC;QAChE,MAAM,OAAO,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,IAAI,CAAC,MAAM,CAAC,eAAe,CAC/B,cAAc,OAAO,CAAC,IAAI,KAAK,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,EAC/D;gBACE,UAAU,EAAE;oBACV,CAAC,yBAAyB,CAAC,EAAE,IAAI;oBACjC,CAAC,mCAAmC,CAAC,EAAE,WAAW;oBAClD,CAAC,kBAAkB,CAAC,EAAE,GAAG,CAAC,IAAI,EAAE,KAAK,IAAI,EAAE;oBAC3C,CAAC,cAAc,CAAC,EAAE,GAAG,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE;oBAC5C,CAAC,sBAAsB,CAAC,EAAE,MAAM,CAC9B,SAAS,EAAE,CAAC,eAAe,CAAC,IAAI,GAAG,CAAC,SAAS,EAAE,MAAM,IAAI,EAAE,CAC5D;oBACD,CAAC,cAAc,CAAC,EAAE,MAAM,CAAC,SAAS,EAAE,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;oBACpD,CAAC,cAAc,CAAC,EAAE,MAAM,CACtB,SAAS,EAAE,CAAC,QAAQ,CAAC,IAAI,SAAS,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,CACzD;oBACD,CAAC,iBAAiB,CAAC,EAAE,MAAM,CAAC,SAAS,EAAE,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;iBAC3D;gBACD,IAAI,EAAE,QAAQ,CAAC,MAAM;aACtB,EACD,KAAK,EAAE,IAAU,EAAE,EAAE;gBACnB,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;gBACpD,MAAM,GAAG,CAAC,MAAM,OAAO,CACrB,GAAG,EAAE,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,EAC3D,IAAI,EACJ,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,EACzB,IAAI,CAAC,sBAAsB,CAC5B,CAAW,CAAC;gBACb,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACnD,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,OAAO,MAAgB,CAAC;IAC1B,CAAC;IAES,KAAK,CAAC,aAAa,CAAC,OAAsB;QAClD,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACpB,MAAM,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;YACxC,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YACrC,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,OAAwB,CAAC;YAEzD,MAAM,OAAO,GAAoC;gBAC/C,UAAU,EAAE,kBAAkB;gBAC9B,WAAW,EAAE,QAAQ;gBACrB,eAAe,EAAE,SAAS;gBAC1B,SAAS,EAAE,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,SAAS;aACpD,CAAC;YAEF,IAAI,OAAO,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;gBAC/B,OAAO,CAAC,sBAAsB,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC;YACrE,CAAC;YACD,IAAI,OAAO,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;gBACvB,OAAO,CAAC,cAAc,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;YACrD,CAAC;YACD,IAAI,OAAO,EAAE,CAAC,QAAQ,CAAC,IAAI,OAAO,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC;gBACnD,OAAO,CAAC,cAAc,CAAC,GAAG,MAAM,CAC9B,OAAO,CAAC,QAAQ,CAAC,IAAI,OAAO,CAAC,YAAY,CAAC,CAC3C,CAAC;YACJ,CAAC;YACD,IAAI,OAAO,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC1B,OAAO,CAAC,iBAAiB,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC;YAC3D,CAAC;YAED,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,EAAE,sBAAsB,QAAQ,EAAE,CAAC,CAAC;YAC1E,MAAM,OAAO,CAAC,IAAI,CAChB,WAAW,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,OAAO,CAAC,EAClD,KAAK,IAAI,EAAE,CACT,MAAM,IAAI,CAAC,MAAM,CAAC,eAAe,CAC/B,aAAa,QAAQ,EAAE,EACvB;gBACE,UAAU,EAAE,CAAC,GAAG,EAAE;oBAChB,MAAM,KAAK,GAAmC;wBAC5C,CAAC,yBAAyB,CAAC,EAAE,IAAI;wBACjC,CAAC,mCAAmC,CAAC,EAAE,WAAW;wBAClD,CAAC,kBAAkB,CAAC,EAAE,QAAQ;qBAC/B,CAAC;oBAEF,IAAI,OAAO,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;wBAC/B,KAAK,CAAC,sBAAsB,CAAC,GAAG,MAAM,CACpC,OAAO,CAAC,eAAe,CAAC,CACzB,CAAC;oBACJ,CAAC;oBACD,IAAI,OAAO,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;wBACvB,KAAK,CAAC,cAAc,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;oBACnD,CAAC;oBACD,IAAI,OAAO,EAAE,CAAC,QAAQ,CAAC,IAAI,OAAO,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC;wBACnD,KAAK,CAAC,cAAc,CAAC,GAAG,MAAM,CAC5B,OAAO,CAAC,QAAQ,CAAC,IAAI,OAAO,CAAC,YAAY,CAAC,CAC3C,CAAC;oBACJ,CAAC;oBACD,IAAI,OAAO,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC;wBAC1B,KAAK,CAAC,iBAAiB,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC;oBACzD,CAAC;oBAED,OAAO,KAAK,CAAC;gBACf,CAAC,CAAC,EAAE;gBACJ,IAAI,EAAE,QAAQ,CAAC,MAAM;aACtB,EACD,KAAK,EAAE,IAAU,EAAE,EAAE;gBACnB,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;gBAC3B,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;gBAChD,MAAM,MAAM,GAAG,MAAM,OAAO,CAC1B,GAAG,EAAE,CAAC,KAAK,CAAC,aAAa,CAAC,OAAO,CAAC,EAClC,IAAI,EACJ,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,EACzB,IAAI,CAAC,sBAAsB,CAC5B,CAAC;gBACF,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;gBAC7C,OAAO,MAAM,CAAC;YAChB,CAAC,CACF,CACJ,CAAC;QACJ,CAAC;aAAM,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YAC5B,IAAI,CAAC,QAAQ,CAAC,sBAAsB,EAAE;gBACpC,CAAC,gBAAgB,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAAE;aACxC,CAAC,CAAC;YACH,OAAO,MAAM,KAAK,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC;IAEM,OAAO,CACZ,MAAc,EACd,MAAc,EACd,aAAsB;QAEtB,OAAO,IAAI,CAAC,MAAM,CAAC,eAAe,CAChC,aAAa,MAAM,EAAE,EACrB,CAAC,IAAI,EAAmB,EAAE;YACxB,IAAI,CAAC,QAAQ,CAAC,iBAAiB,EAAE;gBAC/B,CAAC,mBAAmB,CAAC,EAAE,MAAM;gBAC7B,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,YAAY;aACrC,CAAC,CAAC;YACH,MAAM,MAAM,GAAoB,KAAK,CAAC,OAAO,CAC3C,MAAM,EACN,MAAM,EACN,aAAa,CACd,CAAC;YACF,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,MAAM,CAAC;QAChB,CAAC,CACF,CAAC;IACJ,CAAC;IAES,OAAO,CAAS,SAAiB,EAAE,MAAc;QACzD,IAAI,CAAC,QAAQ,CAAC,kBAAkB,EAAE,EAAE,CAAC,gBAAgB,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;QACrE,qFAAqF;QACrF,MAAM,SAAS,GAAG,MAAiD,CAAC;QACpE,IACE,IAAI,CAAC,cAAc;YACnB,SAAS,CAAC,cAAc,CAAC,EAAE,MAAM,KAAK,SAAS,EAC/C,CAAC;YACD,IAAI,CAAC,cAAc,CAAC,YAAY,CAC9B,wBAAwB,EACxB,SAAS,CAAC,cAAc,CAAC,CAAC,MAAgB,CAC3C,CAAC;QACJ,CAAC;QAED,oCAAoC;QACpC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAC5B;YACE,UAAU,EAAE,eAAe;YAC3B,eAAe,EAAE,SAAS;YAC1B,WAAW,EAAG,SAAS,CAAC,cAAc,CAAC,EAAE,MAAiB,IAAI,GAAG;YACjE,kBAAkB,EAAE,CAAC,GAAG,EAAE;gBACxB,IAAI,CAAC;oBACH,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC;gBACvC,CAAC;gBAAC,MAAM,CAAC;oBACP,OAAO,CAAC,CAAC,CAAC;gBACZ,CAAC;YACH,CAAC,CAAC,EAAE;YACJ,SAAS,EAAE,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,SAAS;SACpD,EACD,uCAAuC,SAAS,EAAE,CACnD,CAAC;QAEF,OAAO,KAAK,CAAC,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IAC1C,CAAC;IAES,YAAY,CACpB,SAAiB,EACjB,KAAkB,EAClB,SAAiB;QAEjB,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,EAAE,CAAC,gBAAgB,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;QAClE,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,IAAI,CAAC,sBAAsB,CACzB,IAAI,CAAC,cAAc,EACnB,SAAS,IAAI,IAAI,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,EACrC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAC1B,CAAC;YACF,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,sBAAsB,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QACvE,CAAC;QAED,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,CAC7B;YACE,UAAU,EAAE,gBAAgB;YAC5B,eAAe,EAAE,SAAS;YAC1B,eAAe,EAAE,KAAK,CAAC,IAAI;YAC3B,kBAAkB,EAAE,KAAK,CAAC,OAAO;YACjC,mBAAmB,EAAE,SAAS,EAAE,WAAW,CAAC,IAAI;YAChD,sBAAsB,EAAE,SAAS,EAAE,OAAO;YAC1C,SAAS,EAAE,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,SAAS;YACnD,KAAK,EAAE,SAAS,IAAI,KAAK;SAC1B,EACD,+BAA+B,SAAS,KAAK,KAAK,CAAC,OAAO,EAAE,CAC7D,CAAC;QAEF,OAAO,KAAK,CAAC,YAAY,CAAC,SAAS,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;IACzD,CAAC;IAEO,QAAQ,CACd,SAAiB,EACjB,UAAkC;QAElC,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,SAAS,EAAE,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC;QAClE,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,MAAe;QAC1B,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAC5B;YACE,UAAU,EAAE,mBAAmB;YAC/B,iBAAiB,EAAE,MAAM,IAAI,QAAQ;YACrC,SAAS,EAAE,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,SAAS;SACpD,EACD,gCAAgC,MAAM,IAAI,QAAQ,EAAE,CACrD,CAAC;QAEF,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;YACvD,IAAI,CAAC,GAAG,EAAE,CAAC;QACb,CAAC;QACD,IAAI,CAAC,qBAAqB,GAAG,IAAI,GAAG,EAAE,CAAC;QAEvC,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;YACvD,IAAI,CAAC,GAAG,EAAE,CAAC;QACb,CAAC;QACD,IAAI,CAAC,qBAAqB,GAAG,IAAI,GAAG,EAAE,CAAC;QACvC,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;QAChC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IACtB,CAAC;CACF"}
|
|
@@ -0,0 +1,179 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* In-Memory Exporters for Testing
|
|
3
|
+
*
|
|
4
|
+
* These exporters capture telemetry data in memory for assertion in tests.
|
|
5
|
+
* No external connections are made.
|
|
6
|
+
*/
|
|
7
|
+
import type { SpanExporter, ReadableSpan } from '@opentelemetry/sdk-trace-base';
|
|
8
|
+
import { ExportResult } from '@opentelemetry/core';
|
|
9
|
+
/**
|
|
10
|
+
* In-memory span exporter for testing.
|
|
11
|
+
* Captures all exported spans for assertion.
|
|
12
|
+
*/
|
|
13
|
+
export declare class InMemorySpanExporter implements SpanExporter {
|
|
14
|
+
private spans;
|
|
15
|
+
private stopped;
|
|
16
|
+
/**
|
|
17
|
+
* Export spans to in-memory storage.
|
|
18
|
+
*/
|
|
19
|
+
export(spans: ReadableSpan[], resultCallback: (result: ExportResult) => void): void;
|
|
20
|
+
/**
|
|
21
|
+
* Shutdown the exporter.
|
|
22
|
+
*/
|
|
23
|
+
shutdown(): Promise<void>;
|
|
24
|
+
/**
|
|
25
|
+
* Force flush (no-op for in-memory).
|
|
26
|
+
*/
|
|
27
|
+
forceFlush(): Promise<void>;
|
|
28
|
+
/**
|
|
29
|
+
* Get all captured spans.
|
|
30
|
+
*/
|
|
31
|
+
getSpans(): ReadableSpan[];
|
|
32
|
+
/**
|
|
33
|
+
* Get the number of captured spans.
|
|
34
|
+
*/
|
|
35
|
+
getSpanCount(): number;
|
|
36
|
+
/**
|
|
37
|
+
* Clear captured spans.
|
|
38
|
+
*/
|
|
39
|
+
reset(): void;
|
|
40
|
+
/**
|
|
41
|
+
* Find spans by name.
|
|
42
|
+
*
|
|
43
|
+
* @param name - Span name to search for
|
|
44
|
+
* @returns Matching spans
|
|
45
|
+
*/
|
|
46
|
+
findSpansByName(name: string): ReadableSpan[];
|
|
47
|
+
/**
|
|
48
|
+
* Find spans by name pattern (regex).
|
|
49
|
+
*
|
|
50
|
+
* @param pattern - Regex pattern to match span names
|
|
51
|
+
* @returns Matching spans
|
|
52
|
+
*/
|
|
53
|
+
findSpansByPattern(pattern: RegExp): ReadableSpan[];
|
|
54
|
+
/**
|
|
55
|
+
* Find spans with a specific attribute value.
|
|
56
|
+
*
|
|
57
|
+
* @param key - Attribute key
|
|
58
|
+
* @param value - Attribute value (or regex for pattern matching)
|
|
59
|
+
* @returns Matching spans
|
|
60
|
+
*/
|
|
61
|
+
findSpansByAttribute(key: string, value: string | RegExp): ReadableSpan[];
|
|
62
|
+
/**
|
|
63
|
+
* Assert no spans have a specific attribute value.
|
|
64
|
+
* Useful for ensuring Tier 1 data doesn't leak.
|
|
65
|
+
*
|
|
66
|
+
* @param key - Attribute key
|
|
67
|
+
* @param value - Forbidden attribute value (or regex pattern)
|
|
68
|
+
* @throws Error if forbidden attribute is found
|
|
69
|
+
*/
|
|
70
|
+
assertNoAttribute(key: string, value: string | RegExp): void;
|
|
71
|
+
/**
|
|
72
|
+
* Assert all spans have a required attribute.
|
|
73
|
+
*
|
|
74
|
+
* @param key - Attribute key
|
|
75
|
+
* @throws Error if any span is missing the attribute
|
|
76
|
+
*/
|
|
77
|
+
assertAllHaveAttribute(key: string): void;
|
|
78
|
+
/**
|
|
79
|
+
* Get span names for debugging.
|
|
80
|
+
*/
|
|
81
|
+
getSpanNames(): string[];
|
|
82
|
+
}
|
|
83
|
+
/**
|
|
84
|
+
* Metric data point captured by in-memory exporter.
|
|
85
|
+
*/
|
|
86
|
+
export interface CapturedMetric {
|
|
87
|
+
name: string;
|
|
88
|
+
description?: string;
|
|
89
|
+
unit?: string;
|
|
90
|
+
type: 'counter' | 'gauge' | 'histogram' | 'observable';
|
|
91
|
+
dataPoints: Array<{
|
|
92
|
+
value: number;
|
|
93
|
+
attributes: Record<string, unknown>;
|
|
94
|
+
timestamp: number;
|
|
95
|
+
}>;
|
|
96
|
+
}
|
|
97
|
+
/**
|
|
98
|
+
* In-memory metric exporter for testing.
|
|
99
|
+
*/
|
|
100
|
+
export declare class InMemoryMetricExporter {
|
|
101
|
+
private metrics;
|
|
102
|
+
/**
|
|
103
|
+
* Export metrics to in-memory storage.
|
|
104
|
+
*/
|
|
105
|
+
export(metrics: CapturedMetric[]): void;
|
|
106
|
+
/**
|
|
107
|
+
* Get all captured metrics.
|
|
108
|
+
*/
|
|
109
|
+
getMetrics(): CapturedMetric[];
|
|
110
|
+
/**
|
|
111
|
+
* Find metrics by name.
|
|
112
|
+
*
|
|
113
|
+
* @param name - Metric name
|
|
114
|
+
* @returns Matching metrics
|
|
115
|
+
*/
|
|
116
|
+
findMetricsByName(name: string): CapturedMetric[];
|
|
117
|
+
/**
|
|
118
|
+
* Clear captured metrics.
|
|
119
|
+
*/
|
|
120
|
+
reset(): void;
|
|
121
|
+
/**
|
|
122
|
+
* Assert no metrics have a specific label value.
|
|
123
|
+
*
|
|
124
|
+
* @param labelKey - Label key
|
|
125
|
+
* @param value - Forbidden value
|
|
126
|
+
* @throws Error if forbidden label is found
|
|
127
|
+
*/
|
|
128
|
+
assertNoLabel(labelKey: string, value: string | RegExp): void;
|
|
129
|
+
}
|
|
130
|
+
/**
|
|
131
|
+
* Log record captured by in-memory exporter.
|
|
132
|
+
*/
|
|
133
|
+
export interface CapturedLogRecord {
|
|
134
|
+
timestamp: number;
|
|
135
|
+
severityNumber: number;
|
|
136
|
+
severityText?: string;
|
|
137
|
+
body?: unknown;
|
|
138
|
+
attributes: Record<string, unknown>;
|
|
139
|
+
}
|
|
140
|
+
/**
|
|
141
|
+
* In-memory log exporter for testing.
|
|
142
|
+
*/
|
|
143
|
+
export declare class InMemoryLogExporter {
|
|
144
|
+
private logs;
|
|
145
|
+
/**
|
|
146
|
+
* Export logs to in-memory storage.
|
|
147
|
+
*/
|
|
148
|
+
export(logs: CapturedLogRecord[]): void;
|
|
149
|
+
/**
|
|
150
|
+
* Get all captured logs.
|
|
151
|
+
*/
|
|
152
|
+
getLogs(): CapturedLogRecord[];
|
|
153
|
+
/**
|
|
154
|
+
* Find logs by severity.
|
|
155
|
+
*
|
|
156
|
+
* @param severityNumber - Severity level (e.g., 9 for INFO, 13 for WARN, 17 for ERROR)
|
|
157
|
+
* @returns Matching logs
|
|
158
|
+
*/
|
|
159
|
+
findLogsBySeverity(severityNumber: number): CapturedLogRecord[];
|
|
160
|
+
/**
|
|
161
|
+
* Find logs containing text.
|
|
162
|
+
*
|
|
163
|
+
* @param text - Text to search for in log body
|
|
164
|
+
* @returns Matching logs
|
|
165
|
+
*/
|
|
166
|
+
findLogsContaining(text: string | RegExp): CapturedLogRecord[];
|
|
167
|
+
/**
|
|
168
|
+
* Clear captured logs.
|
|
169
|
+
*/
|
|
170
|
+
reset(): void;
|
|
171
|
+
/**
|
|
172
|
+
* Assert no logs contain sensitive data.
|
|
173
|
+
*
|
|
174
|
+
* @param pattern - Pattern to check for (forbidden content)
|
|
175
|
+
* @throws Error if forbidden content is found
|
|
176
|
+
*/
|
|
177
|
+
assertNoLogContains(pattern: RegExp): void;
|
|
178
|
+
}
|
|
179
|
+
//# sourceMappingURL=in-memory-exporter.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"in-memory-exporter.d.ts","sourceRoot":"","sources":["../../src/testing/in-memory-exporter.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAChF,OAAO,EAAE,YAAY,EAAoB,MAAM,qBAAqB,CAAC;AAErE;;;GAGG;AACH,qBAAa,oBAAqB,YAAW,YAAY;IACvD,OAAO,CAAC,KAAK,CAAsB;IACnC,OAAO,CAAC,OAAO,CAAS;IAExB;;OAEG;IACH,MAAM,CAAC,KAAK,EAAE,YAAY,EAAE,EAAE,cAAc,EAAE,CAAC,MAAM,EAAE,YAAY,KAAK,IAAI,GAAG,IAAI;IASnF;;OAEG;IACH,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAKzB;;OAEG;IACH,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAI3B;;OAEG;IACH,QAAQ,IAAI,YAAY,EAAE;IAI1B;;OAEG;IACH,YAAY,IAAI,MAAM;IAItB;;OAEG;IACH,KAAK,IAAI,IAAI;IAIb;;;;;OAKG;IACH,eAAe,CAAC,IAAI,EAAE,MAAM,GAAG,YAAY,EAAE;IAI7C;;;;;OAKG;IACH,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,YAAY,EAAE;IAInD;;;;;;OAMG;IACH,oBAAoB,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,YAAY,EAAE;IAWzE;;;;;;;OAOG;IACH,iBAAiB,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAa5D;;;;;OAKG;IACH,sBAAsB,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;IAQzC;;OAEG;IACH,YAAY,IAAI,MAAM,EAAE;CAGzB;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,SAAS,GAAG,OAAO,GAAG,WAAW,GAAG,YAAY,CAAC;IACvD,UAAU,EAAE,KAAK,CAAC;QAChB,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACpC,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC,CAAC;CACJ;AAED;;GAEG;AACH,qBAAa,sBAAsB;IACjC,OAAO,CAAC,OAAO,CAAwB;IAEvC;;OAEG;IACH,MAAM,CAAC,OAAO,EAAE,cAAc,EAAE,GAAG,IAAI;IAIvC;;OAEG;IACH,UAAU,IAAI,cAAc,EAAE;IAI9B;;;;;OAKG;IACH,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc,EAAE;IAIjD;;OAEG;IACH,KAAK,IAAI,IAAI;IAIb;;;;;;OAMG;IACH,aAAa,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;CAgB9D;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,SAAS,EAAE,MAAM,CAAC;IAClB,cAAc,EAAE,MAAM,CAAC;IACvB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,IAAI,CAAC,EAAE,OAAO,CAAC;IACf,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACrC;AAED;;GAEG;AACH,qBAAa,mBAAmB;IAC9B,OAAO,CAAC,IAAI,CAA2B;IAEvC;;OAEG;IACH,MAAM,CAAC,IAAI,EAAE,iBAAiB,EAAE,GAAG,IAAI;IAIvC;;OAEG;IACH,OAAO,IAAI,iBAAiB,EAAE;IAI9B;;;;;OAKG;IACH,kBAAkB,CAAC,cAAc,EAAE,MAAM,GAAG,iBAAiB,EAAE;IAI/D;;;;;OAKG;IACH,kBAAkB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,iBAAiB,EAAE;IAQ9D;;OAEG;IACH,KAAK,IAAI,IAAI;IAIb;;;;;OAKG;IACH,mBAAmB,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;CAS3C"}
|