mirror of
https://github.com/twentyhq/twenty.git
synced 2024-12-12 11:20:40 +03:00
11a7db5672
In this PR: - Follow up on #5170 as we did not take into account not logged in users - only apply throttler on root fields to avoid performance overhead
491 lines
21 KiB
Diff
491 lines
21 KiB
Diff
diff --git a/dist/cjs/index.js b/dist/cjs/index.js
|
|
index 16843949d8589a299d8195b0a349ac4dac0bacbf..21e7fe2bbcba36b04a274be9d2219fd38790b508 100644
|
|
--- a/dist/cjs/index.js
|
|
+++ b/dist/cjs/index.js
|
|
@@ -3,10 +3,14 @@ Object.defineProperty(exports, "__esModule", { value: true });
|
|
exports.YogaDriver = exports.AbstractYogaDriver = void 0;
|
|
const tslib_1 = require("tslib");
|
|
const graphql_1 = require("graphql");
|
|
+const schema_1 = require("@graphql-tools/schema");
|
|
const graphql_yoga_1 = require("graphql-yoga");
|
|
const common_1 = require("@nestjs/common");
|
|
const graphql_2 = require("@nestjs/graphql");
|
|
class AbstractYogaDriver extends graphql_2.AbstractGraphQLDriver {
|
|
+
|
|
+ schemaCache = new Map();
|
|
+
|
|
async start(options) {
|
|
const platformName = this.httpAdapterHost.httpAdapter.getType();
|
|
options = {
|
|
@@ -27,7 +31,7 @@ class AbstractYogaDriver extends graphql_2.AbstractGraphQLDriver {
|
|
async stop() {
|
|
// noop
|
|
}
|
|
- registerExpress(options, { preStartHook } = {}) {
|
|
+ registerExpress({ conditionalSchema, ...options }, { preStartHook } = {}) {
|
|
const app = this.httpAdapterHost.httpAdapter.getInstance();
|
|
preStartHook?.(app);
|
|
// nest's logger doesnt have the info method
|
|
@@ -42,6 +46,39 @@ class AbstractYogaDriver extends graphql_2.AbstractGraphQLDriver {
|
|
}
|
|
const yoga = (0, graphql_yoga_1.createYoga)({
|
|
...options,
|
|
+ schema: async (request) => {
|
|
+ const workspaceId = request.req.workspace?.id ?? 'anonymous'
|
|
+ const workspaceCacheVersion = request.req.cacheVersion ?? '0'
|
|
+ const url = request.req.baseUrl
|
|
+
|
|
+ const cacheKey = `${workspaceId}-${workspaceCacheVersion}-${url}`
|
|
+
|
|
+ if(this.schemaCache.has(cacheKey)) {
|
|
+ return this.schemaCache.get(cacheKey)
|
|
+ }
|
|
+
|
|
+ const schemas = [];
|
|
+
|
|
+ if (options.schema) {
|
|
+ schemas.push(options.schema);
|
|
+ }
|
|
+
|
|
+ if (conditionalSchema) {
|
|
+ const conditionalSchemaResult = typeof conditionalSchema === 'function' ? await conditionalSchema(request) : await conditionalSchema;
|
|
+ if (conditionalSchemaResult) {
|
|
+ schemas.push(conditionalSchemaResult);
|
|
+ }
|
|
+ }
|
|
+
|
|
+
|
|
+ const mergedSchemas = (0, schema_1.mergeSchemas)({
|
|
+ schemas,
|
|
+ });
|
|
+
|
|
+ this.schemaCache.set(cacheKey, mergedSchemas)
|
|
+
|
|
+ return mergedSchemas;
|
|
+ },
|
|
graphqlEndpoint: options.path,
|
|
// disable logging by default
|
|
// however, if `true` use nest logger
|
|
@@ -54,11 +91,44 @@ class AbstractYogaDriver extends graphql_2.AbstractGraphQLDriver {
|
|
this.yoga = yoga;
|
|
app.use(yoga.graphqlEndpoint, (req, res) => yoga(req, res, { req, res }));
|
|
}
|
|
- registerFastify(options, { preStartHook } = {}) {
|
|
+ registerFastify({ conditionalSchema, ...options }, { preStartHook } = {}) {
|
|
const app = this.httpAdapterHost.httpAdapter.getInstance();
|
|
preStartHook?.(app);
|
|
const yoga = (0, graphql_yoga_1.createYoga)({
|
|
...options,
|
|
+ schema: async (request) => {
|
|
+ const workspaceId = request.req.workspace?.id ?? 'anonymous'
|
|
+ const workspaceCacheVersion = request.req.cacheVersion ?? '0'
|
|
+ const url = request.req.baseUrl
|
|
+
|
|
+ const cacheKey = `${workspaceId}-${workspaceCacheVersion}-${url}`
|
|
+
|
|
+ if(this.schemaCache.has(cacheKey)) {
|
|
+ return this.schemaCache.get(cacheKey)
|
|
+ }
|
|
+
|
|
+ const schemas = [];
|
|
+
|
|
+ if (options.schema) {
|
|
+ schemas.push(options.schema);
|
|
+ }
|
|
+
|
|
+ if (conditionalSchema) {
|
|
+ const conditionalSchemaResult = typeof conditionalSchema === 'function' ? await conditionalSchema(request) : await conditionalSchema;
|
|
+ if (conditionalSchemaResult) {
|
|
+ schemas.push(conditionalSchemaResult);
|
|
+ }
|
|
+ }
|
|
+
|
|
+
|
|
+ const mergedSchemas = (0, schema_1.mergeSchemas)({
|
|
+ schemas,
|
|
+ });
|
|
+
|
|
+ this.schemaCache.set(cacheKey, mergedSchemas)
|
|
+
|
|
+ return mergedSchemas;
|
|
+ },
|
|
graphqlEndpoint: options.path,
|
|
// disable logging by default
|
|
// however, if `true` use fastify logger
|
|
diff --git a/dist/esm/index.js b/dist/esm/index.js
|
|
index 7068c519320b379917c46763cd280b1cdd3e48f0..418e1030373fc1e0fb85a932ac8da9b39f580570 100644
|
|
--- a/dist/esm/index.js
|
|
+++ b/dist/esm/index.js
|
|
@@ -2,8 +2,12 @@ import { __decorate } from "tslib";
|
|
import { printSchema } from 'graphql';
|
|
import { createYoga, filter, pipe } from 'graphql-yoga';
|
|
import { Injectable, Logger } from '@nestjs/common';
|
|
+import { mergeSchemas } from '@graphql-tools/schema';
|
|
import { AbstractGraphQLDriver, GqlSubscriptionService, } from '@nestjs/graphql';
|
|
export class AbstractYogaDriver extends AbstractGraphQLDriver {
|
|
+
|
|
+ schemaCache = new Map();
|
|
+
|
|
async start(options) {
|
|
const platformName = this.httpAdapterHost.httpAdapter.getType();
|
|
options = {
|
|
@@ -24,7 +28,7 @@ export class AbstractYogaDriver extends AbstractGraphQLDriver {
|
|
async stop() {
|
|
// noop
|
|
}
|
|
- registerExpress(options, { preStartHook } = {}) {
|
|
+ registerExpress({ conditionalSchema, ...options }, { preStartHook } = {}) {
|
|
const app = this.httpAdapterHost.httpAdapter.getInstance();
|
|
preStartHook?.(app);
|
|
// nest's logger doesnt have the info method
|
|
@@ -39,6 +43,39 @@ export class AbstractYogaDriver extends AbstractGraphQLDriver {
|
|
}
|
|
const yoga = createYoga({
|
|
...options,
|
|
+ schema: async (request) => {
|
|
+ const workspaceId = request.req.workspace?.id ?? 'anonymous'
|
|
+ const workspaceCacheVersion = request.req.cacheVersion ?? '0'
|
|
+ const url = request.req.baseUrl
|
|
+
|
|
+ const cacheKey = `${workspaceId}-${workspaceCacheVersion}-${url}`
|
|
+
|
|
+ if (this.schemaCache.has(cacheKey)) {
|
|
+ return this.schemaCache.get(cacheKey)
|
|
+ }
|
|
+
|
|
+ const schemas = [];
|
|
+
|
|
+ if (options.schema) {
|
|
+ schemas.push(options.schema);
|
|
+ }
|
|
+
|
|
+ if (conditionalSchema) {
|
|
+ const conditionalSchemaResult = typeof conditionalSchema === 'function' ? await conditionalSchema(request) : await conditionalSchema;
|
|
+
|
|
+ if (conditionalSchemaResult) {
|
|
+ schemas.push(conditionalSchemaResult);
|
|
+ }
|
|
+ }
|
|
+
|
|
+ const mergedSchemas = mergeSchemas({
|
|
+ schemas,
|
|
+ });
|
|
+
|
|
+ this.schemaCache.set(cacheKey, mergedSchemas)
|
|
+
|
|
+ return mergedSchemas;
|
|
+ },
|
|
graphqlEndpoint: options.path,
|
|
// disable logging by default
|
|
// however, if `true` use nest logger
|
|
@@ -51,11 +88,44 @@ export class AbstractYogaDriver extends AbstractGraphQLDriver {
|
|
this.yoga = yoga;
|
|
app.use(yoga.graphqlEndpoint, (req, res) => yoga(req, res, { req, res }));
|
|
}
|
|
- registerFastify(options, { preStartHook } = {}) {
|
|
+ registerFastify({ conditionalSchema, ...options }, { preStartHook } = {}) {
|
|
const app = this.httpAdapterHost.httpAdapter.getInstance();
|
|
preStartHook?.(app);
|
|
const yoga = createYoga({
|
|
...options,
|
|
+ schema: async (request) => {
|
|
+ const workspaceId = request.req.workspace?.id ?? 'anonymous'
|
|
+ const workspaceCacheVersion = request.req.cacheVersion ?? '0'
|
|
+ const url = request.req.baseUrl
|
|
+
|
|
+ const cacheKey = `${workspaceId}-${workspaceCacheVersion}-${url}`
|
|
+
|
|
+ if (this.schemaCache.has(cacheKey)) {
|
|
+ return this.schemaCache.get(cacheKey)
|
|
+ }
|
|
+
|
|
+ const schemas = [];
|
|
+
|
|
+ if (options.schema) {
|
|
+ schemas.push(options.schema);
|
|
+ }
|
|
+
|
|
+ if (conditionalSchema) {
|
|
+ const conditionalSchemaResult = typeof conditionalSchema === 'function' ? await conditionalSchema(request) : await conditionalSchema;
|
|
+
|
|
+ if (conditionalSchemaResult) {
|
|
+ schemas.push(conditionalSchemaResult);
|
|
+ }
|
|
+ }
|
|
+
|
|
+ const mergedSchemas = mergeSchemas({
|
|
+ schemas,
|
|
+ });
|
|
+
|
|
+ this.schemaCache.set(cacheKey, mergedSchemas)
|
|
+
|
|
+ return mergedSchemas;
|
|
+ },
|
|
graphqlEndpoint: options.path,
|
|
// disable logging by default
|
|
// however, if `true` use fastify logger
|
|
diff --git a/dist/typings/index.d.cts b/dist/typings/index.d.cts
|
|
index 2c6a9656193392680121487c7147db459d6b69ab..2f2b59f0e311f0526a7cfdad97372229301aabd7 100644
|
|
--- a/dist/typings/index.d.cts
|
|
+++ b/dist/typings/index.d.cts
|
|
@@ -1,7 +1,8 @@
|
|
import type { Express, Request as ExpressRequest, Response as ExpressResponse } from 'express';
|
|
import type { FastifyInstance, FastifyReply, FastifyRequest } from 'fastify';
|
|
-import { YogaServerInstance, YogaServerOptions } from 'graphql-yoga';
|
|
+import { YogaServerInstance, YogaServerOptions, GraphQLSchemaWithContext, PromiseOrValue, YogaInitialContext } from 'graphql-yoga';
|
|
import { AbstractGraphQLDriver, GqlModuleOptions, SubscriptionConfig } from '@nestjs/graphql';
|
|
+export type YogaSchemaDefinition<TContext> = PromiseOrValue<GraphQLSchemaWithContext<TContext>> | ((context: TContext & YogaInitialContext) => PromiseOrValue<GraphQLSchemaWithContext<TContext>>);
|
|
export type YogaDriverPlatform = 'express' | 'fastify';
|
|
export type YogaDriverServerContext<Platform extends YogaDriverPlatform> = Platform extends 'fastify' ? {
|
|
req: FastifyRequest;
|
|
@@ -10,7 +11,10 @@ export type YogaDriverServerContext<Platform extends YogaDriverPlatform> = Platf
|
|
req: ExpressRequest;
|
|
res: ExpressResponse;
|
|
};
|
|
-export type YogaDriverServerOptions<Platform extends YogaDriverPlatform> = Omit<YogaServerOptions<YogaDriverServerContext<Platform>, never>, 'context' | 'schema'>;
|
|
+
|
|
+export type YogaDriverServerOptions<Platform extends YogaDriverPlatform> = Omit<YogaServerOptions<YogaDriverServerContext<Platform>, never>, 'context' | 'schema'> & {
|
|
+ conditionalSchema?: YogaSchemaDefinition<YogaDriverServerContext<Platform>> | undefined;
|
|
+};
|
|
export type YogaDriverServerInstance<Platform extends YogaDriverPlatform> = YogaServerInstance<YogaDriverServerContext<Platform>, never>;
|
|
export type YogaDriverConfig<Platform extends YogaDriverPlatform = 'express'> = GqlModuleOptions & YogaDriverServerOptions<Platform> & {
|
|
/**
|
|
@@ -26,10 +30,10 @@ export declare abstract class AbstractYogaDriver<Platform extends YogaDriverPlat
|
|
protected yoga: YogaDriverServerInstance<Platform>;
|
|
start(options: YogaDriverConfig<Platform>): Promise<void>;
|
|
stop(): Promise<void>;
|
|
- protected registerExpress(options: YogaDriverConfig<'express'>, { preStartHook }?: {
|
|
+ protected registerExpress({ conditionalSchema, ...options }: YogaDriverConfig<'express'>, { preStartHook }?: {
|
|
preStartHook?: (app: Express) => void;
|
|
}): void;
|
|
- protected registerFastify(options: YogaDriverConfig<'fastify'>, { preStartHook }?: {
|
|
+ protected registerFastify({ conditionalSchema, ...options }: YogaDriverConfig<'fastify'>, { preStartHook }?: {
|
|
preStartHook?: (app: FastifyInstance) => void;
|
|
}): void;
|
|
subscriptionWithFilter<TPayload, TVariables, TContext>(instanceRef: unknown, filterFn: (payload: TPayload, variables: TVariables, context: TContext) => boolean | Promise<boolean>, createSubscribeContext: Function): (args_0: TPayload, args_1: TVariables, args_2: TContext) => Promise<import("graphql-yoga").Repeater<TPayload, void, unknown>>;
|
|
diff --git a/dist/typings/index.d.ts b/dist/typings/index.d.ts
|
|
index 2c6a9656193392680121487c7147db459d6b69ab..fd86daccf3e5a93ff44b568c9793c16d761f4f53 100644
|
|
--- a/dist/typings/index.d.ts
|
|
+++ b/dist/typings/index.d.ts
|
|
@@ -1,7 +1,8 @@
|
|
import type { Express, Request as ExpressRequest, Response as ExpressResponse } from 'express';
|
|
import type { FastifyInstance, FastifyReply, FastifyRequest } from 'fastify';
|
|
-import { YogaServerInstance, YogaServerOptions } from 'graphql-yoga';
|
|
+import { YogaServerInstance, YogaServerOptions, GraphQLSchemaWithContext, PromiseOrValue, YogaInitialContext } from 'graphql-yoga';
|
|
import { AbstractGraphQLDriver, GqlModuleOptions, SubscriptionConfig } from '@nestjs/graphql';
|
|
+export type YogaSchemaDefinition<TContext> = PromiseOrValue<GraphQLSchemaWithContext<TContext>> | ((context: TContext & YogaInitialContext) => PromiseOrValue<GraphQLSchemaWithContext<TContext>>);
|
|
export type YogaDriverPlatform = 'express' | 'fastify';
|
|
export type YogaDriverServerContext<Platform extends YogaDriverPlatform> = Platform extends 'fastify' ? {
|
|
req: FastifyRequest;
|
|
@@ -10,7 +11,9 @@ export type YogaDriverServerContext<Platform extends YogaDriverPlatform> = Platf
|
|
req: ExpressRequest;
|
|
res: ExpressResponse;
|
|
};
|
|
-export type YogaDriverServerOptions<Platform extends YogaDriverPlatform> = Omit<YogaServerOptions<YogaDriverServerContext<Platform>, never>, 'context' | 'schema'>;
|
|
+export type YogaDriverServerOptions<Platform extends YogaDriverPlatform> = Omit<YogaServerOptions<YogaDriverServerContext<Platform>, never>, 'context' | 'schema'> & {
|
|
+ conditionalSchema?: YogaSchemaDefinition<YogaDriverServerContext<Platform>> | undefined;
|
|
+};
|
|
export type YogaDriverServerInstance<Platform extends YogaDriverPlatform> = YogaServerInstance<YogaDriverServerContext<Platform>, never>;
|
|
export type YogaDriverConfig<Platform extends YogaDriverPlatform = 'express'> = GqlModuleOptions & YogaDriverServerOptions<Platform> & {
|
|
/**
|
|
@@ -26,10 +29,10 @@ export declare abstract class AbstractYogaDriver<Platform extends YogaDriverPlat
|
|
protected yoga: YogaDriverServerInstance<Platform>;
|
|
start(options: YogaDriverConfig<Platform>): Promise<void>;
|
|
stop(): Promise<void>;
|
|
- protected registerExpress(options: YogaDriverConfig<'express'>, { preStartHook }?: {
|
|
+ protected registerExpress({ conditionalSchema, ...options }: YogaDriverConfig<'express'>, { preStartHook }?: {
|
|
preStartHook?: (app: Express) => void;
|
|
}): void;
|
|
- protected registerFastify(options: YogaDriverConfig<'fastify'>, { preStartHook }?: {
|
|
+ protected registerFastify({ conditionalSchema, ...options }: YogaDriverConfig<'fastify'>, { preStartHook }?: {
|
|
preStartHook?: (app: FastifyInstance) => void;
|
|
}): void;
|
|
subscriptionWithFilter<TPayload, TVariables, TContext>(instanceRef: unknown, filterFn: (payload: TPayload, variables: TVariables, context: TContext) => boolean | Promise<boolean>, createSubscribeContext: Function): (args_0: TPayload, args_1: TVariables, args_2: TContext) => Promise<import("graphql-yoga").Repeater<TPayload, void, unknown>>;
|
|
diff --git a/src/index.ts b/src/index.ts
|
|
index ce142f61ede52499485b19d8af057f4cb828d0f7..5888d31cae1b7aca57ed0819209812ac941edabb 100644
|
|
--- a/src/index.ts
|
|
+++ b/src/index.ts
|
|
@@ -1,9 +1,10 @@
|
|
import type { Express, Request as ExpressRequest, Response as ExpressResponse } from 'express';
|
|
import type { FastifyInstance, FastifyReply, FastifyRequest } from 'fastify';
|
|
-import { printSchema } from 'graphql';
|
|
-import { createYoga, filter, pipe, YogaServerInstance, YogaServerOptions } from 'graphql-yoga';
|
|
+import { printSchema, GraphQLSchema } from 'graphql';
|
|
+import { createYoga, filter, pipe, YogaServerInstance, YogaServerOptions, GraphQLSchemaWithContext, PromiseOrValue, YogaInitialContext } from 'graphql-yoga';
|
|
import type { ExecutionParams } from 'subscriptions-transport-ws';
|
|
import { Injectable, Logger } from '@nestjs/common';
|
|
+import { mergeSchemas } from '@graphql-tools/schema';
|
|
import {
|
|
AbstractGraphQLDriver,
|
|
GqlModuleOptions,
|
|
@@ -11,23 +12,31 @@ import {
|
|
SubscriptionConfig,
|
|
} from '@nestjs/graphql';
|
|
|
|
+export type YogaSchemaDefinition<TContext> =
|
|
+ | PromiseOrValue<GraphQLSchemaWithContext<TContext>>
|
|
+ | ((
|
|
+ context: TContext & YogaInitialContext,
|
|
+ ) => PromiseOrValue<GraphQLSchemaWithContext<TContext>>);
|
|
+
|
|
export type YogaDriverPlatform = 'express' | 'fastify';
|
|
|
|
export type YogaDriverServerContext<Platform extends YogaDriverPlatform> =
|
|
Platform extends 'fastify'
|
|
- ? {
|
|
- req: FastifyRequest;
|
|
- reply: FastifyReply;
|
|
- }
|
|
- : {
|
|
- req: ExpressRequest;
|
|
- res: ExpressResponse;
|
|
- };
|
|
+ ? {
|
|
+ req: FastifyRequest;
|
|
+ reply: FastifyReply;
|
|
+ }
|
|
+ : {
|
|
+ req: ExpressRequest;
|
|
+ res: ExpressResponse;
|
|
+ };
|
|
|
|
export type YogaDriverServerOptions<Platform extends YogaDriverPlatform> = Omit<
|
|
YogaServerOptions<YogaDriverServerContext<Platform>, never>,
|
|
'context' | 'schema'
|
|
->;
|
|
+> & {
|
|
+ conditionalSchema?: YogaSchemaDefinition<YogaDriverServerContext<Platform>> | undefined;
|
|
+};
|
|
|
|
export type YogaDriverServerInstance<Platform extends YogaDriverPlatform> = YogaServerInstance<
|
|
YogaDriverServerContext<Platform>,
|
|
@@ -53,6 +62,8 @@ export type YogaDriverSubscriptionConfig = {
|
|
export abstract class AbstractYogaDriver<
|
|
Platform extends YogaDriverPlatform,
|
|
> extends AbstractGraphQLDriver<YogaDriverConfig<Platform>> {
|
|
+ schemaCache = new Map();
|
|
+
|
|
protected yoga!: YogaDriverServerInstance<Platform>;
|
|
|
|
public async start(options: YogaDriverConfig<Platform>) {
|
|
@@ -78,7 +89,7 @@ export abstract class AbstractYogaDriver<
|
|
}
|
|
|
|
protected registerExpress(
|
|
- options: YogaDriverConfig<'express'>,
|
|
+ { conditionalSchema, ...options }: YogaDriverConfig<'express'>,
|
|
{ preStartHook }: { preStartHook?: (app: Express) => void } = {},
|
|
) {
|
|
const app: Express = this.httpAdapterHost.httpAdapter.getInstance();
|
|
@@ -98,6 +109,39 @@ export abstract class AbstractYogaDriver<
|
|
|
|
const yoga = createYoga<YogaDriverServerContext<'express'>>({
|
|
...options,
|
|
+ schema: async request => {
|
|
+ const workspaceId = request.req.workspace.id
|
|
+ const workspaceCacheVersion = request.req.cacheVersion
|
|
+ const url = request.req.baseUrl
|
|
+
|
|
+ const cacheKey = `${workspaceId}-${workspaceCacheVersion}-${url}`
|
|
+
|
|
+ if (this.schemaCache.has(cacheKey)) {
|
|
+ return this.schemaCache.get(cacheKey)
|
|
+ }
|
|
+
|
|
+ const schemas: GraphQLSchema[] = [];
|
|
+
|
|
+ if (options.schema) {
|
|
+ schemas.push(options.schema);
|
|
+ }
|
|
+
|
|
+ if (conditionalSchema) {
|
|
+ const conditionalSchemaResult = typeof conditionalSchema === 'function' ? await conditionalSchema(request) : await conditionalSchema;
|
|
+
|
|
+ if (conditionalSchemaResult) {
|
|
+ schemas.push(conditionalSchemaResult);
|
|
+ }
|
|
+ }
|
|
+
|
|
+ const mergedSchemas = mergeSchemas({
|
|
+ schemas,
|
|
+ });
|
|
+
|
|
+ this.schemaCache.set(cacheKey, mergedSchemas)
|
|
+
|
|
+ return mergedSchemas;
|
|
+ },
|
|
graphqlEndpoint: options.path,
|
|
// disable logging by default
|
|
// however, if `true` use nest logger
|
|
@@ -105,8 +149,8 @@ export abstract class AbstractYogaDriver<
|
|
options.logging == null
|
|
? false
|
|
: options.logging
|
|
- ? new LoggerWithInfo('YogaDriver')
|
|
- : options.logging,
|
|
+ ? new LoggerWithInfo('YogaDriver')
|
|
+ : options.logging,
|
|
});
|
|
|
|
this.yoga = yoga as YogaDriverServerInstance<Platform>;
|
|
@@ -115,7 +159,7 @@ export abstract class AbstractYogaDriver<
|
|
}
|
|
|
|
protected registerFastify(
|
|
- options: YogaDriverConfig<'fastify'>,
|
|
+ { conditionalSchema, ...options }: YogaDriverConfig<'fastify'>,
|
|
{ preStartHook }: { preStartHook?: (app: FastifyInstance) => void } = {},
|
|
) {
|
|
const app: FastifyInstance = this.httpAdapterHost.httpAdapter.getInstance();
|
|
@@ -124,6 +168,39 @@ export abstract class AbstractYogaDriver<
|
|
|
|
const yoga = createYoga<YogaDriverServerContext<'fastify'>>({
|
|
...options,
|
|
+ schema: async request => {
|
|
+ const workspaceId = request.req.workspace.id
|
|
+ const workspaceCacheVersion = request.req.cacheVersion
|
|
+ const url = request.req.baseUrl
|
|
+
|
|
+ const cacheKey = `${workspaceId}-${workspaceCacheVersion}-${url}`
|
|
+
|
|
+ if (this.schemaCache.has(cacheKey)) {
|
|
+ return this.schemaCache.get(cacheKey)
|
|
+ }
|
|
+
|
|
+ const schemas: GraphQLSchema[] = [];
|
|
+
|
|
+ if (options.schema) {
|
|
+ schemas.push(options.schema);
|
|
+ }
|
|
+
|
|
+ if (conditionalSchema) {
|
|
+ const conditionalSchemaResult = typeof conditionalSchema === 'function' ? await conditionalSchema(request) : await conditionalSchema;
|
|
+
|
|
+ if (conditionalSchemaResult) {
|
|
+ schemas.push(conditionalSchemaResult);
|
|
+ }
|
|
+ }
|
|
+
|
|
+ const mergedSchemas = mergeSchemas({
|
|
+ schemas,
|
|
+ });
|
|
+
|
|
+ this.schemaCache.set(cacheKey, mergedSchemas)
|
|
+
|
|
+ return mergedSchemas;
|
|
+ },
|
|
graphqlEndpoint: options.path,
|
|
// disable logging by default
|
|
// however, if `true` use fastify logger
|
|
@@ -191,8 +268,8 @@ export class YogaDriver<
|
|
const config: SubscriptionConfig =
|
|
options.subscriptions === true
|
|
? {
|
|
- 'graphql-ws': true,
|
|
- }
|
|
+ 'graphql-ws': true,
|
|
+ }
|
|
: options.subscriptions;
|
|
|
|
if (config['graphql-ws']) {
|