Batch methods for testing kafka producer with ruby gem

multi_batch_producer-kafka.rb

require 'kafka'

def batch(m1, m2, m3)
  producer = Kafka::Producer.new
  producer.batch do |msgs|
    puts "batching multi-message..."
    msgs << Kafka::Message.new("#{m1} ONE")
    msgs << Kafka::Message.new("#{m2} TWO")
    msgs << Kafka::Message.new("#{m3} THREE")
  end
end

def batch_multi(m1, m2, m3)
  producer = Kafka::Producer.new
  producer.batch do |msgs1|
    puts "batching multi-message..."
    msgs1 << Kafka::Message.new("#{m1} ONE")
    msgs1 << Kafka::Message.new("#{m2} TWO")
    msgs1 << Kafka::Message.new("#{m3} THREE")
  end
  producer.batch do |msgs2|
    puts "batching multi-message..."
    msgs2 << Kafka::Message.new("#{m1} 2... ONE")
    msgs2 << Kafka::Message.new("#{m2} 2... TWO")
    msgs2 << Kafka::Message.new("#{m3} 2... THREE")
  end
end

10.times do
  batch_multi "howdy", "there", "partner"
end