module Fluent::PluginHelper::Inject

Public Class Methods

included(mod) click to toggle source
# File lib/fluent/plugin_helper/inject.rb, line 80
def self.included(mod)
  mod.include InjectParams
end
new() click to toggle source
Calls superclass method
# File lib/fluent/plugin_helper/inject.rb, line 84
def initialize
  super
  @_inject_enabled = false
  @_inject_hostname_key = nil
  @_inject_hostname = nil
  @_inject_tag_key = nil
  @_inject_time_key = nil
  @_inject_time_formatter = nil
end

Public Instance Methods

configure(conf) click to toggle source
Calls superclass method
# File lib/fluent/plugin_helper/inject.rb, line 94
def configure(conf)
  super

  if @inject_config
    @_inject_hostname_key = @inject_config.hostname_key
    if @_inject_hostname_key
      @_inject_hostname =  @inject_config.hostname
      unless @_inject_hostname
        @_inject_hostname = Socket.gethostname
        log.info "using hostname for specified field", host_key: @_inject_hostname_key, host_name: @_inject_hostname
      end
    end
    @_inject_tag_key = @inject_config.tag_key
    @_inject_time_key = @inject_config.time_key
    if @_inject_time_key
      @_inject_time_formatter = case @inject_config.time_type
                                when :float then ->(time){ time.to_r.truncate(+6).to_f } # microsecond floating point value
                                when :unixtime then ->(time){ time.to_i }
                                else
                                  localtime = @inject_config.localtime && !@inject_config.utc
                                  Fluent::TimeFormatter.new(@inject_config.time_format, localtime, @inject_config.timezone)
                                end
    end

    @_inject_enabled = @_inject_hostname_key || @_inject_tag_key || @_inject_time_key
  end
end
inject_values_to_event_stream(tag, es) click to toggle source
# File lib/fluent/plugin_helper/inject.rb, line 42
def inject_values_to_event_stream(tag, es)
  return es unless @_inject_enabled

  new_es = Fluent::MultiEventStream.new
  es.each do |time, record|
    r = record.dup
    if @_inject_hostname_key
      r[@_inject_hostname_key] = @_inject_hostname
    end
    if @_inject_tag_key
      r[@_inject_tag_key] = tag
    end
    if @_inject_time_key
      r[@_inject_time_key] = @_inject_time_formatter.call(time)
    end
    new_es.add(time, r)
  end

  new_es
end
inject_values_to_record(tag, time, record) click to toggle source
# File lib/fluent/plugin_helper/inject.rb, line 25
def inject_values_to_record(tag, time, record)
  return record unless @_inject_enabled

  r = record.dup
  if @_inject_hostname_key
    r[@_inject_hostname_key] = @_inject_hostname
  end
  if @_inject_tag_key
    r[@_inject_tag_key] = tag
  end
  if @_inject_time_key
    r[@_inject_time_key] = @_inject_time_formatter.call(time)
  end

  r
end