基于Apollo GraphQL查询指定字段对象数组的Query与Resolver设计咨询

GraphQL原生支持客户端按需选择返回字段，你提到的两个场景不需要定义多个查询接口，仅需设计1个通用的teamInfo查询即可满足不同字段返回需求。

1. GraphQL Schema 定义（SDL写法） #

# 团队类型定义，涵盖所有业务需要的字段
type Team {
  team: Int! # 对应团队ID
  teamName: String
  teamMemberCount: Int
  slogun: String
}

type Query {
  """
  根据团队ID数组批量查询团队信息
  客户端可按需指定返回字段
  """
  teamInfo(teamIds: [Int!]!): [Team!]!
}

2. TypeORM 实体定义（TypeScript） #

对应MySQL数据库的team表结构：

import { Entity, Column, PrimaryColumn } from 'typeorm';

@Entity('team')
export class TeamEntity {
  @PrimaryColumn()
  team: number; // 团队ID主键

  @Column({ nullable: true })
  teamName: string;

  @Column({ nullable: true })
  teamMemberCount: number;

  @Column({ nullable: true })
  slogun: string;
}

3. Apollo Resolver 实现（TypeScript） #

NestJS + Apollo 版本 #

import { Query, Resolver, Args, ResolveField, Parent } from '@nestjs/graphql';
import { InjectRepository } from '@nestjs/typeorm';
import { Repository } from 'typeorm';
import { TeamEntity } from './team.entity';
import { Team } from './team.type';

@Resolver(() => Team)
export class TeamResolver {
  constructor(
    @InjectRepository(TeamEntity)
    private readonly teamRepository: Repository<TeamEntity>,
  ) {}

  @Query(() => [Team])
  async teamInfo(@Args('teamIds', { type: () => [Number] }) teamIds: number[]) {
    // 参数校验，避免空数组查询全表
    if (!teamIds?.length) return [];
    // 批量查询对应ID的团队基础数据
    return this.teamRepository.findByIds(teamIds);
  }
}

纯Apollo Server 版本 #

const resolvers = {
  Query: {
    teamInfo: async (_parent, { teamIds }, { dataSource }) => {
      if (!teamIds?.length) return [];
      return dataSource.teamRepository.findByIds(teamIds);
    }
  }
}

可选性能优化：如果teamMemberCount是动态关联统计字段而非表中固有字段，可单独编写FieldResolver，只有当客户端请求该字段时才执行统计逻辑，避免不必要的数据库查询：

@ResolveField(() => Number)
async teamMemberCount(@Parent() team: TeamEntity) {
  return this.teamMemberRepository.count({ where: { teamId: team.team } });
}

Resolver不需要额外做字段过滤逻辑，GraphQL执行引擎会自动根据客户端请求的字段集对返回结果做序列化裁剪，仅返回客户端请求的字段。

4. 客户端调用示例 #

场景1调用 #

query GetTeamBaseInfo($teamIds: [Int!]!) {
  teamInfo(teamIds: $teamIds) {
    team
    teamName
    teamMemberCount
  }
}
// 变量传值：{ "teamIds": [1,2] }

场景2调用 #

query GetTeamSlogun($teamIds: [Int!]!) {
  teamInfo(teamIds: $teamIds) {
    team
    slogun
  }
}
// 变量传值：{ "teamIds": [3,4] }

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