aboutsummaryrefslogtreecommitdiffstats
path: root/actioncable/lib/action_cable/connection/tagged_logger_proxy.rb
blob: e5319087fb098e815dea520273c3ef8b2bd1ed5c (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
module ActionCable
  module Connection
    # Allows the use of per-connection tags against the server logger. This wouldn't work using the tradional
    # ActiveSupport::TaggedLogging-enhanced Rails.logger, as that logger will reset the tags between requests.
    # The connection is long-lived, so it needs its own set of tags for its independent duration.
    class TaggedLoggerProxy
      attr_reader :tags

      def initialize(logger, tags:)
        @logger = logger
        @tags = tags.flatten
      end

      def add_tags(*tags)
        @tags += tags.flatten
        @tags = @tags.uniq
      end

      def tag(logger)
        if logger.respond_to?(:tagged)
          current_tags = tags - logger.formatter.current_tags
          logger.tagged(*current_tags) { yield }
        else
          yield
        end
      end

      %i( debug info warn error fatal unknown ).each do |severity|
        define_method(severity) do |message|
          log severity, message
        end
      end

      protected
        def log(type, message)
          tag(@logger) { @logger.send type, message }
        end
    end
  end
end