class Fluent::ForwardInput::Handler

def initialize(io, on_message)
  super(io)
  if io.is_a?(TCPSocket)
    opt = [1, @timeout.to_i].pack('I!I!')  # { int l_onoff; int l_linger; }
    io.setsockopt(Socket::SOL_SOCKET, Socket::SO_LINGER, opt)
  end
  $log.trace { "accepted fluent socket object_id=#{self.object_id}" }
  @on_message = on_message
end

def on_close
  $log.trace { "closed fluent socket object_id=#{self.object_id}" }
end

def on_connect
end

def on_read(data)
  first = data[0]
  if first == '{' || first == '['
    m = method(:on_read_json)
    @y = Yajl::Parser.new
    @y.on_parse_complete = @on_message
  else
    m = method(:on_read_msgpack)
    @u = MessagePack::Unpacker.new
  end
  (class<<self;self;end).module_eval do
    define_method(:on_read, m)
  end
  m.call(data)
end

def on_read_json(data)
  @y << data
rescue
  $log.error "forward error: #{$!.to_s}"
  $log.error_backtrace
  close
end

def on_read_msgpack(data)
  @u.feed_each(data, &@on_message)
rescue
  $log.error "forward error: #{$!.to_s}"
  $log.error_backtrace
  close
end