Adds scheduling support to Mocha

src/All.slnx

@@ -352,6 +352,7 @@
     <Project Path="Mocha/src/Mocha.Mediator.Abstractions/Mocha.Mediator.Abstractions.csproj" />
     <Project Path="Mocha/src/Mocha.Mediator/Mocha.Mediator.csproj" />
     <Project Path="Mocha/src/Mocha.Outbox/Mocha.Outbox.csproj" />
+    <Project Path="Mocha/src/Mocha.Scheduling/Mocha.Scheduling.csproj" />
     <Project Path="Mocha/src/Mocha.Threading/Mocha.Threading.csproj" />
     <Project Path="Mocha/src/Mocha.Transport.InMemory/Mocha.Transport.InMemory.csproj" />
     <Project Path="Mocha/src/Mocha.Transport.Postgres/Mocha.Transport.Postgres.csproj" />

src/Mocha/Mocha.slnx

@@ -18,6 +18,7 @@
     <Project Path="src/Mocha.Mediator.Abstractions/Mocha.Mediator.Abstractions.csproj" />
     <Project Path="src/Mocha.Mediator/Mocha.Mediator.csproj" />
     <Project Path="src/Mocha.Outbox/Mocha.Outbox.csproj" />
+    <Project Path="src/Mocha.Scheduling/Mocha.Scheduling.csproj" />
     <Project Path="src/Mocha.Threading/Mocha.Threading.csproj" />
     <Project Path="src/Mocha.Mediator/Mocha.Mediator.csproj" />
     <Project Path="src/Mocha.Analyzers/Mocha.Analyzers.csproj" />

src/Mocha/benchmarks/Mocha.Mediator.Benchmarks/Internal/PoolingBenchmarks.cs

@@ -73,7 +73,7 @@ public void Cleanup()
         obj.Message = s_message;
         obj.MessageType = s_type;
         var result = obj.Message;
-        // No return — GC collects it
+        // No return - GC collects it
         return result;
     }
 

src/Mocha/src/Demo/Demo.Billing/Data/BillingDbContext.cs

@@ -3,6 +3,7 @@
 using Microsoft.EntityFrameworkCore.Design;
 using Mocha.Inbox;
 using Mocha.Outbox;
+using Mocha.Scheduling;
 
 namespace Demo.Billing.Data;
 
@@ -20,6 +21,7 @@ protected override void OnModelCreating(ModelBuilder modelBuilder)
 
         modelBuilder.AddPostgresOutbox();
         modelBuilder.AddPostgresInbox();
+        modelBuilder.AddPostgresScheduledMessages();
 
         modelBuilder.Entity<Invoice>(entity =>
         {

src/Mocha/src/Demo/Demo.Billing/Handlers/OrderPlacedEventHandler.cs

@@ -70,5 +70,21 @@ await messageBus.PublishAsync(
             cancellationToken);
 
         logger.LogInformation("PaymentCompletedEvent published for order {OrderId}", message.OrderId);
+
+        // Schedule a payment reminder 30 seconds from now
+        await messageBus.SchedulePublishAsync(
+            new PaymentReminderEvent
+            {
+                InvoiceId = invoice.Id,
+                OrderId = message.OrderId,
+                CustomerId = message.CustomerId,
+                Amount = invoice.Amount
+            },
+            DateTimeOffset.UtcNow.AddSeconds(30),
+            cancellationToken);
+
+        logger.LogInformation(
+            "Payment reminder scheduled for order {OrderId} in 30 seconds",
+            message.OrderId);
     }
 }

src/Mocha/src/Demo/Demo.Billing/Handlers/PaymentReminderEventHandler.cs

@@ -0,0 +1,40 @@
+using Demo.Billing.Data;
+using Demo.Billing.Entities;
+using Demo.Contracts.Events;
+using Microsoft.EntityFrameworkCore;
+using Mocha;
+
+namespace Demo.Billing.Handlers;
+
+public class PaymentReminderEventHandler(
+    BillingDbContext db,
+    ILogger<PaymentReminderEventHandler> logger) : IEventHandler<PaymentReminderEvent>
+{
+    public async ValueTask HandleAsync(PaymentReminderEvent message, CancellationToken cancellationToken)
+    {
+        var invoice = await db.Invoices.FirstOrDefaultAsync(i => i.Id == message.InvoiceId, cancellationToken);
+
+        if (invoice is null)
+        {
+            logger.LogWarning("Payment reminder for invoice {InvoiceId} - invoice not found", message.InvoiceId);
+            return;
+        }
+
+        if (invoice.Status == InvoiceStatus.Pending)
+        {
+            logger.LogWarning(
+                "Payment reminder: Invoice {InvoiceId} for order {OrderId} is still pending. Amount: {Amount}",
+                message.InvoiceId,
+                message.OrderId,
+                message.Amount);
+        }
+        else
+        {
+            logger.LogInformation(
+                "Payment reminder: Invoice {InvoiceId} for order {OrderId} is already {Status}",
+                message.InvoiceId,
+                message.OrderId,
+                invoice.Status);
+        }
+    }
+}