Skip to content

Instantly share code, notes, and snippets.

@daviddavis
Created November 9, 2016 15:08
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save daviddavis/58c1d81e3b2ee259667fb26d2eab631c to your computer and use it in GitHub Desktop.
Save daviddavis/58c1d81e3b2ee259667fb26d2eab631c to your computer and use it in GitHub Desktop.
diff --git a/server/pulp/server/db/connection.py b/server/pulp/server/db/connection.py
index b419560..ce4462f 100644
--- a/server/pulp/server/db/connection.py
+++ b/server/pulp/server/db/connection.py
@@ -210,7 +210,7 @@ class UnsafeRetry(object):
'find', 'find_one', 'count', 'create_index', 'ensure_index',
'drop_index', 'drop_indexes', 'reindex', 'index_information', 'options',
'group', 'rename', 'distinct', 'map_reduce', 'inline_map_reduce',
- 'find_and_modify')
+ 'find_and_modify', 'first', 'delete')
@classmethod
def decorate_instance(cls, instance, full_name):
diff --git a/server/pulp/server/db/querysets.py b/server/pulp/server/db/querysets.py
index dcc7fce..fb7f370 100644
--- a/server/pulp/server/db/querysets.py
+++ b/server/pulp/server/db/querysets.py
@@ -6,6 +6,7 @@ from mongoengine.queryset import DoesNotExist, QuerySetNoCache
from pymongo import ASCENDING
from pulp.server import exceptions as pulp_exceptions
+from pulp.server.db.connection import UnsafeRetry
class QuerySetPreventCache(QuerySetNoCache):
@@ -18,6 +19,10 @@ class QuerySetPreventCache(QuerySetNoCache):
customized behavior for that QuerySet.
"""
+ def __init__(self, document, collection):
+ super(QuerySetPreventCache, self).__init__(document, collection)
+ UnsafeRetry.decorate_instance(instance=self, full_name=type(document))
+
def cache(self):
"""
Currently, there is no reason to have a caching version of any of our
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment