如何让AWS AppSync GraphQL使用DynamoDB PK/SK查询而非扫描

阿华AIGC实验室

2026-5-11

解决Amplify AppSync GraphQL API针对DynamoDB PK/SK执行查询而非扫描的问题

你的问题核心在于：Amplify自动生成的listDataSummaries API本质上绑定了DynamoDB的Scan操作——哪怕你加了filter条件，它也是先做全表扫描，再在结果里过滤数据。这完全浪费了你设计的PK+SK主键结构的索引优势，不仅消耗更多RCU，还会因为全表扫描的分页机制导致返回不符合预期的结果。

你遗漏的关键点

自动生成的list类API是为通用的全表/索引遍历场景设计的，它不会识别你用PK和SK做过滤的意图，自然不会触发DynamoDB高效的Query操作（只有Query才能利用主键或二级索引的索引结构，直接定位到指定分区内的数据）。要让GraphQL调用DynamoDB的Query，你需要自定义查询字段并配置对应的Resolver。

具体解决方案

1. 修改GraphQL Schema，添加自定义查询

在你的Schema里新增一个专门针对PK和SK范围的查询字段，替代自动生成的list API：

type Query {
  # 新增：按PK和SK范围查询数据，支持分页
  getDataSummariesByPKAndSKRange(
    PK: String!
    minSK: Float!
    maxSK: Float!
    limit: Int
    nextToken: String
  ): DataSummaryConnection
}

type DataSummary {
  PK: String!
  SK: Float!
  Summary: AWSJSON
}

# 定义分页连接类型（可选，用于返回分页标记）
type DataSummaryConnection {
  items: [DataSummary]
  nextToken: String
}

2. 配置AppSync Resolver，映射到DynamoDB Query操作

为这个自定义查询编写请求映射模板，告诉AppSync执行DynamoDB的Query而非Scan：

{
  "version": "2017-02-28",
  "operation": "Query",
  "query": {
    "expression": "PK = :pk AND SK BETWEEN :minSK AND :maxSK",
    "expressionValues": {
      ":pk": { "S": "${context.arguments.PK}" },
      ":minSK": { "N": "${context.arguments.minSK.toString()}" },
      ":maxSK": { "N": "${context.arguments.maxSK.toString()}" }
    }
  },
  # 指定使用主表的主键索引（必须，确保利用PK+SK的索引）
  "index": "primary",
  "select": "ALL_ATTRIBUTES",
  # 处理可选分页参数
  #if($context.arguments.limit)
  "limit": $context.arguments.limit,
  #end
  #if($context.arguments.nextToken)
  "nextToken": "$context.arguments.nextToken",
  #end
}

响应映射模板可以用以下内容，确保返回符合定义的连接类型：

{
  "items": $util.toJson($context.result.items),
  "nextToken": $util.toJson($context.result.nextToken)
}

3. 使用自定义查询获取数据

现在你可以用新的查询高效获取指定PK下SK范围内的数据：

query GetTargetDataSummaries {
  getDataSummariesByPKAndSKRange(
    PK: "your-target-pk-string",
    minSK: 1690000000.0,
    maxSK: 1700000000.0,
    limit: 50
  ) {
    items {
      PK
      SK
      Summary
    }
    nextToken
  }
}

这个查询会直接触发DynamoDB的Query操作，精准定位到指定PK分区内、SK在你设定范围内的数据，完全符合你的表设计初衷，RCU消耗也会大幅降低。