fix: 修改模型方法问题 #7626

TencentBlueKing · Dec 13, 2024 · 79fb051 · 79fb051
1 parent 53ebf3a
commit 79fb051
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/gcloud/contrib/template_market/models.py b/gcloud/contrib/template_market/models.py
@@ -16,6 +16,8 @@
 
 from gcloud import err_code
 
+TEMPLATE_SHARED_RECORD_BATCH_OPERATION_COUNT = 50
+
 
 class TemplateSharedManager(models.Manager):
     def update_shared_record(
@@ -66,9 +68,12 @@ def update_shared_record(
                 new_records.append(new_record)
 
         if new_records:
-            TemplateSharedRecord.objects.bulk_create(new_records)
+            TemplateSharedRecord.objects.bulk_create(batch_size=TEMPLATE_SHARED_RECORD_BATCH_OPERATION_COUNT)
+
         if records_to_update:
-            TemplateSharedRecord.objects.bulk_update(records_to_update, ["extra_info"])
+            TemplateSharedRecord.objects.bulk_update(
+                records_to_update, ["extra_info"], batch_size=TEMPLATE_SHARED_RECORD_BATCH_OPERATION_COUNT
+            )
 
         return {"result": True, "message": "update shared record successfully", "code": err_code.SUCCESS.code}