feat(): Complexity options on Query and Mutation

Signed-off-by: William Sedlacek <wsedlacekc@gmail.com>
nestjs · Aug 12, 2020 · 1c3a64e · 1c3a64e
1 parent a9031d0
commit 1c3a64e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/lib/decorators/mutation.decorator.ts b/lib/decorators/mutation.decorator.ts
@@ -2,6 +2,7 @@ import { Type } from '@nestjs/common';
 import { isString } from '@nestjs/common/utils/shared.utils';
 import 'reflect-metadata';
 import { Resolver } from '../enums/resolver.enum';
+import { Complexity } from '../interfaces';
 import { BaseTypeOptions } from '../interfaces/base-type-options.interface';
 import { ReturnTypeFunc } from '../interfaces/return-type-func.interface';
 import { UndefinedReturnTypeError } from '../schema-builder/errors/undefined-return-type.error';
@@ -27,6 +28,10 @@ export interface MutationOptions extends BaseTypeOptions {
    * Mutation deprecation reason (if deprecated).
    */
   deprecationReason?: string;
+  /**
+   * Mutation complexity options.
+   */
+  complexity?: Complexity;
 }
 
 /**
@@ -78,6 +83,7 @@ export function Mutation(
         returnTypeOptions: typeOptions,
         description: options.description,
         deprecationReason: options.deprecationReason,
+        complexity: options.complexity,
       };
       TypeMetadataStorage.addMutationMetadata(metadata);
     });

diff --git a/lib/decorators/query.decorator.ts b/lib/decorators/query.decorator.ts
@@ -2,6 +2,7 @@ import { Type } from '@nestjs/common';
 import { isString } from '@nestjs/common/utils/shared.utils';
 import 'reflect-metadata';
 import { Resolver } from '../enums/resolver.enum';
+import { Complexity } from '../interfaces';
 import { BaseTypeOptions } from '../interfaces/base-type-options.interface';
 import { ReturnTypeFunc } from '../interfaces/return-type-func.interface';
 import { UndefinedReturnTypeError } from '../schema-builder/errors/undefined-return-type.error';
@@ -27,6 +28,10 @@ export interface QueryOptions extends BaseTypeOptions {
    * Query deprecation reason (if deprecated).
    */
   deprecationReason?: string;
+  /**
+   * Query complexity options.
+   */
+  complexity?: Complexity;
 }
 
 /**
@@ -78,6 +83,7 @@ export function Query(
         returnTypeOptions: typeOptions,
         description: options.description,
         deprecationReason: options.deprecationReason,
+        complexity: options.complexity,
       };
       TypeMetadataStorage.addQueryMetadata(metadata);
     });