feat(schema-compiler): Custom granularities support for AWS Athena/Presto dialect

Author: KSDaemon

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

@@ -1,3 +1,4 @@
+import { parseSqlInterval } from '@cubejs-backend/shared';
 import { BaseQuery } from './BaseQuery';
 import { BaseFilter } from './BaseFilter';
 
@@ -55,6 +56,30 @@ export class PrestodbQuery extends BaseQuery {
       field;
   }
 
+  /**
+   * Returns sql for source expression floored to timestamps aligned with
+   * intervals relative to origin timestamp point.
+   * Athena doesn't support INTERVALs directly — using date_diff/date_add
+   */
+  public dateBin(interval: string, source: string, origin: string): string {
+    const intervalParsed = parseSqlInterval(interval);
+    const intervalParts = Object.entries(intervalParsed);
+
+    if (intervalParts.length > 1) {
+      throw new Error('Athena/Presto supports only simple intervals with one date part');
+    }
+
+    const [unit, count] = intervalParts[0];
+    const originExpr = this.timeStampCast(`'${origin}'`);
+
+    return `date_add('${unit}',
+      floor(
+        date_diff('${unit}', ${originExpr}, ${source}) / ${count}
+      ) * ${count},
+      ${originExpr}
+    )`;
+  }
+
   public timeGroupedColumn(granularity, dimension) {
     return `date_trunc('${GRANULARITY_TO_INTERVAL[granularity]}', ${dimension})`;
   }

packages/cubejs-testing-drivers/fixtures/athena.json

@@ -151,15 +151,8 @@
     "---------------------------------------",
     "Custom Granularities                   ",
     "---------------------------------------",
-    "querying custom granularities ECommerce: count by half_year + no dimension",
-    "querying custom granularities ECommerce: count by half_year_by_1st_april + no dimension",
     "querying custom granularities ECommerce: count by three_months_by_march + no dimension",
-    "querying custom granularities ECommerce: count by half_year + dimension",
-    "querying custom granularities ECommerce: count by half_year_by_1st_april + dimension",
     "querying custom granularities ECommerce: count by three_months_by_march + dimension",
-    "querying custom granularities ECommerce: count by two_mo_by_feb + no dimension + rollingCountByUnbounded",
-    "querying custom granularities ECommerce: count by two_mo_by_feb + no dimension + rollingCountByTrailing",
-    "querying custom granularities ECommerce: count by two_mo_by_feb + no dimension + rollingCountByLeading",
 
     "SKIPPED SQL API (Need work)",
     "---------------------------------------",