如何通过JavaScript对象更新DynamoDB嵌套属性？求现成代码

现有基于AWS DynamoDB DocumentClient的更新代码仅支持平级属性（如{a:1}），无法处理嵌套对象（如{stats:{success:1, failure:{reason:'bla', count:2}}}）。原因是DynamoDB要求UpdateExpression中指定嵌套路径（如SET stats.success = :val1），但表达式属性值的占位符不能使用带点的格式（如:stats.failure.reason是无效的）。

以下是改进后的代码，可支持任意层级的嵌套对象更新：

const docClient = new AWS.DynamoDB.DocumentClient({ region: "us-east-2" });

const PK = (userId) => `USER#${userId}`;
const SK = (file) => `FILE#${file}`;

// 递归遍历嵌套对象，生成合法的UpdateExpression片段和属性值映射
const parseNestedAttributes = (attributes, parentPath = "") => {
  let updateParts = [];
  let attrValues = {};
  let counter = 0;

  for (const [key, value] of Object.entries(attributes)) {
    const currentPath = parentPath ? `${parentPath}.${key}` : key;
    if (typeof value === 'object' && value !== null && !Array.isArray(value)) {
      // 递归处理嵌套子对象
      const nestedResult = parseNestedAttributes(value, currentPath);
      updateParts = [...updateParts, ...nestedResult.updateParts];
      attrValues = { ...attrValues, ...nestedResult.attrValues };
    } else {
      // 生成唯一占位符，避免无效格式和命名冲突
      const placeholder = `:val${counter++}`;
      updateParts.push(`${currentPath} = ${placeholder}`);
      attrValues[placeholder] = value;
    }
  }

  return { updateParts, attrValues };
};

const updateFn = async (tableName, { pk, sk }, attributes) => {
  const { updateParts, attrValues } = parseNestedAttributes(attributes);
  
  if (updateParts.length === 0) {
    console.log("没有需要更新的属性");
    return null;
  }

  const params = {
    Key: { pk: PK(`${pk}`), sk: SK(`${sk}`) },
    UpdateExpression: "SET " + updateParts.join(", "),
    ExpressionAttributeValues: attrValues,
  };

  try {
    const records = await docClient
      .update({
        TableName: tableName,
        ...params,
      })
      .promise();
    console.log("更新结果", records);
    return records;
  } catch (err) {
    console.log("更新错误", err);
    return null;
  }
};

const TABLE_NAME = "FileTracker";

// 测试嵌套对象更新
updateFn(
  TABLE_NAME,
  { pk: "pk", sk: "sk" },
  {
    score: 32,
    level: "1b2",
    stats: {
      success: 1,
      failure: {
        reason: 'bla',
        count: 2
      }
    }
  }
);

关键逻辑说明 #

• parseNestedAttributes函数递归遍历属性对象，为嵌套属性生成DynamoDB支持的路径格式（如stats.failure.reason）
• 使用自增计数器生成唯一占位符（:val0、:val1等），规避带点占位符的无效格式问题，同时防止不同层级属性的命名冲突
• 最终将所有路径片段拼接成合法的UpdateExpression，属性值统一整理到ExpressionAttributeValues中

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