try..except around redis_conn uses

Author: shin-

docker_registry/images.py

@@ -289,8 +289,7 @@ def put_image_checksum(image_id):
     # Checksum is ok, we remove the marker
     store.remove(mark_path)
     # We trigger a task on the diff worker if it's running
-    if cache.redis_conn:
-        layers.diff_queue.push(image_id)
+    layers.enqueue_diff(image_id)
     return toolkit.response()
 
 

docker_registry/lib/layers.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 
+import logging
 import tarfile
 import tempfile
 
@@ -28,10 +29,21 @@
     tarfile.GNUTYPE_SPARSE: 'S',
 }
 
+logger = logging.getLogger(__name__)
+
 # queue for requesting diff calculations from workers
 diff_queue = rqueue.CappedCollection(cache.redis_conn, "diff-worker", 1024)
 
 
+
+def enqueue_diff(image_id):
+    try:
+        if cache.redis_conn:
+            layers.diff_queue.push(image_id)
+    except cache.redis.exceptions.ConnectionError:
+        logger.warning('Diff queue: Redis connection error')
+
+
 def generate_ancestry(image_id, parent_id=None):
     if not parent_id:
         store.put_content(store.image_ancestry_path(image_id),

docker_registry/lib/mirroring.py

@@ -105,9 +105,14 @@ def wrapper(namespace, repository, *args, **kwargs):
             # client GETs a single tag
             tag_path = store.tag_path(namespace, repository, kwargs['tag'])
 
-        data = cache.redis_conn.get('{0}:{1}'.format(
-            cache.cache_prefix, tag_path
-        ))
+        try:
+            data = cache.redis_conn.get('{0}:{1}'.format(
+                cache.cache_prefix, tag_path
+            ))
+        except cache.redis.exceptions.ConnectionError:
+            data = None
+            logger.warning('Tags cache: Redis connection error')
+
         if data is not None:
             return toolkit.response(data=data, raw=True)
         source_resp = lookup_source(
@@ -117,9 +122,13 @@ def wrapper(namespace, repository, *args, **kwargs):
             return resp
         data = source_resp.content
         headers = _response_headers(source_resp.headers)
-        cache.redis_conn.setex('{0}:{1}'.format(
-            cache.cache_prefix, tag_path
-        ), tags_cache_ttl, data)
+        try:
+            cache.redis_conn.setex('{0}:{1}'.format(
+                cache.cache_prefix, tag_path
+            ), tags_cache_ttl, data)
+        except cache.redis.exceptions.ConnectionError:
+            logger.warning('Tags cache: Redis connection error')
+
         return toolkit.response(data=data, headers=headers,
                                 raw=True)
     return wrapper