feat(cubesqlplanner): Base joins support (#8656)

Author: waralexrom

packages/cubejs-backend-native/Cargo.lock

@@ -609,11 +609,18 @@ export class BaseQuery {
     const queryParams = {
       measures: this.options.measures,
       dimensions: this.options.dimensions,
+      timeDimensions: this.options.timeDimensions,
+      timezone: this.options.timezone,
       joinRoot: this.join.root,
+      joinGraph: this.joinGraph,
       cubeEvaluator: this.cubeEvaluator,
+      filters: this.options.filters,
+      baseTools: this,
 
     };
     const res = nativeBuildSqlAndParams(queryParams);
+    // FIXME
+    res[1] = [...res[1]];
     return res;
   }
 
@@ -3233,6 +3240,21 @@ export class BaseQuery {
         ilike: '{{ expr }} {% if negated %}NOT {% endif %}ILIKE {{ pattern }}',
         like_escape: '{{ like_expr }} ESCAPE {{ escape_char }}',
       },
+      filters: {
+        equals: '{{ column }} = {{ value }}{{ is_null_check }}',
+        not_equals: '{{ column }} <> {{ value }}{{ is_null_check }}',
+        or_is_null_check: ' OR {{ column }} IS NULL',
+        set_where: '{{ column }} IS NOT NULL',
+        not_set_where: '{{ column }} IS NULL',
+        in: '{{ column }} IN ({{ values_concat }}){{ is_null_check }}',
+        not_in: '{{ column }} NOT IN ({{ values_concat }}){{ is_null_check }}',
+        time_range_filter: '{{ column }} >= {{ from_timestamp }} AND {{ column }} <= {{ to_timestamp }}',
+        gt: '{{ column }} > {{ param }}',
+        gte: '{{ column }} >= {{ param }}',
+        lt: '{{ column }} < {{ param }}',
+        lte: '{{ column }} <= {{ param }}'
+
+      },
       quotes: {
         identifiers: '"',
         escape: '""'
@@ -3661,6 +3683,10 @@ export class BaseQuery {
     };
   }
 
+  securityContextForRust() {
+    return this.contextSymbolsProxy(this.contextSymbols.securityContext);
+  }
+
   contextSymbolsProxy(symbols) {
     return BaseQuery.contextSymbolsProxyFrom(symbols, this.paramAllocator.allocateParam.bind(this.paramAllocator));
   }

packages/cubejs-schema-compiler/src/adapter/BaseQuery.js

@@ -462,9 +462,10 @@ export class CubeEvaluator extends CubeSymbols {
 
   public isInstanceOfType(type: 'measures' | 'dimensions' | 'segments', path: string | string[]): boolean {
     const cubeAndName = Array.isArray(path) ? path : path.split('.');
-    return this.evaluatedCubes[cubeAndName[0]] &&
+    const symbol = this.evaluatedCubes[cubeAndName[0]] &&
       this.evaluatedCubes[cubeAndName[0]][type] &&
       this.evaluatedCubes[cubeAndName[0]][type][cubeAndName[1]];
+    return symbol !== undefined;
   }
 
   public byPathAnyType(path: string[]) {

packages/cubejs-schema-compiler/src/compiler/CubeEvaluator.ts

@@ -493,8 +493,81 @@ export class CubeSymbols {
     return joinHints;
   }
 
+  resolveSymbolsCallDeps(cubeName, sql) {
+    try {
+      return this.resolveSymbolsCallDeps2(cubeName, sql);
+    } catch (e) {
+      console.log(e);
+      return [];
+    }
+  }
+
+  resolveSymbolsCallDeps2(cubeName, sql) {
+    const deps = [];
+    this.resolveSymbolsCall(sql, (name) => {
+      deps.push({ name, undefined });
+      const resolvedSymbol = this.resolveSymbol(
+        cubeName,
+        name
+      );
+      if (resolvedSymbol._objectWithResolvedProperties) {
+        return resolvedSymbol;
+      }
+      return '';
+    }, {
+      depsResolveFn: (name, parent) => {
+        deps.push({ name, parent });
+        return deps.length - 1;
+      },
+      currResolveIndexFn: () => deps.length - 1,
+      contextSymbols: this.depsContextSymbols(),
+
+    });
+    return deps;
+  }
+
+  depsContextSymbols() {
+    return Object.assign({
+      filterParams: this.filtersProxyDep(),
+      filterGroup: this.filterGroupFunctionDep(),
+      securityContext: BaseQuery.contextSymbolsProxyFrom({}, (param) => param)
+    });
+  }
+
+  filtersProxyDep() {
+    return new Proxy({}, {
+      get: (target, name) => {
+        if (name === '_objectWithResolvedProperties') {
+          return true;
+        }
+        // allFilters is null whenever it's used to test if the member is owned by cube so it should always render to `1 = 1`
+        // and do not check cube validity as it's part of compilation step.
+        const cubeName = this.cubeNameFromPath(name);
+        return new Proxy({ cube: cubeName }, {
+          get: (cubeNameObj, propertyName) => ({
+            filter: (column) => ({
+              __column() {
+                return column;
+              },
+              __member() {
+                return this.pathFromArray([cubeNameObj.cube, propertyName]);
+              },
+              toString() {
+                return '';
+              }
+            })
+          })
+        });
+      }
+    });
+  }
+
+  filterGroupFunctionDep() {
+    return (...filterParamArgs) => '';
+  }
+
   resolveSymbol(cubeName, name) {
-    const { sqlResolveFn, contextSymbols, collectJoinHints } = this.resolveSymbolsCallContext || {};
+    const { sqlResolveFn, contextSymbols, collectJoinHints, depsResolveFn, currResolveIndexFn } = this.resolveSymbolsCallContext || {};
 
     if (name === 'USER_CONTEXT') {
       throw new Error('Support for USER_CONTEXT was removed, please migrate to SECURITY_CONTEXT.');
@@ -528,8 +601,14 @@ export class CubeSymbols {
           name
         );
       }
+    } else if (depsResolveFn) {
+      if (cube) {
+        const newCubeName = this.isCurrentCube(name) ? cubeName : name;
+        const parentIndex = currResolveIndexFn();
+        cube = this.cubeDependenciesProxy(parentIndex, newCubeName);
+        return cube;
+      }
     }
-
     return cube || (this.symbols[cubeName] && this.symbols[cubeName][name]);
   }
 
@@ -623,6 +702,44 @@ export class CubeSymbols {
     return cube && cube[dimName] && cube[dimName][gr] && cube[dimName][gr][granName];
   }
 
+  cubeDependenciesProxy(parentIndex, cubeName) {
+    const self = this;
+    const { depsResolveFn } = self.resolveSymbolsCallContext || {};
+    return new Proxy({}, {
+      get: (v, propertyName) => {
+        if (propertyName === '__cubeName') {
+          depsResolveFn('__cubeName', parentIndex);
+          return cubeName;
+        }
+        const cube = self.symbols[cubeName];
+
+        if (propertyName === 'toString') {
+          depsResolveFn('toString', parentIndex);
+          return () => '';
+        }
+        if (propertyName === 'sql') {
+          depsResolveFn('sql', parentIndex);
+          return () => '';
+        }
+        if (propertyName === '_objectWithResolvedProperties') {
+          return true;
+        }
+        if (cube[propertyName]) {
+          depsResolveFn(propertyName, parentIndex);
+          return '';
+        }
+        if (self.symbols[propertyName]) {
+          const index = depsResolveFn(propertyName, parentIndex);
+          return this.cubeDependenciesProxy(index, propertyName);
+        }
+        if (typeof propertyName === 'string') {
+          throw new UserError(`${cubeName}.${propertyName} cannot be resolved. There's no such member or cube.`);
+        }
+        return undefined;
+      }
+    });
+  }
+
   isCurrentCube(name) {
     return CURRENT_CUBE_CONSTANTS.indexOf(name) >= 0;
   }

packages/cubejs-schema-compiler/src/compiler/CubeSymbols.js

@@ -84,7 +84,7 @@ export class MSSqlDbRunner extends BaseDbRunner {
     (2, 1, 2),
     (3, 3, 6)
     `);
-    await query(`CREATE TABLE ##numbers (num INT);`);
+    await query('CREATE TABLE ##numbers (num INT);');
     await query(`
     INSERT INTO ##numbers (num) VALUES (0), (1), (2), (3), (4), (5), (6), (7), (8), (9),
                                   (10), (11), (12), (13), (14), (15), (16), (17), (18), (19),

packages/cubejs-schema-compiler/test/integration/mssql/MSSqlDbRunner.js

@@ -73,7 +73,7 @@ export class MySqlDbRunner extends BaseDbRunner {
     (2, 1, 2),
     (3, 3, 6)
     `);
-    await query(`CREATE TEMPORARY TABLE numbers (num INT);`);
+    await query('CREATE TEMPORARY TABLE numbers (num INT);');
     await query(`
     INSERT INTO numbers (num) VALUES (0), (1), (2), (3), (4), (5), (6), (7), (8), (9),
                                   (10), (11), (12), (13), (14), (15), (16), (17), (18), (19),

packages/cubejs-schema-compiler/test/integration/mysql/MySqlDbRunner.js

@@ -401,8 +401,6 @@ describe('SQL Generation', () => {
       timezone: 'America/Los_Angeles'
     });
 
-    console.log(query.buildSqlAndParams());
-
     return dbRunner.testQuery(query.buildSqlAndParams()).then(res => {
       console.log(JSON.stringify(res));
       expect(res).toEqual(

packages/cubejs-schema-compiler/test/integration/postgres/sql-generation-logic.test.ts

@@ -561,9 +561,6 @@ describe('SQL Generation', () => {
       }]
     });
 
-    const queryAndParams = query.buildSqlAndParams();
-    console.log(queryAndParams);
-
     return dbRunner.testQuery(query.buildSqlAndParams()).then(res => {
       expect(res).toEqual(
         [

packages/cubejs-schema-compiler/test/integration/postgres/sql-generation.test.ts

@@ -1,6 +1,6 @@
 import R from 'ramda';
 import { PreAggregationPartitionRangeLoader } from '@cubejs-backend/query-orchestrator';
-import { BaseQuery } from "../../../src";
+import { BaseQuery } from '../../../src';
 
 export class BaseDbRunner {
   protected containerLazyInitPromise: any = null;