update DbContextInterceptorSamples.cs

samples/WeihanLi.EntityFramework.Sample/DbContextInterceptorSamples.cs

@@ -0,0 +1,75 @@
+using Microsoft.EntityFrameworkCore;
+using Microsoft.EntityFrameworkCore.Diagnostics;
+using Microsoft.Extensions.DependencyInjection;
+using WeihanLi.Extensions;
+
+namespace WeihanLi.EntityFramework.Sample;
+
+public static class DbContextInterceptorSamples
+{
+    public static async Task RunAsync()
+    {
+        await InterceptorTest2();
+    }
+
+    private static async Task InterceptorTest1()
+    {
+        var services = new ServiceCollection();
+        services.AddScoped<SavingInterceptor>();
+        services.AddDbContext<FileTestDbContext>((provider, options) =>
+        {
+            options.AddInterceptors(provider.GetRequiredService<SavingInterceptor>());
+            options.UseInMemoryDatabase("test");
+        });
+        await using var provider = services.BuildServiceProvider();
+        using var scope = provider.CreateScope();
+        var dbContext = scope.ServiceProvider.GetRequiredService<FileTestDbContext>();
+        await dbContext.Database.EnsureCreatedAsync();
+        dbContext.Entities.Add(new TestEntity { Id = 1, Name = "1" });
+        await dbContext.SaveChangesAsync();
+    }
+
+    private static async Task InterceptorTest2()
+    {
+        var services = new ServiceCollection();
+        services.AddDbContext<FileTestDbContext>(options =>
+        {
+            options.UseInMemoryDatabase("test");
+        });
+        services.AddDbContextInterceptor<FileTestDbContext, SavingInterceptor>();
+        await using var provider = services.BuildServiceProvider();
+        using var scope = provider.CreateScope();
+        var dbContext = scope.ServiceProvider.GetRequiredService<FileTestDbContext>();
+        await dbContext.Database.EnsureCreatedAsync();
+        dbContext.Entities.Add(new TestEntity { Id = 2, Name = "1" });
+        await dbContext.SaveChangesAsync();
+    }
+}
+
+file sealed class FileTestDbContext(DbContextOptions<FileTestDbContext> options) : DbContext(options)
+{
+    public DbSet<TestEntity> Entities { get; set; }
+}
+
+file sealed class TestEntity
+{
+    public int Id { get; set; }
+    public string Name { get; set; }
+}
+
+file sealed class SavingInterceptor : SaveChangesInterceptor
+{
+    public override ValueTask<InterceptionResult<int>> SavingChangesAsync(DbContextEventData eventData, InterceptionResult<int> result,
+        CancellationToken cancellationToken = default)
+    {
+        Console.WriteLine("SavingChangesAsync");
+        return base.SavingChangesAsync(eventData, result, cancellationToken);
+    }
+
+    public override ValueTask<int> SavedChangesAsync(SaveChangesCompletedEventData eventData, int result,
+        CancellationToken cancellationToken = default)
+    {
+        Console.WriteLine("SavedChangesAsync");
+        return base.SavedChangesAsync(eventData, result, cancellationToken);
+    }
+}

samples/WeihanLi.EntityFramework.Sample/Program.cs

@@ -18,11 +18,13 @@ namespace WeihanLi.EntityFramework.Sample;
 
 public static class Program
 {
-    public static void Main(string[] args)
+    public static async Task Main(string[] args)
     {
         // SoftDeleteTest();
         // RepositoryTest();
-        AutoAuditTest();
+        // AutoAuditTest();
+
+        await DbContextInterceptorSamples.RunAsync();
 
         Console.WriteLine("completed");
         Console.ReadLine();
@@ -116,7 +118,7 @@ private static void AutoAuditTest()
                     // .AddInterceptors(ActivatorUtilities.GetServiceOrCreateInstance<AuditInterceptor>(provider))
                     ;
             });
-            services.AddDbContextInterceptor<TestDbContext, AuditInterceptor>();
+            services.AddDbContextInterceptor1<TestDbContext, AuditInterceptor>();
 
             services.AddEFAutoAudit(builder =>
             {
@@ -435,7 +437,7 @@ private static void SoftDeleteTest()
         context.Database.EnsureDeleted();
     }
 
-    private static IServiceCollection AddDbContextInterceptor<TContext, TInterceptor>(
+    private static IServiceCollection AddDbContextInterceptor1<TContext, TInterceptor>(
         this IServiceCollection services,
         ServiceLifetime optionsLifetime = ServiceLifetime.Scoped
         )