如何在单条AWS GraphQL查询中获取正确数量的返回项

你对问题原因的判断完全正确：Amplify GraphQL对接DynamoDB时，limit参数控制的是DynamoDB底层单次扫描/查询的原始记录条数，系统会先取出对应条数的原始记录，再在服务端应用过滤条件，最终返回符合规则的记录，所以最终返回的items数量会小于你设置的limit值，属于底层机制的正常表现。

方案1：创建二级索引（最优方案） #

这是性能最高、完全符合你预期的实现方式。你高频使用type字段作为过滤条件，直接为该字段创建全局二级索引（GSI）即可让查询直接命中索引，跳过全表扫描后过滤的流程：

• 在你的Amplify GraphQL schema中给type字段添加索引配置，示例如下：

type Item @model {
  id: ID!
  type: String! @index(name: "itemsByType", queryField: "listItemsByType", sortKeyFields: ["price"])
  price: Float
  status: String
  name: String
}

• 部署schema更新后，直接调用新生成的listItemsByType查询，传入type: "ropa"和limit:12参数，此时返回的items数量就是符合条件的记录数，最多为12条，不需要额外处理分页拼接逻辑。如果需要排序，在建索引时指定排序键即可，排序逻辑也会在索引层直接完成，不会影响limit的生效规则。

方案2：封装自动分页拉取工具函数 #

如果暂时不想修改schema加索引，可以封装一个通用的查询函数，自动处理多页拉取拼接逻辑，对上层业务屏蔽分页细节：

• 函数逻辑为：接收过滤条件、目标条数、当前nextToken三个参数，调用listItems查询后将符合条件的item存入结果数组，如果结果数组长度达到目标条数或者nextToken为空，就返回结果和新的nextToken；如果没凑够，就递归调用自身传入新的nextToken，直到满足停止条件。
• 示例伪代码如下：

async function queryItemsWithLimit(filter, targetLimit) {
  let result = []
  let nextToken = null
  do {
    const res = await API.graphql({
      query: listItems,
      variables: { filter, limit: 100, nextToken } // 单次底层查询设大一些减少请求次数
    })
    result = result.concat(res.data.listItems.items)
    nextToken = res.data.listItems.nextToken
  } while (result.length < targetLimit && nextToken)
  // 截取刚好目标条数的记录，多余的下次拉取用
  const returnItems = result.slice(0, targetLimit)
  const newNextToken = result.length > targetLimit ? nextToken : null
  return { items: returnItems, nextToken: newNextToken }
}

方案1是官方推荐的标准实现，无论性能还是开发成本都远优于方案2，如果你后续还有其他过滤、排序需求，都可以通过建对应索引的方式解决，不需要在业务层做复杂的分页拼接处理。

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