Kinesis Data Streams に put するだけの Python スクリプト

• yoheia/kds_put2stream.py at master · yoheia/yoheia · GitHub

#!/usr/bin/env python3
import boto3
import json
from datetime import datetime
import calendar
import random
import time

my_stream_name = 'KDS-ThroughputTest'

kinesis_client = boto3.client('kinesis', region_name='ap-northeast-1')

def put_to_stream(thing_id, property_value, property_timestamp):
    payload = {
                'prop': str(property_value),
                'timestamp': str(property_timestamp),
                'thing_id': thing_id
              }
    print(payload)
    put_response = kinesis_client.put_record(
                        StreamName=my_stream_name,
                        Data=json.dumps(payload),
                        PartitionKey=thing_id)

while True:
    property_value = random.randint(40, 120)
    property_timestamp = calendar.timegm(datetime.utcnow().timetuple())
    thing_id = 'aa-bb'
    put_to_stream(thing_id, property_value, property_timestamp)