Fixed #3 - graphql non-null links now work

cult-of-coders · Jul 31, 2018 · 4054131 · 4054131
1 parent dec6e41
commit 4054131
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/LinkDirective.js b/LinkDirective.js
@@ -26,11 +26,15 @@ export default class LinkDirective extends SchemaDirectiveVisitor {
       referencedType = field.type;
     }
 
-    if (!(referencedType instanceof GraphQLObjectType)) {
-      throw new Meteor.Error(
-        'invalid-type',
-        `You are trying to attach a link on a invalid type. @link directive only works with GraphQLObjectType `
-      );
+    if (referencedType instanceof GraphQLNonNull) {
+      referencedType = referencedType.ofType;
+    } else {
+      if (!(referencedType instanceof GraphQLObjectType)) {
+        throw new Meteor.Error(
+          'invalid-type',
+          `You are trying to attach a link on a invalid type. @link directive only works with GraphQLObjectType `
+        );
+      }
     }
 
     let referencedCollectionName = referencedType._mongoCollectionName;