From 7833eadee22bb5fea797fb8a0d779547f2e8af68 Mon Sep 17 00:00:00 2001
From: ysmoradi <ysmoradi@outlook.com>
Date: Sat, 8 Mar 2025 10:35:01 +0100
Subject: [PATCH] Fix bit Besql issue with lock during in-app migration
 (#10204)

---
 src/Besql/Bit.Besql/IServiceCollectionBesqlExtentions.cs | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/Besql/Bit.Besql/IServiceCollectionBesqlExtentions.cs b/src/Besql/Bit.Besql/IServiceCollectionBesqlExtentions.cs
index ded8f5e149..195b27a22e 100644
--- a/src/Besql/Bit.Besql/IServiceCollectionBesqlExtentions.cs
+++ b/src/Besql/Bit.Besql/IServiceCollectionBesqlExtentions.cs
@@ -52,7 +52,13 @@ public static IServiceCollection AddBesqlDbContextFactory<TDbContext>(this IServ
         else
         {
             services.TryAddSingleton<IBitBesqlStorage, BitBesqlNoopStoage>();
-            services.AddDbContextFactory<TDbContext, PooledDbContextFactoryBase<TDbContext>>(optionsAction);
+            services.AddDbContextFactory<TDbContext, PooledDbContextFactoryBase<TDbContext>>((serviceProvider, options) =>
+            {
+#if NET9_0_OR_GREATER
+                options.ReplaceService<IHistoryRepository, BesqlHistoryRepository>();
+#endif
+                optionsAction.Invoke(serviceProvider, options);
+            });
         }
 
         return services;