本文详解如何在 dynamodb 中通过条件更新与异常重试机制，实现请求计数（requestcount）按月自动重置（新月份归零）或递增（同月+1），规避单次 updateitem 无法执行分支逻辑的限制。
本文详解如何在 dynamodb 中通过条件更新与异常重试机制，实现请求计数（requestcount）按月自动重置（新月份归零）或递增（同月+1），规避单次 updateitem 无法执行分支逻辑的限制。
在 DynamoDB 中，UpdateItem 操作不支持原生的 IF-ELSE 分支逻辑（如“若 activeMonth 匹配则 +1，否则重置为 1”），其 UpdateExpression 仅支持函数式操作（如 if_not_exists, list_append），无法根据属性值动态选择更新路径。因此，试图用一条命令同时完成“同月递增、跨月重置”是不可行的——这正是原始代码中无条件覆盖 activeMonth 并始终 +1 所导致的逻辑缺陷。
✅ 正确做法是采用 “乐观条件更新 + 异常驱动回退” 的两阶段策略：
第一尝试：假设仍在同一月份
使用 ConditionExpression 校验当前 activeMonth 是否等于本月值；若匹配，则安全地递增 requestCount 并保持 activeMonth 不变。失败回退：捕获 ConditionalCheckFailedException
若条件不满足（即已跨月），DynamoDB 将拒绝该请求并抛出异常。此时捕获异常，在同一事务上下文中发起第二次 UpdateItem，将 requestCount 重置为 1，并更新 activeMonth 为当前月份。
以下是生产就绪的 TypeScript 实现（基于 AWS SDK v3）：
import {
UpdateItemCommand,
ConditionalCheckFailedException
} from '@aws-sdk/client-dynamodb';
import { dynamoClient } from './dynamo-client'; // your configured client
const getCurrentMonth = (): number => new Date().getMonth() + 1; // 1–12 (Jan=1)
export const updateRequestCount = async (accessKey: string) => {
const currentMonth = getCurrentMonth();
// ✅ Step 1: Try to increment only if month matches
try {
const command = new UpdateItemCommand({
TableName: TABLE_NAME,
Key: { accessKey: { S: accessKey } },
UpdateExpression: 'SET requestCount = requestCount + :inc',
ConditionExpression: 'activeMonth = :month',
ExpressionAttributeValues: {
':inc': { N: '1' },
':month': { N: currentMonth.toString() },
},
ReturnValues: 'ALL_NEW',
});
return await dynamoClient.send(command);
} catch (error) {
if (error instanceof ConditionalCheckFailedException) {
// ❌ Month changed → reset counter and update month
const resetCommand = new UpdateItemCommand({
TableName: TABLE_NAME,
Key: { accessKey: { S: accessKey } },
UpdateExpression: 'SET requestCount = :reset, activeMonth = :month',
ExpressionAttributeValues: {
':reset': { N: '1' },
':month': { N: currentMonth.toString() },
},
// No condition — allow overwrite on month change
ReturnValues: 'ALL_NEW',
});
return await dynamoClient.send(resetCommand);
}
throw error; // re-throw other errors (e.g., network, permission)
}
};⚠️ 关键注意事项：
- 月份字段设计建议：使用 1–12（而非 0–11）更符合业务直觉，避免 getMonth() 返回 0（January）引发混淆；也可考虑存储 YYYY-MM 字符串（如 "2024-06"）以支持跨年判断，此时需将 activeMonth 类型设为 S，ConditionExpression 改为 'activeMonth = :month'，ExpressionAttributeValues 中传入字符串。
- 幂等性保障：该方案天然具备幂等性——重复调用不会导致计数错乱（因条件更新只在匹配时生效，重置操作也只在必要时触发）。
- 性能与成本：绝大多数请求（同月场景）仅消耗 1 次写容量单位（WCU）；跨月场景最多 2 次 WCU，远优于先 GetItem 再 UpdateItem 的两轮往返（且避免竞态条件）。
- 错误处理边界：务必显式捕获 ConditionalCheckFailedException，避免误将其他异常（如 ProvisionedThroughputExceededException）当作“月份变更”处理。
总结而言，DynamoDB 的强一致性条件更新能力，配合 SDK 的异常分类机制，可优雅支撑此类“状态感知型计数”需求。无需引入外部协调服务（如 Lambda + EventBridge），也无需妥协数据一致性——这是 Serverless 架构下轻量、可靠、可扩展的最佳实践。
