snapshot: preserve reliability hardening and Workdock identity pass

backend/workflows/middleware.py

@@ -1,9 +1,34 @@
+import uuid
+
 from django.shortcuts import render
 
 from .branding import is_trial_expired, is_trial_mode_enabled
+from .logging_utils import clear_request_id, set_request_id
 from .roles import ROLE_PLATFORM_OWNER, get_user_role_key
 
 
+class RequestIDMiddleware:
+    HEADER_NAME = 'X-Request-ID'
+
+    def __init__(self, get_response):
+        self.get_response = get_response
+
+    def __call__(self, request):
+        request_id = (
+            request.META.get('HTTP_X_REQUEST_ID')
+            or request.META.get('HTTP_X_CORRELATION_ID')
+            or uuid.uuid4().hex
+        )
+        request.request_id = request_id
+        set_request_id(request_id)
+        try:
+            response = self.get_response(request)
+        finally:
+            clear_request_id()
+        response[self.HEADER_NAME] = request_id
+        return response
+
+
 class TrialModeMiddleware:
     EXEMPT_PREFIXES = (
         '/healthz/',