update correlation_id if cid module installed

Fixes #119

Author: Drew Shafer

django_declarative_apis/machinery/tasks.py

@@ -26,8 +26,10 @@
     import cid.locals
 
     _get_correlation_id = cid.locals.get_cid
+    _set_correlation_id = cid.locals.set_cid
 except ImportError:
     _get_correlation_id = lambda: None  # noqa: E731
+    _set_correlation_id = lambda _: None  # noqa: E731
 
 JOB_COUNT_CACHE_KEY = "future_task_runner:job_id"
 QUEUE_LENGTH_CACHE_KEY = "future_task_runner:current_queue_length"
@@ -151,6 +153,9 @@ def future_task_runner(
     resource_instance = resource_class.objects.get(pk=resource_instance_id)
     endpoint_task = getattr(endpoint_class, endpoint_method_name)
 
+    if correlation_id:
+        _set_correlation_id(correlation_id)
+
     _log_task_stats(
         endpoint_method_name,
         resource_instance_id,
@@ -291,6 +296,9 @@ def resource_task_runner(
     resource_instance = resource_class.objects.get(pk=resource_instance_id)
     resource_method = getattr(resource_instance, resource_method_name)
 
+    if correlation_id:
+        _set_correlation_id(correlation_id)
+
     _log_task_stats(
         "{0}.{1}".format(resource_class_name, resource_method_name),
         resource_instance_id,