Contents

class Fluent::RewriteTagFilterOutput < Fluent::Output
  Fluent::Plugin.register_output('rewrite_tag_filter', self)

  PATTERN_MAX_NUM = 200

  config_param :rewriterule1, :string # string: NAME REGEXP
  (2..PATTERN_MAX_NUM).each do |i|
    config_param ('rewriterule' + i.to_s).to_sym, :string, :default => nil # NAME REGEXP
  end

  def configure(conf)
    super

    @rewriterules = []
    rewriterule_names = []

    invalids = conf.keys.select{|k| k =~ /^rewriterule(\d+)$/}.select{|arg| arg =~ /^rewriterule(\d+)/ and not (1..PATTERN_MAX_NUM).include?($1.to_i)}
    if invalids.size > 0
      $log.warn "invalid number rewriterules (valid rewriterule number:1-{PATTERN_MAX_NUM}):" + invalids.join(",")
    end
    (1..PATTERN_MAX_NUM).each do |i|
      next unless conf["rewriterule#{i}"]
      name,regexp,tag = conf["rewriterule#{i}"].split(' ', 3)
      @rewriterules.push([i, name, Regexp.new(regexp), tag])
      rewriterule_names.push(name + regexp)
    end
    rewriterule_index_list = conf.keys.select{|s| s =~ /^rewriterule\d$/}.map{|v| (/^rewriterule(\d)$/.match(v))[1].to_i}
    unless rewriterule_index_list.reduce(true){|v,i| v and @rewriterules[i - 1]}
      raise Fluent::ConfigError, "jump of rewriterule index found" + @rewriterules.inspect
    end
    unless @rewriterules.length == rewriterule_names.uniq.length
      raise Fluent::ConfigError, "duplicated rewriterules found" + @rewriterules.inspect
    end
  end

  def emit(tag, es, chain)
    es.each do |time,record|
      rewrite = false
      @rewriterules.each do |index, rewritekey, regexp, rewritetag|
        rewritevalue = record[rewritekey]
        next if rewritevalue.nil?
        if (regexp && regexp.match(rewritevalue))
          rewrite = true
          tag = rewritetag
          break
        end
      end
      Fluent::Engine.emit(tag, time, record) if (rewrite)
    end

    chain.next
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
fluent-plugin-rewrite-tag-filter-0.0.3	lib/fluent/plugin/out_rewrite_tag_filter.rb
fluent-plugin-rewrite-tag-filter-0.0.2	lib/fluent/plugin/out_rewrite_tag_filter.rb