github/reddit
1
1
Fork 0
mirror of https://github.com/reddit-archive/reddit.git synced 2026-01-30 17:28:03 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fix memoization of X_ids and reverse_X_ids.

Browse Source
This commit is contained in:
Logan Hanks
2013-02-20 17:39:05 -08:00
parent e22d31be91
commit 81acc136f4
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
r2/r2/lib/db/userrel.py
View File

@@ -105,21 +105,21 @@ class MemoizedUserRelManager(UserRelManager):
self.add = self._update_caches_on_success(sup.add)
self.remove = self._update_caches_on_success(sup.remove)
def _update_caches(self, thing):
def _update_caches(self, thing, user):
if not self.disable_ids_fn:
self.ids(thing, _update=True)
if not self.disable_reverse_ids_fn:
self.reverse_ids(thing, _update=True)
self.reverse_ids(user, _update=True)
def _update_caches_on_success(self, method):
@functools.wraps(method)
def wrapper(thing, *args, **kwargs):
def wrapper(thing, user, *args, **kwargs):
try:
result = method(thing, *args, **kwargs)
result = method(thing, user, *args, **kwargs)
except:
raise
else:
self._update_caches(thing)
self._update_caches(thing, user)
return result
return wrapper