使用PutRecords将多个记录加载到Kinesis-如何在失败的情况下仅重新发送失败的记录?

时间:2019-12-15 22:47:57

标签: amazon-web-services aws-lambda amazon-kinesis amazon-kinesis-kpl

我正在使用Lambda将数据记录加载到Kinesis中,并且经常想添加多达500K条记录,我将这些记录分为500条,然后使用Boto的put_records方法将其发送到Kinesis。有时会由于超出允许的吞吐量而看到失败。

发生这种情况时重试的最佳方法是什么?理想情况下,我不想在数据流中出现重复的消息,所以我不想简单地重新发送所有500条记录,但是我在努力查看如何仅重试失败的消息。 put_records方法的响应似乎不太有用。

我可以依靠响应记录列表的顺序与传递给putRecords的列表的顺序相同吗?

我知道我可以增加分片的数量,但是我想显着增加向该Kinesis流加载数据的并行Lambda函数的数量。我们计划基于源系统对数据进行分区,但我不能保证多个功能不会将数据写入同一分片并且不会超过允许的吞吐量。因此,我不认为增加分片将消除重试策略的必要性。

或者,有人知道KPL是否会为我自动处理此问题吗?

1 个答案:

答案 0 :(得分:1)

  

我可以依靠响应记录列表的顺序与传递给putRecords的列表的顺序相同吗?

是的。您将不得不依靠响应的顺序。响应记录的顺序与请求记录的顺序相同。

请检查putrecords的响应,https://docs.aws.amazon.com/kinesis/latest/APIReference/API_PutRecords.html

  

记录:   一系列成功和未成功处理的记录结果,通过自然排序与请求相关联。成功添加到流中的记录在结果中包含SequenceNumber和ShardId。未能添加到流中的记录在结果中包含ErrorCode和ErrorMessage。

要重试失败的记录,您必须开发自己的重试机制。我已经在python中使用递归函数编写了重试机制,并以以下方式在重试之间进行了增量等待。

import boto3
import time

kinesis_client = boto3.client('kinesis')
KINESIS_RETRY_COUNT = 10
KINESIS_RETRY_WAIT_IN_SEC = 0.1
KINESIS_STREAM_NAME = "your-kinesis-stream"

def send_to_stream(kinesis_records, retry_count):
    put_response = kinesis_client.put_records(
        Records=kinesis_records,
        StreamName=KINESIS_STREAM_NAME
    )
    failed_count = put_response['FailedRecordCount']
    if failed_count > 0:
        if retry_count > 0:
            retry_kinesis_records = []
            for idx, record in enumerate(put_response['Records']):
                if 'ErrorCode' in record:
                    retry_kinesis_records.append(kinesis_records[idx])
            time.sleep(KINESIS_RETRY_WAIT_IN_SEC * (KINESIS_RETRY_COUNT - retry_count + 1))
            send_to_stream(retry_kinesis_records, retry_count - 1)
        else:
            print(f'Not able to put records after retries. Records = {put_response["Records"]}')

在上面的示例中,您可以根据需要更改KINESIS_RETRY_COUNTKINESIS_RETRY_WAIT_IN_SEC。另外,您还必须确保Lambda超时足以重试。

  

或者,有人知道KPL是否会自动处理此问题   对我来说问题?

我不确定KPL,但是从文档看来,它具有自己的重试机制。 https://docs.aws.amazon.com/streams/latest/dev/kinesis-producer-adv-retries-rate-limiting.html