DynamoDB-Toolbox 中批量获取数据的正确实践

在使用 DynamoDB-Toolbox 进行批量数据获取时，开发者经常会遇到如何处理大容量数据请求和未处理键(UnprocessedKeys)的问题。本文将深入探讨如何高效地实现批量数据获取，并正确处理各种边界情况。

批量获取的基本原理

DynamoDB 的 BatchGetItem 操作允许开发者一次性获取多个项目，但存在几个重要限制：

• 单次请求最多只能获取100个项目
• 响应总大小不能超过16MB
• 可能因容量限制返回部分未处理的项目

DynamoDB-Toolbox 提供了优雅的封装来处理这些复杂情况，开发者无需从头实现重试逻辑。

核心实现策略

1. 分块处理

对于超过100个项目的大批量请求，首先需要进行分块处理：

const chunkSize = 100;
const chunks: KeyInputItem<this>[][] = [];
for (let i = 0; i < keys.length; i += chunkSize) {
  chunks.push(keys.slice(i, i + chunkSize));
}

2. 自动重试机制

DynamoDB-Toolbox 内置了自动重试功能，通过 maxAttempts 参数控制：

const command = this.table.build(BatchGetCommand)
  .options({ maxAttempts: Infinity });

设置为 Infinity 时，工具会持续重试直到所有项目都成功获取或遇到不可恢复的错误。

高级场景处理

大项目处理

当单个项目过大导致无法在16MB限制内返回所有请求项目时，DynamoDB 会返回部分结果和未处理键。DynamoDB-Toolbox 会自动处理这种情况，开发者无需额外代码。

预置容量场景

在使用预置容量模式时，可能出现吞吐量超出限制的情况。此时：

• 如果没有任何项目能处理，会抛出 ProvisionedThroughputExceededException
• 如果有部分项目成功处理，会返回成功并包含未处理键

DynamoDB-Toolbox 会正确处理第二种情况，自动重试未处理键。

最佳实践建议

1. 对于常规使用，直接依赖 DynamoDB-Toolbox 的内置重试机制即可
2. 对于超大请求(>100项目)，先进行手动分块
3. 考虑设置合理的 maxAttempts 值，避免无限重试
4. 按需付费模式(PAY_PER_REQUEST)下吞吐量问题较少，但仍需处理大项目场景

类型安全提示

在使用返回类型时，可以引用 ExecuteBatchGetResponses 类型来保证类型安全，避免使用 any 类型。

通过合理利用 DynamoDB-Toolbox 的这些特性，开发者可以构建出健壮、高效的批量数据获取逻辑，同时保持代码简洁。