# HG changeset patch
# User Marcin Kuzminski <marcin@rhodecode.com>
# Date 2018-07-12 08:40:18
# Node ID bb23dd31cce4120d172f932e818b1864a6bfc88c
# Parent  5151caf6780086785103de5a0bc1ddf6f6b5250d

caches: use a gevent compatible file-lock mechanism.

- because of gevent we can run into a deadlock situation and worker freeze because of
each of the workers will try to aquire a lock.
- This should help with the issue and gevent.

diff --git a/rhodecode/lib/rc_cache/backends.py b/rhodecode/lib/rc_cache/backends.py
--- a/rhodecode/lib/rc_cache/backends.py
+++ b/rhodecode/lib/rc_cache/backends.py
@@ -17,16 +17,24 @@
 # This program is dual-licensed. If you wish to learn more about the
 # RhodeCode Enterprise Edition, including its added features, Support services,
 # and proprietary license terms, please see https://rhodecode.com/licenses/
+import time
+import errno
+import logging
+
+import gevent
 
 from dogpile.cache.backends import memory as memory_backend
 from dogpile.cache.backends import file as file_backend
 from dogpile.cache.backends import redis as redis_backend
-from dogpile.cache.backends.file import NO_VALUE, compat
+from dogpile.cache.backends.file import NO_VALUE, compat, FileLock
+from dogpile.cache.util import memoized_property
 
 from rhodecode.lib.memory_lru_debug import LRUDict
 
 _default_max_size = 1024
 
+log = logging.getLogger(__name__)
+
 
 class LRUMemoryBackend(memory_backend.MemoryBackend):
 
@@ -44,9 +52,45 @@ class Serializer(object):
         return compat.pickle.loads(value)
 
 
+class CustomLockFactory(FileLock):
+
+    @memoized_property
+    def _module(self):
+        import fcntl
+        flock_org = fcntl.flock
+
+        def gevent_flock(fd, operation):
+            """
+            Gevent compatible flock
+            """
+            # set non-blocking, this will cause an exception if we cannot acquire a lock
+            operation |= fcntl.LOCK_NB
+            start_lock_time = time.time()
+            timeout = 60 * 5  # 5min
+            while True:
+                try:
+                    flock_org(fd, operation)
+                    # lock has been acquired
+                    break
+                except (OSError, IOError) as e:
+                    # raise on other errors than Resource temporarily unavailable
+                    if e.errno != errno.EAGAIN:
+                        raise
+                    elif (time.time() - start_lock_time) > timeout:
+                        # waited to much time on a lock, better fail than loop for ever
+                        raise
+
+                    log.debug('Failed to acquire lock, retry in 0.1')
+                    gevent.sleep(0.1)
+
+        fcntl.flock = gevent_flock
+        return fcntl
+
+
 class FileNamespaceBackend(Serializer, file_backend.DBMBackend):
 
     def __init__(self, arguments):
+        arguments['lock_factory'] = CustomLockFactory
         super(FileNamespaceBackend, self).__init__(arguments)
 
     def list_keys(self, prefix=''):