feat(cubejs-server-core): add contextToCubeStoreRouterId config option

Author: bsod90

packages/cubejs-server-core/src/core/optionsValidate.ts

@@ -74,6 +74,7 @@ const schemaOptions = Joi.object().keys({
   contextToAppId: Joi.func(),
   contextToRoles: Joi.func(),
   contextToOrchestratorId: Joi.func(),
+  contextToCubeStoreRouterId: Joi.func(),
   contextToDataSourceId: Joi.func(),
   contextToApiScopes: Joi.func(),
   repositoryFactory: Joi.func(),

packages/cubejs-server-core/src/core/server.ts

@@ -46,6 +46,7 @@ import type {
   DriverContext,
   LoggerFn,
   DriverConfig,
+  ContextToCubeStoreRouterIdFn,
 } from './types';
 import { ContextToOrchestratorIdFn, ContextAcceptanceResult, ContextAcceptanceResultHttp, ContextAcceptanceResultWs, ContextAcceptor } from './types';
 
@@ -115,7 +116,9 @@ export class CubejsServerCore {
 
   protected compilerCache: LRUCache<string, CompilerApi>;
 
-  protected readonly contextToOrchestratorId: ContextToOrchestratorIdFn;
+  protected readonly contextToOrchestratorId: ContextToOrchestratorIdFn | null;
+
+  protected readonly contextToCubeStoreRouterId: ContextToCubeStoreRouterIdFn;
 
   protected readonly preAggregationsSchema: PreAggregationsSchemaFn;
 
@@ -192,6 +195,7 @@ export class CubejsServerCore {
     }
 
     this.contextToOrchestratorId = this.options.contextToOrchestratorId || (() => 'STANDALONE');
+    this.contextToCubeStoreRouterId = this.options.contextToCubeStoreRouterId;
 
     // proactively free up old cache values occasionally
     if (this.options.maxCompilerCacheKeepAlive) {

packages/cubejs-server-core/src/core/types.ts

@@ -122,6 +122,7 @@ export type DatabaseType =
 export type ContextToAppIdFn = (context: RequestContext) => string | Promise<string>;
 export type ContextToRolesFn = (context: RequestContext) => string[] | Promise<string[]>;
 export type ContextToOrchestratorIdFn = (context: RequestContext) => string | Promise<string>;
+export type ContextToCubeStoreRouterIdFn = (context: RequestContext) => string | Promise<string>;
 
 export type OrchestratorOptionsFn = (context: RequestContext) => OrchestratorOptions | Promise<OrchestratorOptions>;
 
@@ -179,6 +180,7 @@ export interface CreateOptions {
   contextToAppId?: ContextToAppIdFn;
   contextToRoles?: ContextToRolesFn;
   contextToOrchestratorId?: ContextToOrchestratorIdFn;
+  contextToCubeStoreRouterId?: ContextToCubeStoreRouterIdFn;
   contextToApiScopes?: ContextToApiScopesFn;
   repositoryFactory?: (context: RequestContext) => SchemaFileRepository;
   checkAuth?: CheckAuthFn;