Added unsafe mode for shadow primary keys

BUTR · Nov 10, 2023 · 0e91d95 · 0e91d95
1 parent f022d28
commit 0e91d95
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/src/BUTR.Site.NexusMods.Server/Extensions/DbSetExtensions.cs b/src/BUTR.Site.NexusMods.Server/Extensions/DbSetExtensions.cs
@@ -45,17 +45,17 @@ public static async Task SynchronizeOnSaveAsync<TEntity>(this DbSet<TEntity> dbS
     }
 
 
-    public static async Task UpsertAsync<TEntity>(this DbSet<TEntity> dbSet, IEnumerable<TEntity> entities) where TEntity : class, IEntity
+    public static async Task UpsertAsync<TEntity>(this DbSet<TEntity> dbSet, IEnumerable<TEntity> entities, bool @unsafe = false) where TEntity : class, IEntity
     {
-        await dbSet.BulkMergeAsync(entities);
+        await dbSet.BulkMergeAsync(entities, o => o.UnsafeMode = @unsafe);
     }
-    public static async Task UpsertAsync<TEntity>(this DbSet<TEntity> dbSet, IAsyncEnumerable<TEntity> entities) where TEntity : class, IEntity
+    public static async Task UpsertAsync<TEntity>(this DbSet<TEntity> dbSet, IAsyncEnumerable<TEntity> entities, bool @unsafe = false) where TEntity : class, IEntity
     {
-        await dbSet.BulkMergeAsync(await entities.ToArrayAsync());
+        await dbSet.BulkMergeAsync(await entities.ToArrayAsync(), o => o.UnsafeMode = @unsafe);
     }
-    public static async Task UpsertAsync<TEntity>(this DbSet<TEntity> dbSet, params TEntity[] entities) where TEntity : class, IEntity
+    public static async Task UpsertAsync<TEntity>(this DbSet<TEntity> dbSet, bool @unsafe = false, params TEntity[] entities) where TEntity : class, IEntity
     {
-        await dbSet.BulkMergeAsync(entities);
+        await dbSet.BulkMergeAsync(entities, o => o.UnsafeMode = @unsafe);
     }
 
     public static async Task SynchronizeAsync<TEntity>(this DbSet<TEntity> dbSet, IEnumerable<TEntity> entities) where TEntity : class, IEntity

diff --git a/src/BUTR.Site.NexusMods.Server/Jobs/AutocompleteProcessorProcessorJob.cs b/src/BUTR.Site.NexusMods.Server/Jobs/AutocompleteProcessorProcessorJob.cs
@@ -95,7 +95,7 @@ private static async Task HandleTenantAsync(TenantId tenant, IServiceProvider se
                 TenantId = tenant,
                 Type = key,
                 Value = x,
-            }));
+            }), true);
         }
 
         /*