diff --git a/inventory_provider/routes/common.py b/inventory_provider/routes/common.py index 6dda30525fb971c0c46987d62e17846710ef985e..c74e5dfea1e2c0ff926db8764004f2b132727e91 100644 --- a/inventory_provider/routes/common.py +++ b/inventory_provider/routes/common.py @@ -137,7 +137,8 @@ def after_request(response): return response -def _redis_client_proc(key_queue, value_queue, config_params, doc_type): +def _redis_client_proc( + key_queue, value_queue, config_params, doc_type, use_next_redis=False): """ create a local redis connection with the current db index, lookup the values of the keys that come from key_queue @@ -165,7 +166,10 @@ def _redis_client_proc(key_queue, value_queue, config_params, doc_type): return etree.XML(value) try: - r = tasks_common.get_current_redis(config_params) + if use_next_redis: + r = tasks_common.get_next_redis(config_params) + else: + r = tasks_common.get_current_redis(config_params) while True: key = key_queue.get() @@ -215,7 +219,7 @@ def _load_redis_docs( q = queue.Queue() t = threading.Thread( target=_redis_client_proc, - args=[q, response_queue, config_params, doc_type]) + args=[q, response_queue, config_params, doc_type, use_next_redis]) t.start() threads.append({'thread': t, 'queue': q})