Pass CacheMode from /cubesql to backend

Author: KSDaemon

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

@@ -425,7 +425,7 @@ class ApiGateway {
         try {
           await this.assertApiScope('data', req.context?.securityContext);
 
-          await this.sqlServer.execSql(req.body.query, res, req.context?.securityContext);
+          await this.sqlServer.execSql(req.body.query, res, req.context?.securityContext, req.body.cache);
         } catch (e: any) {
           this.handleError({
             e,

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

@@ -8,6 +8,7 @@ import {
   Request as NativeRequest,
   LoadRequestMeta,
   Sql4SqlResponse,
+  CacheMode,
 } from '@cubejs-backend/native';
 import type { ShutdownMode } from '@cubejs-backend/native';
 import { displayCLIWarning, getEnv } from '@cubejs-backend/shared';
@@ -65,8 +66,8 @@ export class SQLServer {
     throw new Error('Native api gateway is not enabled');
   }
 
-  public async execSql(sqlQuery: string, stream: any, securityContext?: any) {
-    await execSql(this.sqlInterfaceInstance!, sqlQuery, stream, securityContext);
+  public async execSql(sqlQuery: string, stream: any, securityContext?: any, cacheMode?: CacheMode) {
+    await execSql(this.sqlInterfaceInstance!, sqlQuery, stream, securityContext, cacheMode);
   }
 
   public async sql4sql(sqlQuery: string, disablePostProcessing: boolean, securityContext?: unknown): Promise<Sql4SqlResponse> {

packages/cubejs-backend-native/js/index.ts

@@ -429,16 +429,18 @@ export const registerInterface = async (options: SQLInterfaceOptions): Promise<S
 
 export type ShutdownMode = 'fast' | 'semifast' | 'smart';
 
+export type CacheMode = 'stale-if-slow' | 'stale-while-revalidate' | 'must-revalidate' | 'no-cache';
+
 export const shutdownInterface = async (instance: SqlInterfaceInstance, shutdownMode: ShutdownMode): Promise<void> => {
   const native = loadNative();
 
   await native.shutdownInterface(instance, shutdownMode);
 };
 
-export const execSql = async (instance: SqlInterfaceInstance, sqlQuery: string, stream: any, securityContext?: any): Promise<void> => {
+export const execSql = async (instance: SqlInterfaceInstance, sqlQuery: string, stream: any, securityContext?: any, cacheMode: CacheMode = 'stale-if-slow'): Promise<void> => {
   const native = loadNative();
 
-  await native.execSql(instance, sqlQuery, stream, securityContext ? JSON.stringify(securityContext) : null);
+  await native.execSql(instance, sqlQuery, stream, securityContext ? JSON.stringify(securityContext) : null, cacheMode);
 };
 
 // TODO parse result from native code

packages/cubejs-backend-native/src/node_export.rs

@@ -226,6 +226,7 @@ async fn handle_sql_query(
     channel: Arc<Channel>,
     stream_methods: WritableStreamMethods,
     sql_query: &str,
+    cache_mode: &str,
 ) -> Result<(), CubeError> {
     let span_id = Some(Arc::new(SpanId::new(
         Uuid::new_v4().to_string(),
@@ -427,6 +428,8 @@ fn exec_sql(mut cx: FunctionContext) -> JsResult<JsValue> {
         Err(_) => None,
     };
 
+    let cache_mode = cx.argument::<JsString>(4)?.value(&mut cx);
+
     let js_stream_on_fn = Arc::new(
         node_stream
             .get::<JsFunction, _, _>(&mut cx, "on")?
@@ -474,6 +477,7 @@ fn exec_sql(mut cx: FunctionContext) -> JsResult<JsValue> {
             channel.clone(),
             stream_methods,
             &sql_query,
+            &cache_mode,
         )
         .await;