本文详解如何在 dynamodb 中通过条件更新与异常重试机制，安全、高效地实现“同月递增、跨月重置”的请求计数逻辑，避免竞态问题并确保数据一致性。
本文详解如何在 dynamodb 中通过条件更新与异常重试机制，安全、高效地实现“同月递增、跨月重置”的请求计数逻辑，避免竞态问题并确保数据一致性。
在 DynamoDB 中，原生不支持 IF-ELSE 类型的条件赋值表达式（如“若 activeMonth 等于当前月则 +1，否则重置为 1”）。因此，无法仅靠单条 UpdateItemCommand 的 UpdateExpression 完成带分支逻辑的原子更新。但可通过条件更新（ConditionExpression）+ 异常捕获 + 二次提交的组合策略，在应用层实现语义等价的原子行为。
✅ 推荐实现方案：两阶段条件更新
核心思路是分两步尝试：
- 乐观更新：假设当前月未变，仅对 requestCount 执行 +1，且要求 activeMonth 必须等于当前月份；
- 失败回退：若条件不满足（即 ConditionExpression 失败，抛出 ConditionalCheckFailedException），说明已跨月，此时执行重置逻辑——将 requestCount 设为 1，activeMonth 更新为当前月。
以下是完整、健壮的 TypeScript 实现（基于 AWS SDK v3）：
import { UpdateItemCommand, ConditionalCheckFailedException } from '@aws-sdk/client-dynamodb';
import { dynamoClient } from './dynamo-client'; // 你的客户端实例
const TABLE_NAME = 'YourTable';
const getCurrentMonth = () => new Date().getMonth() + 1; // 注意：getMonth() 返回 0~11，建议 +1 得到 1~12（更符合业务直觉）
export const updateRequestCount = async (accessKey: string) => {
const currentMonth = getCurrentMonth();
// 阶段一：尝试同月递增（带月条件校验）
try {
const command = new UpdateItemCommand({
TableName: TABLE_NAME,
Key: { accessKey: { S: accessKey } },
UpdateExpression:
'SET requestCount = if_not_exists(requestCount, :zero) + :inc',
ConditionExpression: 'activeMonth = :currentMonth',
ExpressionAttributeValues: {
':inc': { N: '1' },
':zero': { N: '0' },
':currentMonth': { N: currentMonth.toString() },
},
ReturnValues: 'ALL_NEW',
});
const response = await dynamoClient.send(command);
return response.Attributes!;
} catch (error) {
if (error instanceof ConditionalCheckFailedException) {
// 阶段二：跨月重置 —— 原子写入新月+计数=1
const resetCommand = new UpdateItemCommand({
TableName: TABLE_NAME,
Key: { accessKey: { S: accessKey } },
UpdateExpression:
'SET requestCount = :one, activeMonth = :currentMonth',
ExpressionAttributeValues: {
':one': { N: '1' },
':currentMonth': { N: currentMonth.toString() },
},
ReturnValues: 'ALL_NEW',
});
const response = await dynamoClient.send(resetCommand);
return response.Attributes!;
}
throw error; // 其他异常透传
}
};⚠️ 关键注意事项
- activeMonth 字段类型建议统一用 N（数字）：避免字符串比较歧义（如 '03' vs '3'），推荐存储为 1–12 的整数。
- if_not_exists(requestCount, :zero) 是安全兜底：首次写入时自动初始化为 0，后续递增才生效；重置路径中显式设为 1，语义清晰。
- 无竞态风险：两次操作均为原子 UpdateItem，且第二步不依赖读取旧值，完全规避了“读-改-写”窗口期。
- 性能友好：99% 场景下仅需一次请求（同月高频场景）；仅跨月瞬间触发一次额外请求，延迟可控。
- 可扩展性提示：若需按年+月复合判断（防跨年归零），可将 activeMonth 改为 YYYYMM 格式数字（如 202403），逻辑不变。
✅ 总结
DynamoDB 虽不支持服务端分支更新，但借助 ConditionExpression 的强一致性校验能力，配合客户端的异常驱动重试，即可优雅实现“月粒度计数器”的业务需求。该模式简洁、可靠、符合无状态服务设计原则，是 Serverless 和高并发场景下的推荐实践。
