batch index to have single batch mode for source/collections of an or…

…g/user
OpenConceptLab · Nov 17, 2023 · e61d65d · rkorytkowski · Dec 8, 2023 · snyaggarwal
1 parent 3c5d1af
commit e61d65d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/core/common/models.py b/core/common/models.py
@@ -177,10 +177,13 @@ def get_exact_or_criteria(attr, values):
         return criteria
 
     @staticmethod
-    def batch_index(queryset, document):
+    def batch_index(queryset, document, single_batch=False):
         if not get(settings, 'TEST_MODE'):
-            for batch in queryset.iterator(chunk_size=500):
-                document().update(batch, parallel=True)
+            if single_batch:
 instance.batch_index(instance.source_set, SourceDocument, True) 
 instance.batch_index(instance.source_set, SourceDocument, True) 
+                document().update(queryset, parallel=True)
+            else:
+                for batch in queryset.iterator(chunk_size=500):
+                    document().update(batch, parallel=True)
 
     @staticmethod
     @transaction.atomic

diff --git a/core/common/signals.py b/core/common/signals.py
@@ -25,7 +25,7 @@ def propagate_owner_status(sender, instance=None, created=False, **kwargs):  # p
 
         if updated_sources:
             from core.sources.documents import SourceDocument
-            instance.batch_index(instance.source_set, SourceDocument)
+            instance.batch_index(instance.source_set, SourceDocument, True)
         if updated_collections:
             from core.collections.documents import CollectionDocument
-            instance.batch_index(instance.collection_set, CollectionDocument)
+            instance.batch_index(instance.collection_set, CollectionDocument, True)