Add the ability to override Single/List fields resolvers

SeveralBrands · Jan 15, 2023 · 6b21bc5 · 6b21bc5
1 parent 855e18e
commit 6b21bc5
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/lib/SchemaBuilder.js b/lib/SchemaBuilder.js
@@ -184,6 +184,10 @@ class SchemaBuilder {
   }
 
   _rootSingleField(modelData) {
+    if (modelData.modelClass.hasOwnProperty('getSingleNode')) {
+      return modelData.modelClass.getSingleNode
+    }
+
     return {
       type: this._typeForModel(modelData),
       args: modelData.args,
@@ -194,6 +198,10 @@ class SchemaBuilder {
   }
 
   _rootListField(modelData) {
+    if (modelData.modelClass.hasOwnProperty('getListNode')) {
+      return modelData.modelClass.getListNode
+    }
+
     return {
       type: new GraphQLList(this._typeForModel(modelData)),
       args: modelData.args,