本文详解如何在 dynamodb 中通过条件更新与异常重试机制，实现“同月递增、跨月重置”的请求计数逻辑，规避单次 updateitem 无法原生支持分支判断的限制。
本文详解如何在 dynamodb 中通过条件更新与异常重试机制，实现“同月递增、跨月重置”的请求计数逻辑，规避单次 updateitem 无法原生支持分支判断的限制。
在 DynamoDB 中，UpdateExpression 不支持 IF/ELSE 类型的条件分支语法（如 if(activeMonth == :currentMonth, requestCount + 1, 1)），因此无法仅靠一次 UpdateItemCommand 完全实现“同月递增、跨月重置”的语义。但借助 条件表达式（ConditionExpression）+ 异常捕获 + 二次更新 的组合策略，我们可在应用层达成原子性行为，且保持高可靠性与低延迟。
✅ 推荐实现方案：两阶段条件更新
核心思路是分两步尝试：
- 第一阶段（乐观更新）：假设当前月份未变更，仅对 requestCount 执行自增，并用 ConditionExpression 确保 activeMonth 与当前月份一致；
- 第二阶段（兜底重置）：若第一步因条件不满足而抛出 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 currentMonth = new Date().getMonth(); // 注意：0~11，无需 toString()
// 第一阶段：尝试同月递增
const tryIncrement = async (accessKey: string) => {
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: String(currentMonth) },
},
ReturnValues: 'ALL_NEW',
});
try {
const response = await dynamoClient.send(command);
return response.Attributes;
} catch (err) {
if (err instanceof ConditionalCheckFailedException) {
// 条件失败 → 说明月份已变更，进入重置流程
return await resetAndSet(accessKey);
}
throw err; // 其他错误透传
}
};
// 第二阶段：跨月重置（原子写入新月+计数=1）
const resetAndSet = async (accessKey: string) => {
const command = new UpdateItemCommand({
TableName: TABLE_NAME,
Key: { accessKey: { S: accessKey } },
UpdateExpression:
'SET requestCount = :resetValue, activeMonth = :newMonth',
ExpressionAttributeValues: {
':resetValue': { N: '1' },
':newMonth': { N: String(currentMonth) },
},
// 可选：添加存在性检查防止误创建（按需）
ConditionExpression: 'attribute_exists(accessKey)',
ReturnValues: 'ALL_NEW',
});
return await dynamoClient.send(command);
};
// 使用示例
try {
const result = await tryIncrement('ak-12345');
console.log('Updated count:', result?.requestCount?.N); // e.g., "5" or "1"
} catch (error) {
console.error('Update failed:', error);
}⚠️ 关键注意事项
- 月份字段设计建议：使用 Number 类型存储 getMonth() 返回值（0–11），避免字符串比较歧义；如需年份维度（防跨年混淆），应扩展为 yearMonth: 202406 格式并同步校验。
- 并发安全：该方案在单 key 粒度下天然线程安全（DynamoDB 单 item 更新是原子的），无需额外锁机制。
- 性能考量：99% 场景下仅触发一次请求；仅在月初极短时间内可能触发两次，但 DynamoDB 条件检查开销极低，整体影响可忽略。
- 幂等性保障：若业务需严格幂等（如重试场景），建议在 UpdateExpression 中加入 if_not_exists(requestCount, :zero) 防空初始化，并统一由重置分支控制初始值。
✅ 总结
虽然 DynamoDB 不支持服务端 IF-ELSE 表达式，但通过 ConditionExpression 的精准约束 + 应用层异常驱动的 fallback 逻辑，我们能以简洁、可靠、高性能的方式实现“月粒度计数状态机”。该模式也适用于类似场景：如每日配额重置、会话活跃窗口刷新、周期性指标归零等。
