Class: ActFluentLoggerRails::FluentLogger

Inherits:
ActiveSupport::Logger
  • Object
show all
Defined in:
lib/act-fluent-logger-rails/logger.rb

Instance Method Summary collapse

Constructor Details

#initialize(options, level, log_tags) ⇒ FluentLogger

Returns a new instance of FluentLogger.



80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
# File 'lib/act-fluent-logger-rails/logger.rb', line 80

def initialize(options, level, log_tags)
  self.level = level
  port    = options[:port]
  host    = options[:host]
  nanosecond_precision = options[:nanosecond_precision]
  @messages_type = (options[:messages_type] || :array).to_sym
  @tag = options[:tag]
  @severity_key = (options[:severity_key] || :severity).to_sym
  @flush_immediately = options[:flush_immediately]
  logger_opts = {host: host, port: port, nanosecond_precision: nanosecond_precision}
  logger_opts[:tls_options] = options[:tls_options] unless options[:tls_options].nil?
  @fluent_logger = ::Fluent::Logger::FluentLogger.new(nil, logger_opts)
  @severity = 0
  @log_tags = log_tags
  after_initialize if respond_to?(:after_initialize) && Rails::VERSION::MAJOR < 6
end

Instance Method Details

#[](key) ⇒ Object



128
129
130
# File 'lib/act-fluent-logger-rails/logger.rb', line 128

def [](key)
  map[key]
end

#[]=(key, value) ⇒ Object



132
133
134
# File 'lib/act-fluent-logger-rails/logger.rb', line 132

def []=(key, value)
  map[key] = value
end

#add(severity, message = nil, progname = nil, &block) ⇒ Object



97
98
99
100
101
102
103
# File 'lib/act-fluent-logger-rails/logger.rb', line 97

def add(severity, message = nil, progname = nil, &block)
  return true if severity < level
  message = (block_given? ? block.call : progname) if message.blank?
  return true if message.blank?
  add_message(severity, message)
  true
end

#add_message(severity, message) ⇒ Object



105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
# File 'lib/act-fluent-logger-rails/logger.rb', line 105

def add_message(severity, message)
  @severity = severity if @severity < severity

  message =
    case message
    when ::String
      message
    when ::Exception
      "#{ message.message } (#{ message.class })\n" <<
        (message.backtrace || []).join("\n")
    else
      message.inspect
    end

  if message.encoding == Encoding::UTF_8
    logger_messages << message
  else
    logger_messages << message.dup.force_encoding(Encoding::UTF_8)
  end

  flush if @flush_immediately
end

#add_tagsObject



154
155
156
157
158
159
# File 'lib/act-fluent-logger-rails/logger.rb', line 154

def add_tags
  return unless @tags_thread_key && Thread.current.key?(@tags_thread_key)
  @log_tags.keys.zip(Thread.current[@tags_thread_key]).each do |k, v|
    map[k] = v
  end
end

#closeObject



171
172
173
# File 'lib/act-fluent-logger-rails/logger.rb', line 171

def close
  @fluent_logger.close
end

#flushObject



136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
# File 'lib/act-fluent-logger-rails/logger.rb', line 136

def flush
  return if logger_messages.empty?
  messages = if @messages_type == :string
               logger_messages.join("\n")
             else
               logger_messages
             end
  map[:messages] = messages
  map[@severity_key] = format_severity(@severity)
  add_tags

  @fluent_logger.post(@tag, map)
  @severity = 0
  logger_messages.clear
  Thread.current[@tags_thread_key] = nil if @tags_thread_key
  map.clear
end

#format_severity(severity) ⇒ Object



183
184
185
# File 'lib/act-fluent-logger-rails/logger.rb', line 183

def format_severity(severity)
  ActFluentLoggerRails::Logger::SEV_LABEL[severity] || 'ANY'
end

#levelObject



175
176
177
# File 'lib/act-fluent-logger-rails/logger.rb', line 175

def level
  @level
end

#level=(l) ⇒ Object



179
180
181
# File 'lib/act-fluent-logger-rails/logger.rb', line 179

def level=(l)
  @level = l
end

#logger_messagesObject



161
162
163
164
# File 'lib/act-fluent-logger-rails/logger.rb', line 161

def logger_messages
  @messages_thread_key ||= "fluentd_logger_messages:#{object_id}".freeze
  Thread.current[@messages_thread_key] ||= []
end

#mapObject



166
167
168
169
# File 'lib/act-fluent-logger-rails/logger.rb', line 166

def map
  @map_thread_key ||= "fluentd_logger_map:#{object_id}".freeze
  Thread.current[@map_thread_key] ||= {}
end