lib/lamby/rack.rb in lamby-0.3.1 vs lib/lamby/rack.rb in lamby-0.3.2

@@ -3,11 +3,11 @@
 
     include SamHelpers
 
     LAMBDA_EVENT = 'lambda.event'.freeze
     LAMBDA_CONTEXT = 'lambda.context'.freeze
-    HTTP_X_APIGWSTAGE = 'HTTP_X_APIGWSTAGE'.freeze
+    HTTP_X_REQUESTID = 'HTTP_X_REQUEST_ID'.freeze
 
     attr_reader :event, :context
 
     def initialize(event, context)
       @event = event
@@ -46,10 +46,12 @@
     end
 
     def env_headers
       headers.transform_keys do |key|
         "HTTP_#{key.to_s.upcase.tr '-', '_'}"
+      end.tap do |hdrs|
+        hdrs[HTTP_X_REQUESTID] = request_id
       end
     end
 
     def content_type
       headers.delete('Content-Type') || headers.delete('content-type')
@@ -80,9 +82,13 @@
       event['isBase64Encoded']
     end
 
     def server_protocol
       event.dig('requestContext', 'protocol') || 'HTTP/1.1'
+    end
+
+    def request_id
+      context.aws_request_id
     end
 
   end
 end