feat: base 优化safe_get_or_create以处理并发

apps/utils/models.py

@@ -112,15 +112,21 @@ class BaseModel(models.Model):
         abstract = True
 
     @classmethod
-    def safe_get_or_create(cls, **kwargs):
+    def safe_get_or_create(cls, defaults=None, **kwargs):
-        defaults = kwargs.pop('defaults', {})
+        defaults = defaults or {}
-        for attempt in range(5):
+        
+        with transaction.atomic():
             try:
-                return cls.objects.get_or_create(defaults=defaults, **kwargs)
+                ins = cls.objects.select_for_update().get(**kwargs)
-            except IntegrityError:
+                return ins, False
-                if attempt < 4:
+            except cls.DoesNotExist:
-                    time.sleep(0.1)
+                try:
-                else:
+                    ins = cls.objects.create(**defaults, **kwargs)
+                except IntegrityError:
+                    try:
+                        return cls.objects.get(**kwargs), False
+                    except cls.DoesNotExist:
+                        pass
                     raise
         
     def handle_parent(self):