pass cacheMode to getSqlResponseInternal

Author: KSDaemon

packages/cubejs-api-gateway/src/gateway.ts

@@ -12,6 +12,7 @@ import {
   getRealType,
   parseUtcIntoLocalDate,
   QueryAlias,
+  CacheMode,
 } from '@cubejs-backend/shared';
 import {
   ResultArrayWrapper,
@@ -317,6 +318,7 @@ class ApiGateway {
         context: req.context,
         res: this.resToResultFn(res),
         queryType: req.query.queryType,
+        cacheMode: req.query.cache,
       });
     }));
 
@@ -326,7 +328,8 @@ class ApiGateway {
         query: req.body.query,
         context: req.context,
         res: this.resToResultFn(res),
-        queryType: req.body.queryType
+        queryType: req.body.queryType,
+        cacheMode: req.body.cache,
       });
     }));
 
@@ -335,7 +338,8 @@ class ApiGateway {
         query: req.query.query,
         context: req.context,
         res: this.resToResultFn(res),
-        queryType: req.query.queryType
+        queryType: req.query.queryType,
+        cacheMode: req.query.cache,
       });
     }));
 
@@ -1642,6 +1646,7 @@ class ApiGateway {
     context: RequestContext,
     normalizedQuery: NormalizedQuery,
     sqlQuery: any,
+    cacheMode: CacheMode = 'stale-if-slow',
   ): Promise<ResultWrapper> {
     const queries = [{
       ...sqlQuery,

packages/cubejs-api-gateway/src/graphql.ts

@@ -653,6 +653,7 @@ export function makeSchema(metaConfig: any): GraphQLSchema {
             apiGateway.load({
               query,
               queryType: QueryType.REGULAR_QUERY,
+              ...(query.cache ? { cacheMode: query.cache } : {}),
               context: req.context,
               res: async (message) => {
                 if (message.error) {

packages/cubejs-api-gateway/src/types/request.ts

@@ -7,6 +7,7 @@
 
 import type { Request as ExpressRequest } from 'express';
 import type { DataResult } from '@cubejs-backend/native';
+import { CacheMode } from '@cubejs-backend/shared';
 import { RequestType, ApiType, ResultType } from './strings';
 import { Query } from './query';
 
@@ -133,6 +134,7 @@ type QueryRequest = BaseRequest & {
   memberExpressions?: boolean;
   disableExternalPreAggregations?: boolean;
   disableLimitEnforcing?: boolean;
+  cacheMode?: CacheMode;
 };
 
 type SqlApiRequest = BaseRequest & {