指令访问者模式:用 AST 遍历实现自定义校验与转换
本教程将展示如何自定义 GraphQL 指令(@validate、@transform),并通过遍历 Schema AST 或操作文档 AST 实现运行时校验和数据预处理。 · 难度:入门 · +10XP
指令访问者模式:用 AST 遍历实现自定义校验与转换
大多数指令教程停留在简单参数传递,但通过 AST 遍历你可以实现强大的功能:在查询执行前修改选择集、注入额外字段、甚至重写查询。本课程将演示构建一个 @validate 指令,自动根据参数的 JSON Schema 校验输入,以及 @transform 指令将服务器返回的日期字符串自动转为时间戳。你将熟悉 graphql 库中的 visit 函数和 SchemaDirectiveVisitor 类。
class ValidateDirective extends SchemaDirectiveVisitor {
visitFieldDefinition(field) {
const originalResolve = field.resolve;
field.resolve = async (source, args, context, info) => {
const schema = this.args.schema;
if (schema) {
Object.keys(schema).forEach(key => {
if (typeof args[key] !== schema[key].type) {
throw new Error(Validation failed for ${key});
}
});
}
return originalResolve.call(this, source, args, context, info);
};
}
}