v1.0.15

framework/src/BBT.Aether.Core/BBT/Aether/DependencyInjection/ServiceScopeFactoryExtensions.cs

@@ -0,0 +1,160 @@
+using System;
+using System.Threading;
+using System.Threading.Tasks;
+using BBT.Aether.Uow;
+using Microsoft.Extensions.DependencyInjection;
+
+namespace BBT.Aether.DependencyInjection;
+
+/// <summary>
+/// Extension methods for IServiceScopeFactory.
+/// </summary>
+public static class ServiceScopeFactoryExtensions
+{
+    /// <summary>
+    /// Executes the given action within a new dependency injection scope and a new unit of work.
+    /// Manages ambient service provider propagation.
+    /// </summary>
+    /// <param name="scopeFactory">The service scope factory.</param>
+    /// <param name="action">The action to execute, receiving the scoped service provider.</param>
+    /// <param name="options">Unit of work options. Defaults to a new UoW with default settings.</param>
+    /// <param name="cancellationToken">Cancellation token.</param>
+    /// <returns>A task representing the asynchronous operation.</returns>
+    public static async Task ExecuteInNewUnitOfWorkScopeAsync(
+        this IServiceScopeFactory scopeFactory,
+        Func<IServiceProvider, Task> action,
+        UnitOfWorkOptions? options = null,
+        CancellationToken cancellationToken = default)
+    {
+        await using var scope = scopeFactory.CreateAsyncScope();
+        var sp = scope.ServiceProvider;
+
+        // Propagate ambient service provider for the new scope
+        var previousAmbient = AmbientServiceProvider.Current;
+        AmbientServiceProvider.Current = sp;
+
+        try
+        {
+            var uowManager = sp.GetRequiredService<IUnitOfWorkManager>();
+
+            // Begin UnitOfWork
+            await using var uow = await uowManager.BeginAsync(options, cancellationToken);
+
+            // Execute action
+            await action(sp);
+
+            // Commit UnitOfWork
+            await uow.CommitAsync(cancellationToken);
+        }
+        finally
+        {
+            // Restore previous ambient context
+            AmbientServiceProvider.Current = previousAmbient;
+        }
+    }
+
+    /// <summary>
+    /// Executes the given function within a new dependency injection scope and a new unit of work, returning a result.
+    /// Manages ambient service provider propagation.
+    /// </summary>
+    /// <typeparam name="T">The type of the result.</typeparam>
+    /// <param name="scopeFactory">The service scope factory.</param>
+    /// <param name="func">The function to execute, receiving the scoped service provider.</param>
+    /// <param name="options">Unit of work options. Defaults to a new UoW with default settings.</param>
+    /// <param name="cancellationToken">Cancellation token.</param>
+    /// <returns>The result of the function execution.</returns>
+    public static async Task<T> ExecuteInNewUnitOfWorkScopeAsync<T>(
+        this IServiceScopeFactory scopeFactory,
+        Func<IServiceProvider, Task<T>> func,
+        UnitOfWorkOptions? options = null,
+        CancellationToken cancellationToken = default)
+    {
+        await using var scope = scopeFactory.CreateAsyncScope();
+        var sp = scope.ServiceProvider;
+
+        // Propagate ambient service provider for the new scope
+        var previousAmbient = AmbientServiceProvider.Current;
+        AmbientServiceProvider.Current = sp;
+
+        try
+        {
+            var uowManager = sp.GetRequiredService<IUnitOfWorkManager>();
+
+            // Begin UnitOfWork
+            await using var uow = await uowManager.BeginAsync(options, cancellationToken);
+
+            // Execute function
+            var result = await func(sp);
+
+            // Commit UnitOfWork
+            await uow.CommitAsync(cancellationToken);
+
+            return result;
+        }
+        finally
+        {
+            // Restore previous ambient context
+            AmbientServiceProvider.Current = previousAmbient;
+        }
+    }
+
+    /// <summary>
+    /// Executes the given action within a new dependency injection scope without starting a unit of work.
+    /// Manages ambient service provider propagation.
+    /// </summary>
+    /// <param name="scopeFactory">The service scope factory.</param>
+    /// <param name="action">The action to execute, receiving the scoped service provider.</param>
+    /// <returns>A task representing the asynchronous operation.</returns>
+    public static async Task ExecuteInNewScopeAsync(
+        this IServiceScopeFactory scopeFactory,
+        Func<IServiceProvider, Task> action)
+    {
+        await using var scope = scopeFactory.CreateAsyncScope();
+        var sp = scope.ServiceProvider;
+
+        // Propagate ambient service provider for the new scope
+        var previousAmbient = AmbientServiceProvider.Current;
+        AmbientServiceProvider.Current = sp;
+
+        try
+        {
+            await action(sp);
+        }
+        finally
+        {
+            // Restore previous ambient context
+            AmbientServiceProvider.Current = previousAmbient;
+        }
+    }
+
+    /// <summary>
+    /// Executes the given function within a new dependency injection scope without starting a unit of work, returning a result.
+    /// Manages ambient service provider propagation.
+    /// </summary>
+    /// <typeparam name="T">The type of the result.</typeparam>
+    /// <param name="scopeFactory">The service scope factory.</param>
+    /// <param name="func">The function to execute, receiving the scoped service provider.</param>
+    /// <returns>The result of the function execution.</returns>
+    public static async Task<T> ExecuteInNewScopeAsync<T>(
+        this IServiceScopeFactory scopeFactory,
+        Func<IServiceProvider, Task<T>> func)
+    {
+        await using var scope = scopeFactory.CreateAsyncScope();
+        var sp = scope.ServiceProvider;
+
+        // Propagate ambient service provider for the new scope
+        var previousAmbient = AmbientServiceProvider.Current;
+        AmbientServiceProvider.Current = sp;
+
+        try
+        {
+            return await func(sp);
+        }
+        finally
+        {
+            // Restore previous ambient context
+            AmbientServiceProvider.Current = previousAmbient;
+        }
+    }
+}
+}

