Node.js操作DynamoDB如何实现新增条目时Sort Key自动整数自增

该需求完全可以实现，核心通过DynamoDB原生原子计数器特性保证同user下id自增的唯一性和并发安全性，不会出现重复id的问题。你需要提前准备两个DynamoDB表：

• 主业务表：你当前使用的业务表，分区键为user，排序键为id（数字类型），存储实际业务数据
• 计数器表：专门存储每个user对应的当前最大id，仅需设置分区键为user，额外保留普通数字属性current_max_id即可

• 每次新增主表数据前，先调用UpdateItem接口操作计数器表，对指定user的current_max_id执行原子加1操作，同时返回更新后的最新值
• 将返回的新id作为主表数据的排序键，调用PutItem接口写入完整业务数据
• 原子加操作为DynamoDB原生支持，即使多并发请求同时操作同一个user的计数器，也不会出现id重复问题

以下代码基于AWS SDK v3编写，是官方当前推荐的版本：

1. 先安装依赖
   npm install @aws-sdk/client-dynamodb @aws-sdk/util-dynamodb

2. 业务代码示例

const { DynamoDBClient, UpdateItemCommand, PutItemCommand } = require("@aws-sdk/client-dynamodb");
const { marshall, unmarshall } = require("@aws-sdk/util-dynamodb");

// 初始化DynamoDB客户端，替换为你自己的AWS区域
const client = new DynamoDBClient({ region: "ap-northeast-1" });
// 替换为你自己的实际表名
const COUNTER_TABLE = "user_id_counter";
const MAIN_TABLE = "your_business_table";

async function addItemWithAutoIncrementId(user, businessData) {
  // 第一步：原子更新计数器，获取新的自增id
  const updateCounterCmd = new UpdateItemCommand({
    TableName: COUNTER_TABLE,
    Key: marshall({ user }),
    // 如果是user首次生成id，自动初始化初始值，初始id从1开始计算
    UpdateExpression: "SET current_max_id = if_not_exists(current_max_id, :init) + :step",
    ExpressionAttributeValues: marshall({
      ":init": 0,
      ":step": 1
    }),
    ReturnValues: "UPDATED_NEW"
  });

  const counterRes = await client.send(updateCounterCmd);
  const newId = unmarshall(counterRes.Attributes).current_max_id;

  // 第二步：写入主业务表
  const putItemCmd = new PutItemCommand({
    TableName: MAIN_TABLE,
    Item: marshall({
      user,
      id: newId,
      ...businessData
    })
  });

  await client.send(putItemCmd);

  return { user, id: newId, ...businessData };
}

// 调用示例
// addItemWithAutoIncrementId("user_001", { content: "测试内容", createTime: Date.now() })
//   .then(res => console.log("写入成功", res))
//   .catch(err => console.error("写入失败", err));

• 如果你需要首个生成的id为0，只需将代码中:init的取值改为-1即可
• 如果业务要求极强一致性，可将两次操作合并为DynamoDB事务操作，避免计数器更新成功但主表写入失败导致的id跳号问题
• 计数器表无需提前为每个user创建条目，if_not_exists参数会在user首次请求时自动初始化计数器
• 不建议通过每次查询主表最大值再加1的方式生成id，并发场景下会出现id重复冲突的问题

内容的提问来源于stack exchange，提问作者M H