lib/phobos/batch_handler.rb in phobos-1.9.0.pre.beta1 vs lib/phobos/batch_handler.rb in phobos-1.9.0.pre.beta2

@@ -4,19 +4,15 @@
   module BatchHandler
     def self.included(base)
       base.extend(ClassMethods)
     end
 
-    def before_consume_batch(payloads, _metadata)
-      payloads
-    end
-
     def consume_batch(_payloads, _metadata)
       raise NotImplementedError
     end
 
-    def around_consume_batch(_payloads, _metadata)
-      yield
+    def around_consume_batch(payloads, metadata)
+      yield payloads, metadata
     end
 
     module ClassMethods
       def start(kafka_client); end