ebc25c8695
## Context @lucasbordeau introduced a new Yoga plugin that allows us to cache our requests (👏), see https://github.com/twentyhq/twenty/pull/5189 I'm simply updating the implementation to allow us to use different cache storage types such as redis Also adding a check so it does not use cache for other operations than ObjectMetadataItems ## Test locally, first call takes 340ms, 2nd takes 30ms with 'redis' and 13ms with 'memory'
62 lines
2.4 KiB
TypeScript
62 lines
2.4 KiB
TypeScript
import { YogaDriverConfig } from '@graphql-yoga/nestjs';
|
|
import GraphQLJSON from 'graphql-type-json';
|
|
|
|
import { EnvironmentService } from 'src/engine/integrations/environment/environment.service';
|
|
import { ExceptionHandlerService } from 'src/engine/integrations/exception-handler/exception-handler.service';
|
|
import { useExceptionHandler } from 'src/engine/integrations/exception-handler/hooks/use-exception-handler.hook';
|
|
import { useThrottler } from 'src/engine/api/graphql/graphql-config/hooks/use-throttler';
|
|
import { MetadataGraphQLApiModule } from 'src/engine/api/graphql/metadata-graphql-api.module';
|
|
import { renderApolloPlayground } from 'src/engine/utils/render-apollo-playground.util';
|
|
import { DataloaderService } from 'src/engine/dataloaders/dataloader.service';
|
|
import { useCachedMetadata } from 'src/engine/api/graphql/graphql-config/hooks/use-cached-metadata';
|
|
import { CacheStorageService } from 'src/engine/integrations/cache-storage/cache-storage.service';
|
|
|
|
export const metadataModuleFactory = async (
|
|
environmentService: EnvironmentService,
|
|
exceptionHandlerService: ExceptionHandlerService,
|
|
dataloaderService: DataloaderService,
|
|
workspaceSchemaCacheStorage: CacheStorageService,
|
|
): Promise<YogaDriverConfig> => {
|
|
const config: YogaDriverConfig = {
|
|
autoSchemaFile: true,
|
|
include: [MetadataGraphQLApiModule],
|
|
renderGraphiQL() {
|
|
return renderApolloPlayground({ path: 'metadata' });
|
|
},
|
|
resolvers: { JSON: GraphQLJSON },
|
|
plugins: [
|
|
useThrottler({
|
|
ttl: environmentService.get('API_RATE_LIMITING_TTL'),
|
|
limit: environmentService.get('API_RATE_LIMITING_LIMIT'),
|
|
identifyFn: (context) => {
|
|
return context.req.user?.id ?? context.req.ip ?? 'anonymous';
|
|
},
|
|
}),
|
|
useExceptionHandler({
|
|
exceptionHandlerService,
|
|
}),
|
|
useCachedMetadata({
|
|
cacheGetter: workspaceSchemaCacheStorage.get.bind(
|
|
workspaceSchemaCacheStorage,
|
|
),
|
|
cacheSetter: workspaceSchemaCacheStorage.set.bind(
|
|
workspaceSchemaCacheStorage,
|
|
),
|
|
operationsToCache: ['ObjectMetadataItems'],
|
|
}),
|
|
],
|
|
path: '/metadata',
|
|
context: () => ({
|
|
loaders: dataloaderService.createLoaders(),
|
|
}),
|
|
};
|
|
|
|
if (environmentService.get('DEBUG_MODE')) {
|
|
config.renderGraphiQL = () => {
|
|
return renderApolloPlayground({ path: 'metadata' });
|
|
};
|
|
}
|
|
|
|
return config;
|
|
};
|