Merge pull request rails#49458 from eugeneius/broadcast_delegate_block

Delegate block in broadcast logger method_missing

Author: eugeneius

activesupport/lib/active_support/broadcast_logger.rb

@@ -203,16 +203,20 @@ def dispatch(&block)
         @broadcasts.each { |logger| block.call(logger) }
       end
 
-      def method_missing(name, *args)
+      def method_missing(name, *args, &block)
         loggers = @broadcasts.select { |logger| logger.respond_to?(name) }
 
         if loggers.none?
-          super(name, *args)
+          super(name, *args, &block)
         elsif loggers.one?
-          loggers.first.send(name, *args)
+          loggers.first.send(name, *args, &block)
         else
-          loggers.map { |logger| logger.send(name, *args) }
+          loggers.map { |logger| logger.send(name, *args, &block) }
         end
       end
+
+      def respond_to_missing?(method, include_all)
+        @broadcasts.any? { |logger| logger.respond_to?(method, include_all) }
+      end
   end
 end

activesupport/test/broadcast_logger_test.rb

@@ -268,6 +268,16 @@ def info(msg, &block)
       assert(logger.foo)
     end
 
+    test "calling a method that accepts a block" do
+      logger = BroadcastLogger.new(CustomLogger.new)
+
+      called = false
+      logger.bar do
+        called = true
+      end
+      assert(called)
+    end
+
     class CustomLogger
       attr_reader :adds, :closed, :chevrons
       attr_accessor :level, :progname, :formatter, :local_level
@@ -286,6 +296,10 @@ def foo
         true
       end
 
+      def bar
+        yield
+      end
+
       def debug(message, &block)
         add(::Logger::DEBUG, message, &block)
       end