如何通过单个查询从DynamoDB按PK和多个SK获取多条数据

DynamoDB的KeyConditionExpression不支持OR逻辑，要实现已知PK和多个SK的批量查询，结合你的场景（USR分区数据≤1000条），推荐以下两种高效方案：

方案一：使用BatchGetItem API（优先推荐） #

BatchGetItem是DynamoDB专门用于批量获取多主键条目的API，支持单次请求最多获取100条数据，内部会并行处理请求，比循环执行单个Query效率更高，且RCU消耗仅按实际读取的条目计算，不会浪费资源。

代码示例：

const params = {
  RequestItems: {
    [MY_TABLE_NAME]: {
      Keys: [
        { PK: "USR", SK: "1234" },
        { PK: "USR", SK: "9101" }
        // 可添加更多(PK, SK)对，单次最多100个
      ]
    }
  }
};

try {
  const data = await dynamoClient.batchGet(params).promise();
  const users = data.Responses[MY_TABLE_NAME] || [];
  
  // 处理未成功获取的键（如请求超过限制时）
  if (data.UnprocessedKeys && Object.keys(data.UnprocessedKeys).length > 0) {
    const retryParams = { RequestItems: data.UnprocessedKeys };
    const retryData = await dynamoClient.batchGet(retryParams).promise();
    users.push(...retryData.Responses[MY_TABLE_NAME]);
  }
  return users;
} catch (err) {
  callback(new Error(`[500] ${err.message}`));
}

说明：若SK列表长度超过100，只需将列表分批，每次请求传入100个Keys即可，保证请求效率。

方案二：Query整个USR分区+客户端过滤（适合SK列表接近分区总条目数） #

由于你的USR分区数据量不超过1000条，一次性Query整个分区的RCU消耗可控（1KB数据对应1个RCU，单条用户数据通常很小，1000条仅需少量RCU），之后在客户端过滤出目标SK的条目，比循环单个Query减少网络开销。

代码示例：

const targetSKs = new Set(["1234", "9101"]); // 用Set提升过滤效率
const params = {
  TableName: MY_TABLE_NAME,
  KeyConditionExpression: "PK = :usr",
  ExpressionAttributeValues: { ":usr": "USR" }
};

try {
  const users = [];
  let data;
  do {
    data = await dynamoClient.query(params).promise();
    // 客户端过滤目标SK条目
    const filteredItems = data.Items.filter(item => targetSKs.has(item.SK));
    users.push(...filteredItems);
    params.ExclusiveStartKey = data.LastEvaluatedKey;
  } while (typeof data.LastEvaluatedKey !== "undefined");
  return users;
} catch (err) {
  callback(new Error(`[500] ${err.message}`));
}

说明：该方案适合需要获取的SK数量接近USR分区总条目数的场景，若SK列表仅包含少量条目，仍优先选择BatchGetItem。

关于KeyConditionExpression不支持OR的原因 #

DynamoDB的Query操作基于主键索引的有序性设计，KeyConditionExpression仅支持与索引有序逻辑兼容的操作符（EQ、LE、LT、GE、GT、BEGINS_WITH、BETWEEN），OR操作不符合有序索引的查询逻辑，因此不被支持。

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