lib/lamby/handler.rb in lamby-0.6.0 vs lib/lamby/handler.rb in lamby-1.0.0

@@ -18,11 +18,11 @@
     end
 
     def response
       { statusCode: status,
         headers: headers,
-        body: body }
+        body: body }.merge(rack_response)
     end
 
     def status
       @status
     end
@@ -51,9 +51,13 @@
       when :api
         Lamby::RackApi.new @event, @context
       else
         Lamby::RackAlb.new @event, @context
       end
+    end
+
+    def rack_response
+      rack.response(self)
     end
 
     def call_app
       if Debug.on?(@event)
         Debug.call @event, @context, rack.env