aws_kinesis_agg_record_aggregator_quickstart.py

1import aws_kinesis_agg.aggregator
2
3# Create an aggregator object
4kinesis_aggregator = aws_kinesis_agg.aggregator.RecordAggregator()
5
6# Add records to the aggregator
7# The add_user_record method takes: partition_key, data, and an optional explicit_hash_key
8kinesis_aggregator.add_user_record('partition_key_1', 'my_data_1')
9kinesis_aggregator.add_user_record('partition_key_2', 'my_data_2')
10
11# When you are ready to send the records, call clear_and_get_aggregate()
12# This returns a single AggregatedRecord object (if any records were added)
13agg_record = kinesis_aggregator.clear_and_get_aggregate()
14
15if agg_record:
16    # Use the aggregated record's properties to send to Kinesis via boto3
17    # pk = agg_record.partition_key
18    # ehk = agg_record.explicit_hash_key
19    # data = agg_record.get_contents()
20    print(f"Successfully created aggregated record with {len(agg_record.user_records)} user records.")