Try again in cache race condition update_or_create

Author: davegaeddert

bolt-cache/bolt/cache/core.py

@@ -69,10 +69,12 @@ def set(self, value, expiration: datetime | timedelta | int | float | None = Non
                     key=self.key, defaults=defaults
                 )
         except IntegrityError:
-            # Most likely a race condition in creating the item
-            # so we'll effectively do a get and return the stored value
-            self.reload()
-            return self.value
+            # Most likely a race condition in creating the item,
+            # so trying again should do an update
+            with transaction.atomic():
+                item, _ = self._model_class.objects.update_or_create(
+                    key=self.key, defaults=defaults
+                )
 
         self.reload()
         return item.value