# File lib/fluent/plugin/in_stream.rb, line 117 def initialize(io, log, 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 @on_message = on_message @log = log @log.trace { remote_port, remote_addr = *Socket.unpack_sockaddr_in(@_io.getpeername) rescue nil "accepted fluent socket from '#{remote_addr}:#{remote_port}': object_id=#{self.object_id}" } end
# File lib/fluent/plugin/in_stream.rb, line 167 def on_close @log.trace { "closed fluent socket object_id=#{self.object_id}" } end
# File lib/fluent/plugin/in_stream.rb, line 131 def on_connect end
# File lib/fluent/plugin/in_stream.rb, line 134 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 = Fluent::Engine.msgpack_factory.unpacker end (class << self; self; end).module_eval do define_method(:on_read, m) end m.call(data) end
# File lib/fluent/plugin/in_stream.rb, line 151 def on_read_json(data) @y << data rescue @log.error "unexpected error", error: $!.to_s @log.error_backtrace close end
# File lib/fluent/plugin/in_stream.rb, line 159 def on_read_msgpack(data) @u.feed_each(data, &@on_message) rescue @log.error "unexpected error", error: $!.to_s @log.error_backtrace close end