fix(mssql-driver): Support MS SQL dialect for Tesseract (cube-js#10343)

Author: ovr

.github/workflows/push.yml

@@ -432,6 +432,9 @@ jobs:
           - db: postgres
             node-version: 22.x
             use_tesseract_sql_planner: true
+          - db: mssql
+            node-version: 22.x
+            use_tesseract_sql_planner: true
       fail-fast: false
 
     steps:

CLAUDE.md

@@ -75,6 +75,7 @@ yarn build  # Build for production
 3. **API Gateway**: Provides REST, GraphQL, and SQL APIs
 4. **CubeSQL**: Postgres-compatible SQL interface (Rust)
 5. **CubeStore**: Distributed OLAP storage engine (Rust)
+6. **Tesseract**: Native SQL planner (Rust) located in `/rust/cubesqlplanner` - enabled via `CUBESQL_SQL_PUSH_DOWN=true` environment variable
 
 ### Package Management
 - Uses Yarn workspaces with Lerna for package management

packages/cubejs-schema-compiler/src/adapter/MssqlQuery.ts

@@ -270,6 +270,36 @@ export class MssqlQuery extends BaseQuery {
     delete templates.types.interval;
     templates.types.binary = 'VARBINARY';
     templates.params.param = '@_{{ param_index + 1 }}';
+    // MSSQL does not support ordinal GROUP BY (GROUP BY 1, 2), must use expressions
+    templates.statements.group_by_exprs = '{{ group_by | map(attribute=\'expr\') | join(\', \') }}';
+    // MSSQL does not support ordinal ORDER BY (ORDER BY 1, 2), must use expressions
+    templates.expressions.order_by = '{{ expr }} {% if asc %}ASC{% else %}DESC{% endif %}';
+    // MSSQL uses CAST(...AS DATETIME2) instead of ::timestamp
+    templates.statements.time_series_select = 'SELECT CAST(date_from AS DATETIME2) AS "date_from",\n' +
+      'CAST(date_to AS DATETIME2) AS "date_to" \n' +
+      'FROM(\n' +
+      '    VALUES ' +
+      '{% for time_item in seria  %}' +
+      '(\'{{ time_item[0] }}\', \'{{ time_item[1] }}\')' +
+      '{% if not loop.last %}, {% endif %}' +
+      '{% endfor %}' +
+      ') AS dates (date_from, date_to)';
+    // MSSQL uses OFFSET/FETCH instead of LIMIT/OFFSET
+    templates.statements.select = '{% if ctes %} WITH \n' +
+      '{{ ctes | join(\',\n\') }}\n' +
+      '{% endif %}' +
+      'SELECT {% if distinct %}DISTINCT {% endif %}' +
+      '{{ select_concat | map(attribute=\'aliased\') | join(\', \') }} {% if from %}\n' +
+      'FROM (\n' +
+      '{{ from | indent(2, true) }}\n' +
+      ') AS {{ from_alias }}{% elif from_prepared %}\n' +
+      'FROM {{ from_prepared }}' +
+      '{% endif %}' +
+      '{% if filter %}\nWHERE {{ filter }}{% endif %}' +
+      '{% if group_by %}\nGROUP BY {{ group_by }}{% endif %}' +
+      '{% if having %}\nHAVING {{ having }}{% endif %}' +
+      '{% if order_by %}\nORDER BY {{ order_by | map(attribute=\'expr\') | join(\', \') }}\nOFFSET {% if offset is not none %}{{ offset }}{% else %}0{% endif %} ROWS{% endif %}' +
+      '{% if limit is not none %}\nFETCH NEXT {{ limit }} ROWS ONLY{% endif %}';
     return templates;
   }
 }

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

@@ -143,3 +143,10 @@ export class MSSqlDbRunner extends BaseDbRunner {
     return new MssqlQuery(compilers, query);
   }
 }
+
+export const dbRunner = new MSSqlDbRunner();
+
+// eslint-disable-next-line no-undef
+afterAll(async () => {
+  await dbRunner.tearDown();
+});

packages/cubejs-schema-compiler/test/integration/mssql/custom-granularities.test.ts

@@ -1,11 +1,9 @@
 import { prepareYamlCompiler } from '../../unit/PrepareCompiler';
-import { MSSqlDbRunner } from './MSSqlDbRunner';
+import { dbRunner } from './MSSqlDbRunner';
 
 describe('Custom Granularities', () => {
   jest.setTimeout(200000);
 
-  const dbRunner = new MSSqlDbRunner();
-
   const { compiler, joinGraph, cubeEvaluator } = prepareYamlCompiler(`
   cubes:
     - name: orders

packages/cubejs-schema-compiler/test/integration/mssql/mssql-cumulative-measures.test.ts

@@ -1,18 +1,12 @@
 import R from 'ramda';
 import { MssqlQuery } from '../../../src/adapter/MssqlQuery';
 import { prepareJsCompiler } from '../../unit/PrepareCompiler';
-import { MSSqlDbRunner } from './MSSqlDbRunner';
+import { dbRunner } from './MSSqlDbRunner';
 import { createJoinedCubesSchema } from '../../unit/utils';
 
 describe('MSSqlPreAggregations', () => {
   jest.setTimeout(200000);
 
-  const dbRunner = new MSSqlDbRunner();
-
-  afterAll(async () => {
-    await dbRunner.tearDown();
-  });
-
   const { compiler, joinGraph, cubeEvaluator } = prepareJsCompiler(`
     cube(\`visitors\`, {
       sql: \`

packages/cubejs-schema-compiler/test/integration/mssql/mssql-pre-aggregations.test.ts

@@ -1,16 +1,10 @@
 import { MssqlQuery } from '../../../src/adapter/MssqlQuery';
 import { prepareJsCompiler } from '../../unit/PrepareCompiler';
-import { MSSqlDbRunner } from './MSSqlDbRunner';
+import { dbRunner } from './MSSqlDbRunner';
 
 describe('MSSqlUngrouped', () => {
   jest.setTimeout(200000);
 
-  const dbRunner = new MSSqlDbRunner();
-
-  afterAll(async () => {
-    await dbRunner.tearDown();
-  });
-
   const { compiler, joinGraph, cubeEvaluator } = prepareJsCompiler(`
     const perVisitorRevenueMeasure = {
       type: 'number',