Stream context now gets set

shinyorg · Jan 23, 2025 · d587a8c · d587a8c
1 parent 9edad71
commit d587a8c
Show file tree

Hide file tree

Showing 9 changed files with 41 additions and 19 deletions.
diff --git a/samples/Sample.Api/Handlers/TheStreamHandler.cs b/samples/Sample.Api/Handlers/TheStreamHandler.cs
@@ -9,7 +9,11 @@ public record TestStreamRequest(int SecondsBetween) : IStreamRequest<string>;
 [MediatorHttpPost("GetStream", "/stream")]
 public class TheStreamHandler(ILogger<TheStreamHandler> logger) : IStreamRequestHandler<TestStreamRequest, string>
 {
-    public async IAsyncEnumerable<string> Handle(TestStreamRequest request, [EnumeratorCancellation] CancellationToken cancellationToken)
+    public async IAsyncEnumerable<string> Handle(
+        TestStreamRequest request, 
+        RequestContext<TestStreamRequest> context, 
+        [EnumeratorCancellation] CancellationToken cancellationToken
+    )
     {
         while (!cancellationToken.IsCancellationRequested)
         {

diff --git a/samples/Sample/Handlers/AutoRefreshStreamRequestHandler.cs b/samples/Sample/Handlers/AutoRefreshStreamRequestHandler.cs
@@ -7,7 +7,11 @@ namespace Sample.Handlers;
 public class AutoRefreshStreamRequestHandler : IStreamRequestHandler<AutoRefreshRequest, string>
 {
     [TimerRefresh(3000)]
-    public async IAsyncEnumerable<string> Handle(AutoRefreshRequest request, [EnumeratorCancellation] CancellationToken cancellationToken)
+    public async IAsyncEnumerable<string> Handle(
+        AutoRefreshRequest request,
+        RequestContext<AutoRefreshRequest> context,
+        [EnumeratorCancellation] CancellationToken cancellationToken
+    )
     {
         yield return DateTimeOffset.Now.ToString("h:mm:ss tt");
     }

diff --git a/samples/Sample/Handlers/MyRequestMiddleware.cs b/samples/Sample/Handlers/MyRequestMiddleware.cs
@@ -4,7 +4,6 @@
 namespace Sample.Handlers;
 
 
-[SingletonMiddleware]
 public class MyRequestMiddleware(AppSqliteConnection conn) : IRequestMiddleware<MyMessageRequest, MyMessageResponse>
 {
     public async Task<MyMessageResponse> Process(

diff --git a/samples/Sample/Handlers/TickerStreamRequestHandler.cs b/samples/Sample/Handlers/TickerStreamRequestHandler.cs
@@ -7,7 +7,11 @@ namespace Sample.Handlers;
 [SingletonHandler]
 public class TickerStreamRequestHandler : IStreamRequestHandler<TickerRequest, string>
 {
-    public async IAsyncEnumerable<string> Handle(TickerRequest request, [EnumeratorCancellation] CancellationToken cancellationToken)
+    public async IAsyncEnumerable<string> Handle(
+        TickerRequest request, 
+        RequestContext<TickerRequest> context,
+        [EnumeratorCancellation] CancellationToken cancellationToken
+    )
     {
         for (var i = 0; i < request.Repeat; i++)
         {

diff --git a/samples/Sample/MauiProgram.cs b/samples/Sample/MauiProgram.cs
@@ -2,6 +2,7 @@
 using System.Runtime.CompilerServices;
 using Microsoft.Extensions.Configuration;
 using Polly;
+using Sample.Handlers;
 
 namespace Sample;
 
@@ -39,21 +40,17 @@ public static MauiApp CreateMauiApp()
         builder.Logging.AddDebug();
         builder.Services.AddBlazorWebViewDeveloperTools();
 #endif
-        builder.Configuration.AddJsonStream(Assembly.GetExecutingAssembly().GetManifestResourceStream("Sample.appsettings.json")!);
+        builder.Configuration.AddJsonStream(
+            Assembly
+                .GetExecutingAssembly()
+                .GetManifestResourceStream("Sample.appsettings.json")!
+        );
 
         builder.Services.AddShinyMediator(x => x
             .UseMaui()
             .UseBlazor()
-
-            .AddOfflineAvailabilityMiddleware()
             .AddPersistentCache()
-            .AddUserErrorNotificationsMiddleware()
-            .AddPerformanceLoggingMiddleware()
-            .AddTimerRefreshStreamMiddleware()
-            .AddEventExceptionHandlingMiddleware()
             .AddDataAnnotations()
-
-            .AddHttpClient()
             .AddMauiHttpDecorator()
             .AddPrismSupport()
 
@@ -64,6 +61,7 @@ public static MauiApp CreateMauiApp()
                 y.ExpirationScanFrequency = TimeSpan.FromSeconds(5);
             })
         );
+        builder.Services.AddSingletonAsImplementedInterfaces<MyRequestMiddleware>();
         builder.Services.AddDiscoveredMediatorHandlersFromSample();
 
         builder.Services.AddSingleton<AppSqliteConnection>();

diff --git a/src/Shiny.Mediator.AppSupport/Middleware/ReplayStreamMiddleware.cs b/src/Shiny.Mediator.AppSupport/Middleware/ReplayStreamMiddleware.cs
@@ -31,7 +31,8 @@ CancellationToken cancellationToken
 
         logger.LogDebug("ReplayStream Enabled - {Request}", context.Request);
         return this.Iterate(
-            context.Request, 
+            context.Request,
+            context,
             next, 
             cancellationToken
         );
@@ -55,9 +56,9 @@ protected bool IsEnabled(TRequest request, IRequestHandler requestHandler)
     }
 
 
-    // TODO: add cache
     protected virtual async IAsyncEnumerable<TResult> Iterate(
         TRequest request, 
+        RequestContext<TRequest> context,
         StreamRequestHandlerDelegate<TResult> next, 
         [EnumeratorCancellation] CancellationToken ct
     )
@@ -68,13 +69,21 @@ [EnumeratorCancellation] CancellationToken ct
         {
             var item = await cache.Get<TResult>(requestKey).ConfigureAwait(false);
             if (item != null)
+            {
+                // TODO: need cache date
+                logger.LogDebug("ReplayStream Cache Hit - {Request}", context.Request);
+                context.Cache(new CacheContext(requestKey, true, DateTimeOffset.UtcNow));
                 yield return item.Value;
+            }
         }
         else if (offline != null)
         {
             var store = await offline.Get<TResult>(request);
-            if (store != null) // TODO: I need context here to ship out date
+            if (store != null)
+            {
+                context.Offline(new OfflineAvailableContext(requestKey, store.Timestamp));
                 yield return store.Value;
+            }
         }
 
         if (!internet.IsAvailable)

diff --git a/src/Shiny.Mediator/Caching/Infrastructure/CachingRequestMiddleware.cs b/src/Shiny.Mediator/Caching/Infrastructure/CachingRequestMiddleware.cs
@@ -46,7 +46,7 @@ CancellationToken cancellationToken
             logger.LogDebug("Cache Forced Refresh - {Request}", context.Request);
             result = await next().ConfigureAwait(false);
             if (result != null)
-                cacheService.Set(cacheKey, result, config);
+                await cacheService.Set(cacheKey, result, config).ConfigureAwait(false);
         }
         else
         {

diff --git a/src/Shiny.Mediator/IStreamRequestHandler.cs b/src/Shiny.Mediator/IStreamRequestHandler.cs
@@ -3,5 +3,9 @@ namespace Shiny.Mediator;
 
 public interface IStreamRequestHandler<TRequest, TResult> : IRequestHandler where TRequest : IStreamRequest<TResult>
 {
-    IAsyncEnumerable<TResult> Handle(TRequest request, CancellationToken cancellationToken);
+    IAsyncEnumerable<TResult> Handle(
+        TRequest request, 
+        RequestContext<TRequest> context, 
+        CancellationToken cancellationToken
+    );
 }
diff --git a/src/Shiny.Mediator/Infrastructure/Impl/Mediator_Streams.cs b/src/Shiny.Mediator/Infrastructure/Impl/Mediator_Streams.cs
@@ -51,7 +51,7 @@ public RequestResult<IAsyncEnumerable<TResult>> Handle()
                 "Executing streaming request handler {RequestHandlerType}",
                 requestHandler.GetType().FullName
             );
-            return requestHandler.Handle(request, cancellationToken);
+            return requestHandler.Handle(request, context, cancellationToken);
         });
 
         var middlewares = scope.GetServices<IStreamRequestMiddleware<TRequest, TResult>>();