Merge pull request #700 from leancodepl/more-tracing

Improve tracing across CQRS elements

CHANGELOG.md

@@ -5,12 +5,13 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](http://keepachangelog.com/)
 but this project DOES NOT adhere to [Semantic Versioning](http://semver.org/).
 
-## 8.1 (Unreleased)
+## 8.1
 * `LeanCode.Kratos` has switched from official (and at the time outdated) client package `Ory.Kratos.Client` to our own `LeanCode.Kratos.Client` that is maintained at https://github.com/leancodepl/dotnet-kratos-client and with this the usage has changed:
-  * Underlying JSON (de)serializer is now STJ instead of JSON.NET
-  * Values returned from client's API calls are now wrapped in response objects that provide additional info
-  * Models were rebuilt, adding support for NRT and wrapping optional properties in `Option<T>` (importantly, the latter can change how identity metadata should be read)
-  * The client comes with its own DI registration extensions, changing how it should be added to DI container
+* Underlying JSON (de)serializer is now STJ instead of JSON.NET
+* Values returned from client's API calls are now wrapped in response objects that provide additional info
+* Models were rebuilt, adding support for NRT and wrapping optional properties in `Option<T>` (importantly, the latter can change how identity metadata should be read)
+* The client comes with its own DI registration extensions, changing how it should be added to DI container
+* Better tracing across CQRS executions
 
 ## 8.0
 

src/CQRS/LeanCode.CQRS.AspNetCore/Local/MiddlewareBasedLocalExecutor.cs

@@ -5,7 +5,6 @@
 using LeanCode.OpenTelemetry;
 using Microsoft.AspNetCore.Builder;
 using Microsoft.AspNetCore.Http;
-using Microsoft.AspNetCore.Mvc.Infrastructure;
 using Microsoft.Extensions.DependencyInjection;
 
 namespace LeanCode.CQRS.AspNetCore.Local;
@@ -41,8 +40,10 @@ CancellationToken cancellationToken
     {
         var metadata = objectSource.MetadataFor(obj.GetType());
 
-        using var activity = LeanCodeActivitySource.ActivitySource.StartActivity("pipeline.action.local");
-        activity?.AddTag("object", metadata.ObjectType.FullName);
+        using var activity = LeanCodeActivitySource.StartExecution("Local", metadata.ObjectType.FullName);
+        activity?.AddTag("object.kind", metadata.ObjectKind.ToString());
+        activity?.AddTag("object.type", metadata.ObjectType.FullName);
+        activity?.AddTag("object.handler", metadata.HandlerType.FullName);
 
         await using var scope = serviceProvider.CreateAsyncScope();
 

src/CQRS/LeanCode.CQRS.AspNetCore/Middleware/CQRSExceptionTranslationMiddleware.cs

@@ -2,6 +2,7 @@
 using LeanCode.Contracts;
 using LeanCode.Contracts.Validation;
 using LeanCode.CQRS.Execution;
+using LeanCode.OpenTelemetry;
 using Microsoft.AspNetCore.Http;
 using Serilog;
 
@@ -35,6 +36,9 @@ public async Task InvokeAsync(HttpContext httpContext)
         }
         catch (CommandExecutionInvalidException ex)
         {
+            using var activity = LeanCodeActivitySource.StartMiddleware("ExceptionTranslation");
+            activity?.SetTag("error.code", ex.ErrorCode);
+
             var result = WrapInCommandResult(ex);
             logger.Warning("Command {@Command} is not valid with result {@Result}", cqrsPayload.Payload, result);
             var executionResult = ExecutionResult.WithPayload(result, StatusCodes.Status422UnprocessableEntity);

src/CQRS/LeanCode.CQRS.AspNetCore/Middleware/CQRSSecurityMiddleware.cs

@@ -1,6 +1,7 @@
 using LeanCode.Contracts.Security;
 using LeanCode.CQRS.Execution;
 using LeanCode.CQRS.Security;
+using LeanCode.OpenTelemetry;
 using Microsoft.AspNetCore.Http;
 using Serilog;
 
@@ -41,12 +42,12 @@ public async Task InvokeAsync(HttpContext context)
         foreach (var customAuthorizerDefinition in customAuthorizers)
         {
             var authorizerType = customAuthorizerDefinition.Authorizer;
-            var customAuthorizer = context.RequestServices.GetService(authorizerType) as IHttpContextCustomAuthorizer;
+            using var activity = LeanCodeActivitySource.StartMiddleware("Security", authorizerType.FullName);
+            activity?.SetTag("authorizer.type", authorizerType.FullName);
 
-            if (customAuthorizer is null)
-            {
-                throw new CustomAuthorizerNotFoundException(authorizerType);
-            }
+            var customAuthorizer =
+                context.RequestServices.GetService(authorizerType) as IHttpContextCustomAuthorizer
+                ?? throw new CustomAuthorizerNotFoundException(authorizerType);
 
             var authorized = await customAuthorizer.CheckIfAuthorizedAsync(
                 context,
@@ -56,6 +57,7 @@ public async Task InvokeAsync(HttpContext context)
 
             if (!authorized)
             {
+                activity?.SetTag("authorizer.authorized", false);
                 logger.Warning(
                     "User is not authorized for {@Object}, authorizer {AuthorizerType} did not pass",
                     payload.Payload,
@@ -66,6 +68,10 @@ public async Task InvokeAsync(HttpContext context)
                 metrics.CQRSFailure(CQRSMetrics.AuthorizationFailure);
                 return;
             }
+            else
+            {
+                activity?.SetTag("authorizer.authorized", true);
+            }
         }
 
         await next(context);

src/CQRS/LeanCode.CQRS.AspNetCore/Middleware/CQRSTracingMiddleware.cs

@@ -18,8 +18,9 @@ public async Task InvokeAsync(HttpContext httpContext)
     {
         var cqrsMetadata = httpContext.GetCQRSObjectMetadata();
 
-        using var activity = LeanCodeActivitySource.Start(
-            $"{cqrsMetadata.ObjectKind}Handler {cqrsMetadata.HandlerType.FullName}"
+        using var activity = LeanCodeActivitySource.StartExecution(
+            cqrsMetadata.ObjectKind.ToString(),
+            cqrsMetadata.HandlerType.FullName
         );
         activity?.AddTag("object.kind", cqrsMetadata.ObjectKind.ToString());
         activity?.AddTag("object.type", cqrsMetadata.ObjectType.FullName);

src/CQRS/LeanCode.CQRS.AspNetCore/Middleware/CQRSValidationMiddleware.cs

@@ -1,6 +1,7 @@
 using LeanCode.Contracts;
 using LeanCode.CQRS.Execution;
 using LeanCode.CQRS.Validation;
+using LeanCode.OpenTelemetry;
 using Microsoft.AspNetCore.Http;
 
 namespace LeanCode.CQRS.AspNetCore.Middleware;
@@ -32,7 +33,11 @@ public async Task InvokeAsync(HttpContext httpContext, ICommandValidatorResolver
 
         if (validator is not null)
         {
+            using var activity = LeanCodeActivitySource.StartMiddleware("Validation");
+            activity?.SetTag("validation.validator", validator.GetType().FullName);
+
             var result = await validator.ValidateAsync(httpContext, (ICommand)payload.Payload);
+            activity?.SetTag("validation.valid", result.IsValid);
 
             if (!result.IsValid)
             {

src/Infrastructure/LeanCode.OpenTelemetry/ActivitySource.cs

@@ -7,4 +7,11 @@ public static class LeanCodeActivitySource
     public static readonly ActivitySource ActivitySource = new("LeanCode.CoreLibrary");
 
     public static Activity? Start(string name) => ActivitySource.StartActivity(name);
+
+    public static Activity? StartExecution(string type, string? objectType) => Start($"{type} - {objectType}");
+
+    public static Activity? StartMiddleware(string middlewareName) => Start($"middleware - {middlewareName}");
+
+    public static Activity? StartMiddleware(string middlewareName, string? objectName) =>
+        Start($"middleware - {middlewareName} - {objectName}");
 }