客户端侧GraphQL查询代价计算实现方案咨询

以下是可落地的优化方向，按投入产出比从高到低排序：

• 优先加查询代价缓存
  绝大部分业务场景下客户端发送的重复查询占比很高，你可以直接以查询字符串+影响代价的变量参数（如分页的first/last、嵌套层级控制参数）为key，把计算好的代价存在内存缓存里，命中直接返回，完全省去重复解析AST、遍历计算的开销。注意缓存可以设置过期时间，或者按最大容量做LRU淘汰，避免内存溢出。
• 预加载Schema类型元数据，减少运行时判断成本
  纯AST遍历无法直接区分字段的返回类型，你可以提前把服务端的Schema introspection结果拉到本地，预先标记好所有连接类型、对象类型的名称，遍历AST的时候直接匹配类型名称套用代价规则，不需要每次都做动态类型推断，遍历速度至少能提升30%以上。
• 预处理片段展开，降低AST遍历复杂度
  如果查询里大量使用命名片段、内联片段，原生AST遍历需要频繁跳转查询片段定义，效率很低。你可以在计算代价之前先跑一遍片段展开逻辑，把所有片段的选择集合并到对应的父级选择节点里，后续计算代价的时候只需要遍历一层平的选择树即可，不用处理跨节点的引用跳转。
• 拆分固定/动态代价分开计算
  把查询里不随变量变化的固定代价（比如固定选择的对象、基础连接的2点基础分）和随变量变化的动态代价（比如分页参数带来的额外代价、变量控制的可选字段代价）拆分，固定代价部分直接缓存，每次请求只需要重新计算和变量相关的动态部分即可，进一步降低计算量。
• 前置轻量校验拦截极端请求
  对于长度超过阈值的超大查询、嵌套层级肉眼可见过高的查询，可以直接在解析AST之前就按最高代价拦截限流，不需要跑完整的计算逻辑，能有效挡住恶意构造的消耗资源的查询请求。

简化的优化后代码示例：

import { parse, visit } from 'graphql'
// 预加载的类型映射，提前从introspection结果生成
const CONNECTION_TYPES = new Set(['UserConnection', 'PostConnection'])
const OBJECT_TYPES = new Set(['User', 'Post', 'Comment'])
const costCache = new LRUCache({ max: 500 }) // 用LRU缓存控制内存占用

function calcQueryCost(query, variables = {}) {
  // 生成缓存key，仅纳入影响代价的变量
  const cacheKey = JSON.stringify([query, variables.first, variables.last, variables.deep])
  if (costCache.get(cacheKey)) return costCache.get(cacheKey)

  let totalCost = 0
  const ast = parse(query)

  visit(ast, {
    Field(node) {
      // 直接匹配预加载的类型集合，不用动态推断
      const fieldType = getFieldReturnType(node.name.value) // 从预加载的schema元数据取
      if (CONNECTION_TYPES.has(fieldType)) {
        totalCost += 2
        return
      }
      if (OBJECT_TYPES.has(fieldType)) {
        totalCost += 1
      }
    }
  })

  costCache.set(cacheKey, totalCost)
  return totalCost
}

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