feat: skip route key for AddOrUpdateManyAsync

AElfProject · Sep 3, 2024 · 25bd865 · 25bd865
1 parent 2935bf8
commit 25bd865
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/AElf.EntityMapping.Elasticsearch/Repositories/ElasticsearchRepository.cs b/src/AElf.EntityMapping.Elasticsearch/Repositories/ElasticsearchRepository.cs
@@ -150,7 +150,7 @@ await client.IndexAsync(model, ss => ss.Index(indexName).Refresh(_elasticsearchO
     }
 
     public async Task AddOrUpdateManyAsync(List<TEntity> list, string collectionName = null,
-        CancellationToken cancellationToken = default)
+        CancellationToken cancellationToken = default, bool skipRouteKey = false)
     {
         string entityName = typeof(TEntity).Name;
         _logger.LogDebug("[{1}]Before AddOrUpdateManyAsync time: {0} ",
@@ -161,7 +161,7 @@ public async Task AddOrUpdateManyAsync(List<TEntity> list, string collectionName
         var isSharding = _shardingKeyProvider.IsShardingCollection();
 
         var client = await GetElasticsearchClientAsync(cancellationToken);
-        if (!isSharding)
+        if (!isSharding || skipRouteKey)
         {
             await BulkAddAsync(client, indexNames, list, isSharding, cancellationToken);
             return;

diff --git a/src/AElf.EntityMapping/Repositories/IEntityMappingBasicRepository.cs b/src/AElf.EntityMapping/Repositories/IEntityMappingBasicRepository.cs
@@ -8,7 +8,7 @@ public interface IEntityMappingBasicRepository<TEntity, TKey> where TEntity : cl
 
         Task AddOrUpdateAsync(TEntity model, string collectionName = null, CancellationToken cancellationToken = default);
 
-        Task AddOrUpdateManyAsync(List<TEntity> list, string collectionName = null, CancellationToken cancellationToken = default);
+        Task AddOrUpdateManyAsync(List<TEntity> list, string collectionName = null, CancellationToken cancellationToken = default, bool skipRouteKey = false);
 
         Task UpdateAsync(TEntity model, string collectionName = null, CancellationToken cancellationToken = default);