Skip to content

指令(Directives)

指令可以附加到字段或片段包含上,并可以以服务器期望的任何方式影响查询的执行(更多信息请参阅此处)。GraphQL 规范提供了几个默认指令:

  • @include(if: Boolean) - 仅在参数为 true 时在结果中包含此字段
  • @skip(if: Boolean) - 如果参数为 true 则跳过此字段
  • @deprecated(reason: String) - 使用消息将字段标记为已弃用

指令是以 @ 字符开头的标识符,可选地后跟一组命名参数,可以出现在 GraphQL 查询和 schema 语言中几乎所有元素之后。

自定义指令

要指示当 Apollo/Mercurius 遇到你的指令时应该发生什么,你可以创建一个转换器函数。该函数使用 mapSchema 函数遍历 schema 中的位置(字段定义、类型定义等)并执行相应的转换。

typescript
import { getDirective, MapperKind, mapSchema } from '@graphql-tools/utils';
import { defaultFieldResolver, GraphQLSchema } from 'graphql';

export function upperDirectiveTransformer(
  schema: GraphQLSchema,
  directiveName: string,
) {
  return mapSchema(schema, {
    [MapperKind.OBJECT_FIELD]: (fieldConfig) => {
      const upperDirective = getDirective(
        schema,
        fieldConfig,
        directiveName,
      )?.[0];

      if (upperDirective) {
        const { resolve = defaultFieldResolver } = fieldConfig;

        // Replace the original resolver with a function that *first* calls
        // the original resolver, then converts its result to upper case
        fieldConfig.resolve = async function (source, args, context, info) {
          const result = await resolve(source, args, context, info);
          if (typeof result === 'string') {
            return result.toUpperCase();
          }
          return result;
        };
        return fieldConfig;
      }
    },
  });
}

现在,使用 transformSchema 函数在 GraphQLModule#forRoot 方法中应用 upperDirectiveTransformer 转换函数:

typescript
GraphQLModule.forRoot({
  // ...
  transformSchema: (schema) => upperDirectiveTransformer(schema, 'upper'),
});

注册后,@upper 指令就可以在我们的 schema 中使用了。但是,应用指令的方式会因你使用的方法(代码优先或 schema 优先)而异。

代码优先

在代码优先方法中,使用 @Directive() 装饰器来应用指令。

typescript
@Directive('@upper')
@Field()
title: string;

提示

@Directive() 装饰器从 @nestjs/graphql 包导出。

指令可以应用于字段、字段解析器、输入和对象类型,以及查询、变更和订阅。以下是在查询处理器级别应用指令的示例:

typescript
@Directive('@deprecated(reason: "This query will be removed in the next version")')
@Query(() => Author, { name: 'author' })
async getAuthor(@Args({ name: 'id', type: () => Int }) id: number) {
  return this.authorsService.findOneById(id);
}

警告

通过 @Directive() 装饰器应用的指令不会反映在生成的 schema 定义文件中。

最后,确保在 GraphQLModule 中声明指令,如下所示:

typescript
GraphQLModule.forRoot({
  // ...,
  transformSchema: schema => upperDirectiveTransformer(schema, 'upper'),
  buildSchemaOptions: {
    directives: [
      new GraphQLDirective({
        name: 'upper',
        locations: [DirectiveLocation.FIELD_DEFINITION],
      }),
    ],
  },
}),

提示

GraphQLDirectiveDirectiveLocation 都从 graphql 包导出。

Schema 优先

在 schema 优先方法中,直接在 SDL 中应用指令。

graphql
directive @upper on FIELD_DEFINITION

type Post {
  id: Int!
  title: String! @upper
  votes: Int
}

基于 NestJS 官方文档翻译