feat(firebolt-driver): Timezones support (#6458)

* firebolt timestamp support

* lint

* add test

Author: Jeseeq

packages/cubejs-firebolt-driver/src/FireboltQuery.ts

@@ -25,11 +25,22 @@ export class FireboltQuery extends BaseQuery {
   public paramAllocator!: ParamAllocator;
 
   public convertTz(field: string) {
-    return field;
+    return `${field} AT TIME ZONE '${this.timezone}'`;
   }
 
   public timeStampCast(value: string) {
-    return `${value}::timestamp`;
+    return `${value}::timestamptz`;
+  }
+
+  public dateTimeCast(value: string) {
+    return `${value}::timestampntz`;
+  }
+
+  public seriesSql(timeDimension: any) {
+    const values = timeDimension.timeSeries().map(
+      ([from, to]: [string, string]) => `select '${from}' f, '${to}' t`
+    ).join(' UNION ALL ');
+    return `SELECT ${this.dateTimeCast('dates.f')} date_from, ${this.dateTimeCast('dates.t')} date_to FROM (${values}) AS dates`;
   }
 
   public timeGroupedColumn(granularity: string, dimension: string) {

packages/cubejs-firebolt-driver/test/FireboltQuery.test.ts

@@ -59,7 +59,7 @@ cube(\`sales\`, {
     const queryAndParams = query.buildSqlAndParams();
 
     expect(queryAndParams[0]).toContain(
-      'DATE_TRUNC(\'DAY\', "sales".sales_datetime)'
+      'DATE_TRUNC(\'DAY\', "sales".sales_datetime AT TIME ZONE \'America/Los_Angeles\')'
     );
   }));
 
@@ -81,9 +81,39 @@ cube(\`sales\`, {
     const queryAndParams = query.buildSqlAndParams();
 
     expect(queryAndParams[0]).toContain(
-      '("sales".is_shiped = CAST(? AS BOOL))'
+      '("sales".is_shiped = CAST(? AS BOOLEAN))'
     );
 
     expect(queryAndParams[1]).toEqual(["true"]);
   }))
+    it("should cast timestamp", () =>
+    compiler.compile().then(() => {
+      const query = new FireboltQuery(
+        { joinGraph, cubeEvaluator, compiler },
+        {
+          measures: ["sales.count"],
+          timeDimensions: [
+            {
+              dimension: "sales.salesDatetime",
+              granularity: "day",
+              dateRange: ["2017-01-01", "2017-01-02"],
+            },
+          ],
+          timezone: "America/Los_Angeles",
+          order: [
+            {
+              id: "sales.salesDatetime",
+            },
+          ],
+        }
+      );
+
+      const queryAndParams = query.buildSqlAndParams();
+
+      expect(queryAndParams[0]).toContain(
+        '("sales".sales_datetime >= ?::timestamptz AND "sales".sales_datetime <= ?::timestamptz)'
+      );
+    }));
+
+
 })