Skip to content

Commit dab3ea3

Browse files
feat: store tracecontext object in additionalcontext
1 parent 4db8ace commit dab3ea3

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

src/Voxel.MiddyNet.Tracing.ApiGatewayMiddleware/ApiGatewayTracingMiddleware.cs

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,9 @@ namespace Voxel.MiddyNet.Tracing.ApiGatewayMiddleware
66
{
77
public class ApiGatewayTracingMiddleware : ILambdaMiddleware<APIGatewayProxyRequest, APIGatewayProxyResponse>
88
{
9+
private const string TraceContextKey = "TraceContext";
910
private const string TraceParentHeaderName = "traceparent";
1011
private const string TraceStateHeaderName = "tracestate";
11-
private const string TraceIdHeaderName = "trace-id";
1212

1313
public Task Before(APIGatewayProxyRequest apiGatewayEvent, MiddyNetContext context)
1414
{
@@ -22,17 +22,17 @@ public Task Before(APIGatewayProxyRequest apiGatewayEvent, MiddyNetContext conte
2222

2323
var traceContext = TraceContext.Handle(traceParentHeaderValue, traceStateHeaderValue);
2424

25+
context.AdditionalContext.Add(TraceContextKey, traceContext);
26+
2527
context.Logger.EnrichWith(new LogProperty(TraceParentHeaderName, traceContext.TraceParent));
2628
context.Logger.EnrichWith(new LogProperty(TraceStateHeaderName, traceContext.TraceState));
2729
context.Logger.EnrichWith(new LogProperty(TraceIdHeaderName, traceContext.TraceId));
28-
29-
context.AdditionalContext.Add(TraceParentHeaderName, traceContext.TraceParent);
30-
context.AdditionalContext.Add(TraceStateHeaderName, traceContext.TraceState);
31-
context.AdditionalContext.Add(TraceIdHeaderName, traceContext.TraceId);
32-
30+
3331
return Task.CompletedTask;
3432
}
3533

34+
private const string TraceIdHeaderName = "trace-id";
35+
3636
public Task<APIGatewayProxyResponse> After(APIGatewayProxyResponse lambdaResponse, MiddyNetContext context)
3737
{
3838
return Task.FromResult(lambdaResponse);

0 commit comments

Comments
 (0)