lib/kafka/connection.rb in ruby-kafka-0.7.5 vs lib/kafka/connection.rb in ruby-kafka-0.7.6.beta1

@@ -50,11 +50,11 @@
     #   broker. Default is 10 seconds.
     #
     # @return [Connection] a new connection.
     def initialize(host:, port:, client_id:, logger:, instrumenter:, connect_timeout: nil, socket_timeout: nil, ssl_context: nil)
       @host, @port, @client_id = host, port, client_id
-      @logger = logger
+      @logger = TaggedLogger.new(logger)
       @instrumenter = instrumenter
 
       @connect_timeout = connect_timeout || CONNECT_TIMEOUT
       @socket_timeout = socket_timeout || SOCKET_TIMEOUT
       @ssl_context = ssl_context
@@ -91,10 +91,11 @@
         response_size: 0,
       }
 
       raise IdleConnection if idle?
 
+      @logger.push_tags(api_name)
       @instrumenter.instrument("request.connection", notification) do
         open unless open?
 
         @correlation_id += 1
 
@@ -111,9 +112,11 @@
       end
     rescue SystemCallError, EOFError, IOError => e
       close
 
       raise ConnectionError, "Connection error #{e.class}: #{e}"
+    ensure
+      @logger.pop_tags
     end
 
     private
 
     def open