DynamoDB单操作实现请求计数按月重置、同月递增的方案咨询

嗨，我来帮你搞定这个需求！你想要用单个DynamoDB操作实现「同月请求计数递增、跨月自动重置为1」的逻辑是完全可行的，不过你当前的代码还没法实现这个判断——它不管当前存储的月份是否和当前月份匹配，都会直接给requestCount加1，没法触发重置逻辑。

修改后的完整实现代码 #

我们可以借助DynamoDB的if()条件表达式来实现这个判断逻辑，直接在UpdateExpression里完成月份校验和计数更新：

const month = new Date().getMonth().toString();

const command = new UpdateItemCommand({
  TableName: TABLE_NAME,
  Key: {
    accessKey: { S: metadataAccessKey },
  },
  UpdateExpression: `
    SET requestCount = if(activeMonth = :activeMonth, requestCount + :incrementValue, :start + :incrementValue), 
        activeMonth = :activeMonth
  `,
  ExpressionAttributeValues: {
    ':incrementValue': { N: '1' },
    ':start': { N: '0' },
    ':activeMonth': { N: month },
  },
  ConditionExpression: 'attribute_exists(accessKey)',
  ReturnValues: 'ALL_NEW',
});

核心逻辑解释 #

• 这里的if(activeMonth = :activeMonth, requestCount + :incrementValue, :start + :incrementValue)是关键：
  • 当存储的activeMonth和当前月份:activeMonth一致时，执行requestCount + 1实现递增；
  • 当两者不一致时，执行0 + 1把计数重置为1（正好符合新月份的初始计数需求）；
• 同时我们会同步更新activeMonth为当前月份，确保下一次操作时的判断基准是最新的；
• 保留ConditionExpression: 'attribute_exists(accessKey)'是为了确保我们只更新已存在的条目，避免意外创建新条目。

这样就能用单个UpdateItem操作完成你想要的所有逻辑啦，不需要额外的查询或判断步骤。

备注：内容来源于stack exchange，提问作者Michael Sherris Caley