Add diagnostic context overriding (#136)

* Add OverrideDiagnosticContextAttribute * Update * Update * Clean up
mindbox-cloud · Apr 26, 2024 · 0cbdc7b · 0cbdc7b
1 parent 38ccfb7
commit 0cbdc7b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/AspNetCore/UseDiagnosticContextAttribute.cs b/src/AspNetCore/UseDiagnosticContextAttribute.cs
@@ -38,6 +38,8 @@ public override async Task OnActionExecutionAsync(ActionExecutingContext context
 			metricPath: _metricName,
 			metricsTypesOverride: CreateMetricsTypesOverride(context));
 
+		diagnosticContext = CreateDiagnosticContextOverride(diagnosticContext, context);
+
 		context.HttpContext.Response.RegisterForDispose(diagnosticContext);
 		context.HttpContext.StoreDiagnosticContext(diagnosticContext);
 
@@ -48,4 +50,7 @@ public override async Task OnActionExecutionAsync(ActionExecutingContext context
 	}
 
 	protected virtual MetricsType[]? CreateMetricsTypesOverride(ActionExecutingContext context) => null;
+
+	protected virtual IDiagnosticContext CreateDiagnosticContextOverride(IDiagnosticContext original, ActionExecutingContext context)
+		=> original;
 }
diff --git a/src/Directory.Build.props b/src/Directory.Build.props
@@ -15,7 +15,7 @@
   </PropertyGroup>
   <PropertyGroup>
     <VersionMajor>9</VersionMajor>
-    <VersionMinor>0</VersionMinor>
+    <VersionMinor>1</VersionMinor>
     <BuildNumber>$(BuildNumber)</BuildNumber>
     <BuildNumber Condition="'$(BuildNumber)' == ''">0</BuildNumber>
     <PackageVersion>$(VersionMajor).$(VersionMinor).$(BuildNumber)</PackageVersion>