class Fluent::NorikraPlugin::Target

Attributes

auto_field[RW]
escaped_name[R]
fields[RW]
name[RW]
queries[RW]
time_key[RW]

Public Class Methods

escape(src) click to toggle source
# File lib/fluent/plugin/norikra/target.rb, line 6
def self.escape(src)
  if src.nil? || src.empty?
    return 'FluentdGenerated'
  end

  dst = src.gsub(/[^_a-zA-Z0-9]/, '_')
  unless dst =~ /^[a-zA-Z]([_a-zA-Z0-9]*[a-zA-Z0-9])?$/
    unless dst =~ /^[a-zA-Z]/
      dst = 'Fluentd' + dst
    end
    unless dst =~ /[a-zA-Z0-9]$/
      dst = dst + 'Generated'
    end
  end
  dst
end
new(target, config) click to toggle source
# File lib/fluent/plugin/norikra/target.rb, line 23
def initialize(target, config)
  @name = target
  @escaped_name = self.class.escape(@name)
  @auto_field = config.auto_field.nil? ? true : config.auto_field
  @time_key = config.time_key
  @escape_fieldname = config.escape_fieldname

  @filter = RecordFilter.new(*([:include, :include_regexp, :exclude, :exclude_regexp].map{|s| config.filter_params[s]}))
  @fields = config.field_definitions
  if @time_key
    @fields[:integer].push @time_key
  end
  @queries = config.query_generators.map{|g| g.generate(@name, @escaped_name)}
end

Public Instance Methods

escape_recursive(record) click to toggle source
# File lib/fluent/plugin/norikra/target.rb, line 51
def escape_recursive(record)
  return record unless record.is_a?(Hash) || record.is_a?(Array)
  return record.map{|v| escape_recursive(v) } if record.is_a?(Array)

  # Hash
  r = {}
  record.keys.each do |key|
    k = if key =~ /[^$_a-zA-Z0-9]/
          key.gsub(/[^$_a-zA-Z0-9]/, '_')
        else
          key
        end
    v = escape_recursive(record[key])
    r[k] = v
  end
  r
end
filter(time, record) click to toggle source
# File lib/fluent/plugin/norikra/target.rb, line 38
def filter(time, record)
  r = @filter.filter(record)
  if @time_key
    # Fluentd time (sec) -> Norikra timestamp (milliseconds)
    r = r.merge({ @time_key => time * 1000 })
  end
  if @escape_fieldname
    escape_recursive(r)
  else
    r
  end
end
reserve_fields() click to toggle source
# File lib/fluent/plugin/norikra/target.rb, line 69
def reserve_fields
  f = {}
  @fields.keys.each do |type_sym|
    @fields[type_sym].each do |fieldname|
      f[fieldname] = type_sym.to_s
    end
  end
  f
end