kinesis_put.rb

kinesis_put.rb

require 'aws-sdk'

print "Enter target stream name: "
stream_name = gets.chomp

begin
  client = AWS::Kinesis.new(
    access_key_id: ENV['AWS_ACCESS_KEY_ID'],
    secret_access_key: ENV['AWS_SECRET_ACCESS_KEY']
  ).client

  categories = %w{Foods Books Toys Electronics Sports Clothing Shoes Music}
  loop do
    now = Time.now.to_s
    partition_key = categories.sample
    response = client.put_record(
      stream_name: stream_name,
      data: now,
      partition_key: partition_key
    )
    puts "Data: #{now}, Partition Key: #{partition_key}, Shard ID: #{response.shard_id}, Seq Number: #{response.sequence_number}"
    p response
    sleep 1
  end
rescue => e
  puts "Error: #{e.message}"
  abort
end