Contents

class Fluent::StringScrubFilter < Fluent::Filter
  Fluent::Plugin.register_filter('string_scrub', self)

  config_param :replace_char, :string, :default => ''

  def initialize
    super
    require 'string/scrub' if RUBY_VERSION.to_f < 2.1
  end

  def configure(conf)
    super

    if @replace_char =~ /\\u\{*[A-F0-9]{4}\}*/
      @replace_char = eval("\"#{@replace_char}\"")
    end
  end

  def filter_stream(tag, es)
    new_es = Fluent::MultiEventStream.new
    es.each do |time,record|
      begin
        scrubbed = recv_record(record)
        next if scrubbed.nil?
        new_es.add(time, record)
      rescue => e
        router.emit_error_event(tag, time, record, e)
      end
    end

    new_es
  end

  def recv_record(record)
    scrubbed = {}
    record.each do |k,v|
      if v.instance_of? Hash
        scrubbed[with_scrub(k)] = recv_record(v)
      else
        scrubbed[with_scrub(k)] = with_scrub(v)
      end
    end
    scrubbed
  end

  def with_scrub(string)
    begin
      string =~ //
      return string
    rescue ArgumentError => e
      raise e unless e.message.index("invalid byte sequence in") == 0
      if string.frozen?
          string = string.dup.scrub!(@replace_char)
      else
          string.scrub!(@replace_char)
      end
      retry
    end
  end
end if defined?(Fluent::Filter) # Support only >= v0.12

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
fluent-plugin-string-scrub-0.1.0	lib/fluent/plugin/filter_string_scrub.rb