-
Notifications
You must be signed in to change notification settings - Fork 1.3k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[frontend] Format InstrumentationMiddleware.ts (#607)
- Loading branch information
1 parent
5a2c0ef
commit 126f0e8
Showing
1 changed file
with
50 additions
and
51 deletions.
There are no files selected for viewing
101 changes: 50 additions & 51 deletions
101
src/frontend/utils/telemetry/InstrumentationMiddleware.ts
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,62 +1,61 @@ | ||
| import {NextApiHandler} from 'next'; | ||
| import {context, Exception, propagation, Span, SpanKind, SpanStatusCode, trace} from '@opentelemetry/api'; | ||
| import {SemanticAttributes} from '@opentelemetry/semantic-conventions'; | ||
| import { NextApiHandler } from 'next'; | ||
| import { context, Exception, propagation, Span, SpanKind, SpanStatusCode, trace } from '@opentelemetry/api'; | ||
| import { SemanticAttributes } from '@opentelemetry/semantic-conventions'; | ||
|
|
||
| const InstrumentationMiddleware = (handler: NextApiHandler): NextApiHandler => { | ||
| return async (request, response) => { | ||
| const {headers, method, url = '', httpVersion} = request; | ||
| const [target] = url.split('?'); | ||
|
|
||
| let span; | ||
| const baggage = propagation.getBaggage(context.active()); | ||
| if (baggage?.getEntry("synthetic_request")?.value == "true") { | ||
| // if synthetic_request baggage is set, create a new trace linked to the span in context | ||
| // this span will look similar to the auto-instrumented HTTP span | ||
| const syntheticSpan = trace.getSpan(context.active()) as Span; | ||
| const tracer = trace.getTracer(process.env.OTEL_SERVICE_NAME as string); | ||
| span = tracer.startSpan(`HTTP ${method}`, { | ||
| root: true, | ||
| kind: SpanKind.SERVER, | ||
| links: [{context: syntheticSpan.spanContext()}], | ||
| attributes: { | ||
| "app.synthetic_request": true, | ||
| [SemanticAttributes.HTTP_TARGET]: target, | ||
| [SemanticAttributes.HTTP_STATUS_CODE]: response.statusCode, | ||
| [SemanticAttributes.HTTP_METHOD]: method, | ||
| [SemanticAttributes.HTTP_USER_AGENT]: headers['user-agent'] || '', | ||
| [SemanticAttributes.HTTP_URL]: `${headers.host}${url}`, | ||
| [SemanticAttributes.HTTP_FLAVOR]: httpVersion, | ||
| } | ||
| }); | ||
|
|
||
| } else { | ||
| // continue current trace/span | ||
| span = trace.getSpan(context.active()) as Span; | ||
| } | ||
| return async (request, response) => { | ||
| const { headers, method, url = '', httpVersion } = request; | ||
| const [target] = url.split('?'); | ||
|
|
||
| let span; | ||
| const baggage = propagation.getBaggage(context.active()); | ||
| if (baggage?.getEntry('synthetic_request')?.value == 'true') { | ||
| // if synthetic_request baggage is set, create a new trace linked to the span in context | ||
| // this span will look similar to the auto-instrumented HTTP span | ||
| const syntheticSpan = trace.getSpan(context.active()) as Span; | ||
| const tracer = trace.getTracer(process.env.OTEL_SERVICE_NAME as string); | ||
| span = tracer.startSpan(`HTTP ${method}`, { | ||
| root: true, | ||
| kind: SpanKind.SERVER, | ||
| links: [{ context: syntheticSpan.spanContext() }], | ||
| attributes: { | ||
| 'app.synthetic_request': true, | ||
| [SemanticAttributes.HTTP_TARGET]: target, | ||
| [SemanticAttributes.HTTP_STATUS_CODE]: response.statusCode, | ||
| [SemanticAttributes.HTTP_METHOD]: method, | ||
| [SemanticAttributes.HTTP_USER_AGENT]: headers['user-agent'] || '', | ||
| [SemanticAttributes.HTTP_URL]: `${headers.host}${url}`, | ||
| [SemanticAttributes.HTTP_FLAVOR]: httpVersion, | ||
| }, | ||
| }); | ||
| } else { | ||
| // continue current trace/span | ||
| span = trace.getSpan(context.active()) as Span; | ||
| } | ||
|
|
||
| try { | ||
| await runWithSpan(span, async () => handler(request, response)); | ||
| } catch (error) { | ||
| span.recordException(error as Exception); | ||
| span.setStatus({code: SpanStatusCode.ERROR}); | ||
| throw error; | ||
| } finally { | ||
| span.end(); | ||
| } | ||
| }; | ||
| try { | ||
| await runWithSpan(span, async () => handler(request, response)); | ||
| } catch (error) { | ||
| span.recordException(error as Exception); | ||
| span.setStatus({ code: SpanStatusCode.ERROR }); | ||
| throw error; | ||
| } finally { | ||
| span.end(); | ||
| } | ||
| }; | ||
| }; | ||
|
|
||
| async function runWithSpan(parentSpan: Span, fn: () => Promise<unknown>) { | ||
| const ctx = trace.setSpan(context.active(), parentSpan); | ||
| const ctx = trace.setSpan(context.active(), parentSpan); | ||
|
|
||
| try { | ||
| return await context.with(ctx, fn); | ||
| } catch (error) { | ||
| parentSpan.recordException(error as Exception); | ||
| parentSpan.setStatus({ code: SpanStatusCode.ERROR }); | ||
| try { | ||
| return await context.with(ctx, fn); | ||
| } catch (error) { | ||
| parentSpan.recordException(error as Exception); | ||
| parentSpan.setStatus({ code: SpanStatusCode.ERROR }); | ||
|
|
||
| throw error; | ||
| } | ||
| throw error; | ||
| } | ||
| } | ||
|
|
||
| export default InstrumentationMiddleware; |