feat:pass other apollo server options. (#31)

Author: whlsxl

README.md

@@ -67,6 +67,24 @@ exports.graphql = {
   onPreGraphQL: function* (ctx) {},
   // 开发工具 graphiQL 路由前的拦截器，建议用于做权限操作(如只提供开发者使用)
   onPreGraphiQL: function* (ctx) {},
+  // apollo server的透传参数，参考[文档](https://www.apollographql.com/docs/apollo-server/api/apollo-server/#parameters)
+  apolloServerOptions: {
+    rootValue,
+    formatError,
+    formatResponse,
+    mocks,
+    schemaDirectives,
+    introspection,
+    playground,
+    debug,
+    validationRules,
+    tracing,
+    cacheControl,
+    subscriptions,
+    engine,
+    persistedQueries,
+    cors,
+  }
 };
 
 // 添加中间件拦截请求

app/middleware/graphql.js

@@ -45,21 +45,31 @@ module.exports = (_, app) => {
   return async (ctx, next) => {
     /* istanbul ignore else */
     if (ctx.path === graphQLRouter) {
+      const {
+        onPreGraphiQL,
+        onPreGraphQL,
+        apolloServerOptions,
+      } = options;
       if (ctx.request.accepts([ 'json', 'html' ]) === 'html' && graphiql) {
-        if (options.onPreGraphiQL) {
-          await options.onPreGraphiQL(ctx);
+        if (onPreGraphiQL) {
+          await onPreGraphiQL(ctx);
         }
         return graphiqlKoa({
           endpointURL: graphQLRouter,
         })(ctx);
       }
-      if (options.onPreGraphQL) {
-        await options.onPreGraphQL(ctx);
+      if (onPreGraphQL) {
+        await onPreGraphQL(ctx);
       }
-      return graphqlKoa({
-        schema: app.schema,
-        context: ctx,
-      })(ctx);
+      const serverOptions = Object.assign(
+        {},
+        apolloServerOptions,
+        {
+          schema: app.schema,
+          context: ctx,
+        }
+      );
+      return graphqlKoa(serverOptions)(ctx);
     }
     await next();
   };

test/app/graphql-options/graphql.test.js

@@ -0,0 +1,35 @@
+'use strict';
+
+const assert = require('assert');
+const mm = require('egg-mock');
+
+describe('test/app/graphql-options.test.js', () => {
+  let app;
+
+  before(() => {
+    app = mm.app({
+      baseDir: 'apps/graphql-options-app',
+    });
+    return app.ready();
+  });
+
+  after(mm.restore);
+
+  it('should return custom error, use formatError', async () => {
+    const resp = await app.httpRequest()
+      .get('/graphql?query=query+getUser($id:Int){user(id:$id){name}}&variables={"id":1}')
+      .expect(200);
+    assert.equal(resp.body.errors[0].code, 100001);
+  });
+
+  it('should return frameworks, user formatResponse', async () => {
+    const resp = await app.httpRequest()
+      .get('/graphql?query=query+getFramework($id:Int){framework(id:$id){name}}&variables={"id":1}')
+      .expect(200);
+    assert.deepEqual(resp.body.data, {
+      frameworks: {
+        name: 'framework1',
+      },
+    });
+  });
+});

test/fixtures/apps/graphql-options-app/app/extend/application.js

@@ -0,0 +1,5 @@
+'use strict';
+
+module.exports = {
+  changedName: 'frameworks',
+};

test/fixtures/apps/graphql-options-app/app/graphql/directives/directive.js

@@ -0,0 +1,12 @@
+'use strict';
+
+module.exports = {
+  upper(next) {
+    return next().then(str => {
+      if (typeof str === 'string') {
+        return str.toUpperCase();
+      }
+      return str;
+    });
+  },
+};

test/fixtures/apps/graphql-options-app/app/graphql/directives/schema.graphql

@@ -0,0 +1 @@
+directive @upper on FIELD_DEFINITION

test/fixtures/apps/graphql-options-app/app/graphql/group/framework/connector.js

@@ -0,0 +1,30 @@
+'use strict';
+
+const DataLoader = require('dataloader');
+
+class FrameworkConnector {
+  constructor(ctx) {
+    this.ctx = ctx;
+    this.loader = new DataLoader(this.fetch.bind(this));
+  }
+
+  fetch(ids) {
+    return Promise.resolve(ids.map(id => ({
+      id,
+      name: `framework${id}`,
+      projects: [],
+    })));
+  }
+
+  fetchByIds(ids) {
+    return this.loader.loadMany(ids);
+  }
+
+  fetchById(id) {
+    return this.loader.load(id);
+  }
+
+}
+
+module.exports = FrameworkConnector;
+

test/fixtures/apps/graphql-options-app/app/graphql/group/framework/resolver.js

@@ -0,0 +1,9 @@
+'use strict';
+
+module.exports = {
+  Query: {
+    framework(root, { id }, ctx) {
+      return ctx.connector.framework.fetchById(id);
+    },
+  },
+};

test/fixtures/apps/graphql-options-app/app/graphql/group/framework/schema.graphql

@@ -0,0 +1,6 @@
+
+type Framework {
+  id: Int!
+  name: String!
+  projects: [Project]
+}

test/fixtures/apps/graphql-options-app/app/graphql/project/resolver.js

@@ -0,0 +1,12 @@
+'use strict';
+
+module.exports = app => {
+  return {
+    Query: {
+      projects() {
+        console.log(app);
+        return [];
+      },
+    },
+  };
+};