logstash/lib/net/messagestream.rb

require 'lib/net/common'

module LogStash; module Net
  class MessageStream

    def initialize
      @data = Hash.new
      @data["version"] = PROTOCOL_VERSION
      @data["messages"] = Array.new
    end # def initialize

    def <<(message)
      @data["messages"] << message
    end # def <<

    def message_count
      return @data["messages"].length
    end

    def clear
      @data["messages"] = Array.new
    end # def clear

    def encode
      jsonstr = JSON::dump(@data)
      return jsonstr
    end # def encode

    # unused for now
    def _unused____sendto(sock)
      data = self.encode
      puts "Writing #{data.length} bytes to #{sock}"
      bytestream = [data.length, data.checksum, data].pack("NNA*")
      sock.write(bytestream)
      self.clear
    end # def sendto

    def self.decode(string, &block)
      data = JSON::parse(string)
      ms = MessageStream.new
      if data["version"] != PROTOCOL_VERSION
        # throw some kind of error
      end

      responses = []
      data["messages"].each do |msgdata|
        msg = Message.new_from_data(msgdata)
        responses << (yield msg)
      end
      return responses
    end # def self.decode
  end # class MessageStream
end; end # module LogStash::Net