Added millisecond, microsecond and nanosecond support to date fns

Author: matt-aitken

internal-packages/clickhouse/src/tsqlFunctions.test.ts

@@ -413,6 +413,9 @@ describe("TSQL Function Smoke Tests", () => {
       ["dateAdd (keyword unit)", "SELECT dateAdd(day, 7, created_at) AS r FROM task_runs"],
       ["dateSub (string unit)", "SELECT dateSub('hour', 1, created_at) AS r FROM task_runs"],
       ["dateDiff (string unit)", "SELECT dateDiff('minute', created_at, updated_at) AS r FROM task_runs"],
+      ["dateDiff (millisecond)", "SELECT dateDiff('millisecond', created_at, updated_at) AS r FROM task_runs"],
+      ["dateDiff (microsecond)", "SELECT dateDiff('microsecond', created_at, updated_at) AS r FROM task_runs"],
+      ["dateDiff (nanosecond)", "SELECT dateDiff('nanosecond', created_at, updated_at) AS r FROM task_runs"],
       ["dateTrunc (string unit)", "SELECT dateTrunc('month', created_at) AS r FROM task_runs"],
       ["date_add (string unit)", "SELECT date_add('day', 7, created_at) AS r FROM task_runs"],
       ["date_sub (string unit)", "SELECT date_sub('hour', 1, created_at) AS r FROM task_runs"],

internal-packages/tsql/src/query/printer.test.ts

@@ -1312,6 +1312,31 @@ describe("ClickHousePrinter", () => {
 
       expect(sql).toContain("dateAdd(day, 7, created_at)");
     });
+
+    it("should output dateDiff with sub-second units as bare keywords", () => {
+      const { sql } = printQuery(
+        "SELECT dateDiff('millisecond', started_at, completed_at) AS dur FROM task_runs"
+      );
+
+      expect(sql).toContain("dateDiff(millisecond,");
+      expect(sql).not.toContain("'millisecond'");
+    });
+
+    it("should output dateDiff with microsecond as bare keyword", () => {
+      const { sql } = printQuery(
+        "SELECT dateDiff('microsecond', started_at, completed_at) AS dur FROM task_runs"
+      );
+
+      expect(sql).toContain("dateDiff(microsecond,");
+    });
+
+    it("should output dateDiff with nanosecond as bare keyword", () => {
+      const { sql } = printQuery(
+        "SELECT dateDiff('nanosecond', started_at, completed_at) AS dur FROM task_runs"
+      );
+
+      expect(sql).toContain("dateDiff(nanosecond,");
+    });
   });
 
   describe("Tenant isolation", () => {

internal-packages/tsql/src/query/printer.ts

@@ -2940,6 +2940,9 @@ export class ClickHousePrinter {
    * Valid ClickHouse interval unit keywords used by date functions like dateAdd, dateDiff, etc.
    */
   private static readonly INTERVAL_UNITS = new Set([
+    "nanosecond",
+    "microsecond",
+    "millisecond",
     "second",
     "minute",
     "hour",