pass driver as part of the options to custom functions by default, include new helper function

Author: keski

graphql-server/drivers/arangodb/driver.js

@@ -66,6 +66,9 @@ module.exports = {
     isEndOfList: async function (parent, args, info) {
         return await isEndOfList(parent, args, info);
     },
+    addPossibleEdgeTypes: function(query, schema, type_name, field_name){
+        return addPossibleEdgeTypes(query, schema, type_name, field_name);
+    },
     getEdgeCollectionName: function(type, field){
         return getEdgeCollectionName(type, field);
     },
@@ -1095,3 +1098,24 @@ function conditionalThrow(msg){
         throw msg;
     }
 }
+
+/**
+ * Add all possible edge-collections for the given type and field to query
+ * @param query, schema, type_name, field_name, directionString (Optional)
+ */
+function addPossibleEdgeTypes(query, schema, type_name, field_name, directionString = ""){
+    let type = schema._typeMap[type_name];
+    if(graphql.isInterfaceType(type)){
+        let possible_types = schema.getPossibleTypes(type);
+        for (let i in possible_types) {
+            if (i != 0) {
+                query.push(aql`,`);
+            }
+            let collection = db.collection(getEdgeCollectionName(possible_types[i].name, field_name));
+            query.push(aql`${collection}`);
+        }
+    } else {
+        let collection = db.collection(getEdgeCollectionName(type.name, field_name));
+        query.push(aql`${collection}`);
+    }
+}

graphql-server/server.js

@@ -11,7 +11,7 @@ const typeDefs = gql`${text + textExtend}`;
 async function run(){
     // setup resolvers
     const resolvers = await graphqlResolvers.get({ driver: driver });
-    const customResolvers = await graphqlCustomResolvers.get({});
+    const customResolvers = await graphqlCustomResolvers.get({ driver: driver });
     resolvers.Query = {...resolvers.Query, ...customResolvers.Query};
 
     const server = new ApolloServer({typeDefs, resolvers: resolvers});