lib/lamby/handler.rb in lamby-4.0.2 vs lib/lamby/handler.rb in lamby-4.1.0

@@ -86,15 +86,17 @@
         Debug.call @event, @context, rack.env
       elsif rack?
         @status, @headers, @body = @app.call rack.env 
         set_cookies
         rack_response
+      elsif lambdakiq?
+        Lambdakiq.cmd event: @event, context: @context
+      elsif lambda_cable?
+        LambdaCable.cmd event: @event, context: @context
       elsif runner?
         @status, @headers, @body = Runner.call(@event)
         { statusCode: status, headers: headers, body: body }
-      elsif lambdakiq?
-        Lambdakiq.handler(@event)
       elsif event_bridge?
         Lamby.config.event_bridge_handler.call @event, @context
       else
         [404, {}, StringIO.new('')]
       end
@@ -118,8 +120,12 @@
       defined?(::Lambdakiq) && ::Lambdakiq.jobs?(@event)
     end
 
     def runner?
       Runner.handle?(@event)
+    end
+
+    def lambda_cable?
+      defined?(::LambdaCable) && ::LambdaCable.handle?(@event, @context)
     end
   end
 end