# HG changeset patch
# User Martin Bornhold <martin@rhodecode.com>
# Date 2016-07-14 07:33:12
# Node ID af6adc7d1ca920ed8aab350b4cb9c9bcac6f0c31
# Parent  47f28cf8698e819e9cd59717d504dca3a0d63213

caches: Add an argument to make the cache context thread scoped.

diff --git a/rhodecode/lib/caches.py b/rhodecode/lib/caches.py
--- a/rhodecode/lib/caches.py
+++ b/rhodecode/lib/caches.py
@@ -21,6 +21,7 @@
 
 import beaker
 import logging
+import threading
 
 from beaker.cache import _cache_decorate, cache_regions, region_invalidate
 
@@ -170,7 +171,7 @@ class InvalidationContext(object):
             safe_str(self.repo_name), safe_str(self.cache_type))
 
     def __init__(self, compute_func, repo_name, cache_type,
-                 raise_exception=False):
+                 raise_exception=False, thread_scoped=False):
         self.compute_func = compute_func
         self.repo_name = repo_name
         self.cache_type = cache_type
@@ -178,6 +179,13 @@ class InvalidationContext(object):
             repo_name, cache_type)
         self.raise_exception = raise_exception
 
+        # Append the thread id to the cache key if this invalidation context
+        # should be scoped to the current thread.
+        if thread_scoped:
+            thread_id = threading.current_thread().ident
+            self.cache_key = '{cache_key}_{thread_id}'.format(
+                cache_key=self.cache_key, thread_id=thread_id)
+
     def get_cache_obj(self):
         cache_key = CacheKey.get_cache_key(
             self.repo_name, self.cache_type)