framework/src/BBT.Aether.Core/BBT/Aether/Results/ResultExtensions.cs

@@ -231,6 +231,28 @@ public static Result<T> Tap<T>(this Result<T> result, Action<T> sideEffect)
     public static Result<T> Ensure<T>(this Result<T> result, Func<T, bool> predicate, Error error)
         => result.IsSuccess && !predicate(result.Value!) ? Result<T>.Fail(error) : result;
 
+    /// <summary>
+    /// Ensures a predicate is satisfied, otherwise returns an error (lazy evaluation).
+    /// </summary>
+    /// <typeparam name="T">Value type</typeparam>
+    /// <param name="result">The source result</param>
+    /// <param name="predicate">Condition that must be true</param>
+    /// <param name="errorFactory">Function to create error if predicate fails</param>
+    /// <returns>The original result if predicate passes, otherwise a failure</returns>
+    public static Result<T> Ensure<T>(this Result<T> result, Func<T, bool> predicate, Func<Error> errorFactory)
+        => result.IsSuccess && !predicate(result.Value!) ? Result<T>.Fail(errorFactory()) : result;
+
+    /// <summary>
+    /// Ensures a predicate is satisfied, otherwise returns an error (lazy evaluation with value access).
+    /// </summary>
+    /// <typeparam name="T">Value type</typeparam>
+    /// <param name="result">The source result</param>
+    /// <param name="predicate">Condition that must be true</param>
+    /// <param name="errorFactory">Function to create error based on the value if predicate fails</param>
+    /// <returns>The original result if predicate passes, otherwise a failure</returns>
+    public static Result<T> Ensure<T>(this Result<T> result, Func<T, bool> predicate, Func<T, Error> errorFactory)
+        => result.IsSuccess && !predicate(result.Value!) ? Result<T>.Fail(errorFactory(result.Value!)) : result;
+
     /// <summary>
     /// Executes a side effect on failure without altering the error.
     /// Useful for logging, alerting, or cleanup actions on failure.
@@ -349,6 +371,34 @@ public static async Task<Result<T>> EnsureAsync<T>(this Result<T> result, Func<T
         return await predicate(result.Value!) ? result : Result<T>.Fail(error);
     }
 
+    /// <summary>
+    /// Async version of Ensure for sync Result with async predicate (lazy error evaluation).
+    /// </summary>
+    /// <typeparam name="T">Value type</typeparam>
+    /// <param name="result">The source result</param>
+    /// <param name="predicate">Async predicate that must be true</param>
+    /// <param name="errorFactory">Function to create error if predicate fails</param>
+    /// <returns>The original result if predicate passes, otherwise a failure</returns>
+    public static async Task<Result<T>> EnsureAsync<T>(this Result<T> result, Func<T, Task<bool>> predicate, Func<Error> errorFactory)
+    {
+        if (!result.IsSuccess) return result;
+        return await predicate(result.Value!) ? result : Result<T>.Fail(errorFactory());
+    }
+
+    /// <summary>
+    /// Async version of Ensure for sync Result with async predicate (lazy error evaluation with value access).
+    /// </summary>
+    /// <typeparam name="T">Value type</typeparam>
+    /// <param name="result">The source result</param>
+    /// <param name="predicate">Async predicate that must be true</param>
+    /// <param name="errorFactory">Function to create error based on the value if predicate fails</param>
+    /// <returns>The original result if predicate passes, otherwise a failure</returns>
+    public static async Task<Result<T>> EnsureAsync<T>(this Result<T> result, Func<T, Task<bool>> predicate, Func<T, Error> errorFactory)
+    {
+        if (!result.IsSuccess) return result;
+        return await predicate(result.Value!) ? result : Result<T>.Fail(errorFactory(result.Value!));
+    }
+
     /// <summary>
     /// Async version of Map for Task-wrapped results.
     /// </summary>
@@ -458,6 +508,34 @@ public static async Task<Result<T>> EnsureAsync<T>(this Task<Result<T>> task, Fu
         return result.Ensure(predicate, error);
     }
 
+    /// <summary>
+    /// Async version of Ensure for Task-wrapped results (lazy error evaluation).
+    /// </summary>
+    /// <typeparam name="T">Value type</typeparam>
+    /// <param name="task">The source result task</param>
+    /// <param name="predicate">Condition that must be true</param>
+    /// <param name="errorFactory">Function to create error if predicate fails</param>
+    /// <returns>The original result if predicate passes, otherwise a failure</returns>
+    public static async Task<Result<T>> EnsureAsync<T>(this Task<Result<T>> task, Func<T, bool> predicate, Func<Error> errorFactory)
+    {
+        var result = await task;
+        return result.Ensure(predicate, errorFactory);
+    }
+
+    /// <summary>
+    /// Async version of Ensure for Task-wrapped results (lazy error evaluation with value access).
+    /// </summary>
+    /// <typeparam name="T">Value type</typeparam>
+    /// <param name="task">The source result task</param>
+    /// <param name="predicate">Condition that must be true</param>
+    /// <param name="errorFactory">Function to create error based on the value if predicate fails</param>
+    /// <returns>The original result if predicate passes, otherwise a failure</returns>
+    public static async Task<Result<T>> EnsureAsync<T>(this Task<Result<T>> task, Func<T, bool> predicate, Func<T, Error> errorFactory)
+    {
+        var result = await task;
+        return result.Ensure(predicate, errorFactory);
+    }
+
     /// <summary>
     /// Async version of Match for Task-wrapped results with sync handlers.
     /// </summary>