Sha256: f984e0f33970108eb3a371969017a77584a4479be8913c21d66eabb95d773758

Contents?: true

Size: 1.29 KB

Versions: 11

Compression:

Stored size: 1.29 KB

Contents

module Kafka
  module Protocol
    class MessageSet
      attr_reader :messages

      def initialize(messages: [])
        @messages = messages
      end

      def size
        @messages.size
      end

      def ==(other)
        messages == other.messages
      end

      def encode(encoder)
        # Messages in a message set are *not* encoded as an array. Rather,
        # they are written in sequence.
        @messages.each do |message|
          message.encode(encoder)
        end
      end

      def self.decode(decoder)
        fetched_messages = []

        until decoder.eof?
          begin
            message = Message.decode(decoder)

            if message.compressed?
              fetched_messages.concat(message.decompress)
            else
              fetched_messages << message
            end
          rescue EOFError
            if fetched_messages.empty?
              # If the first message in the set is truncated, it's likely because the
              # message is larger than the maximum size that we have asked for.
              raise MessageTooLargeToRead
            else
              # We tried to decode a partial message at the end of the set; just skip it.
            end
          end
        end

        new(messages: fetched_messages)
      end
    end
  end
end

Version data entries

11 entries across 11 versions & 1 rubygems

Version Path
ruby-kafka-0.6.0.beta4 lib/kafka/protocol/message_set.rb
ruby-kafka-0.6.0.beta3 lib/kafka/protocol/message_set.rb
ruby-kafka-0.6.0.beta2 lib/kafka/protocol/message_set.rb
ruby-kafka-0.6.0.beta1 lib/kafka/protocol/message_set.rb
ruby-kafka-0.5.5 lib/kafka/protocol/message_set.rb
ruby-kafka-0.5.4 lib/kafka/protocol/message_set.rb
ruby-kafka-0.5.4.beta1 lib/kafka/protocol/message_set.rb
ruby-kafka-0.5.3 lib/kafka/protocol/message_set.rb
ruby-kafka-0.5.2 lib/kafka/protocol/message_set.rb
ruby-kafka-0.5.2.beta3 lib/kafka/protocol/message_set.rb
ruby-kafka-0.5.2.beta2 lib/kafka/protocol/message_set.rb