feat(nodejs): nanos precision for timestamps

Author: glasstiger

src/buffer/base.ts

@@ -9,12 +9,7 @@ import {
   DEFAULT_BUFFER_SIZE,
   DEFAULT_MAX_BUFFER_SIZE,
 } from "./index";
-import {
-  isInteger,
-  timestampToMicros,
-  timestampToNanos,
-  TimestampUnit,
-} from "../utils";
+import { isInteger, TimestampUnit } from "../utils";
 
 // Default maximum length for table and column names.
 const DEFAULT_MAX_NAME_LENGTH = 127;
@@ -269,6 +264,12 @@ abstract class SenderBufferBase implements SenderBuffer {
     return this;
   }
 
+  protected abstract writeTimestamp(
+    timestamp: number | bigint,
+    unit: TimestampUnit,
+    designated: boolean,
+  ): void;
+
   /**
    * Writes a timestamp column with its value into the buffer. <br>
    * Use it to insert into TIMESTAMP columns.
@@ -286,13 +287,9 @@ abstract class SenderBufferBase implements SenderBuffer {
     if (typeof value !== "bigint" && !Number.isInteger(value)) {
       throw new Error(`Value must be an integer or BigInt, received ${value}`);
     }
-    this.writeColumn(name, value, () => {
-      const valueMicros = timestampToMicros(BigInt(value), unit);
-      const valueStr = valueMicros.toString();
-      this.checkCapacity([valueStr], 1);
-      this.write(valueStr);
-      this.write("t");
-    });
+    this.writeColumn(name, value, () =>
+      this.writeTimestamp(value, unit, false),
+    );
     return this;
   }
 
@@ -313,11 +310,9 @@ abstract class SenderBufferBase implements SenderBuffer {
         `Designated timestamp must be an integer or BigInt, received ${timestamp}`,
       );
     }
-    const timestampNanos = timestampToNanos(BigInt(timestamp), unit);
-    const timestampStr = timestampNanos.toString();
-    this.checkCapacity([timestampStr], 2);
+    this.checkCapacity([], 1);
     this.write(" ");
-    this.write(timestampStr);
+    this.writeTimestamp(timestamp, unit, true);
     this.write("\n");
     this.startNewRow();
   }

src/buffer/bufferv1.ts

@@ -2,6 +2,7 @@
 import { SenderOptions } from "../options";
 import { SenderBuffer } from "./index";
 import { SenderBufferBase } from "./base";
+import { timestampToMicros, timestampToNanos, TimestampUnit } from "../utils";
 
 /**
  * Buffer implementation for protocol version 1. <br>
@@ -39,6 +40,23 @@ class SenderBufferV1 extends SenderBufferBase {
     return this;
   }
 
+  protected writeTimestamp(
+    timestamp: number | bigint,
+    unit: TimestampUnit = "us",
+    designated: boolean,
+  ): void {
+    const biValue = BigInt(timestamp);
+    const timestampValue = designated
+      ? timestampToNanos(biValue, unit)
+      : timestampToMicros(biValue, unit);
+    const timestampStr = timestampValue.toString();
+    this.checkCapacity([timestampStr], 2);
+    this.write(timestampStr);
+    if (!designated) {
+      this.write("t");
+    }
+  }
+
   /**
    * Array columns are not supported in protocol v1.
    *

src/buffer/bufferv2.ts

@@ -2,7 +2,13 @@
 import { SenderOptions } from "../options";
 import { SenderBuffer } from "./index";
 import { SenderBufferBase } from "./base";
-import { ArrayPrimitive, getDimensions, validateArray } from "../utils";
+import {
+  ArrayPrimitive,
+  getDimensions,
+  timestampToMicros,
+  TimestampUnit,
+  validateArray,
+} from "../utils";
 
 // Column type constants for protocol v2.
 const COLUMN_TYPE_DOUBLE: number = 10;
@@ -53,6 +59,28 @@ class SenderBufferV2 extends SenderBufferBase {
     return this;
   }
 
+  protected writeTimestamp(
+    value: number | bigint,
+    unit: TimestampUnit = "us",
+  ): void {
+    let biValue: bigint;
+    let suffix: string;
+    switch (unit) {
+      case "ns":
+        biValue = BigInt(value);
+        suffix = "n";
+        break;
+      default:
+        biValue = timestampToMicros(BigInt(value), unit);
+        suffix = "t";
+    }
+
+    const timestampStr = biValue.toString();
+    this.checkCapacity([timestampStr], 2);
+    this.write(timestampStr);
+    this.write(suffix);
+  }
+
   /**
    * Write an array column with its values into the buffer using v2 format.
    *

test/logging.test.ts

@@ -9,7 +9,7 @@ import {
   vi,
 } from "vitest";
 
-import { Logger } from "../src/logging";
+import { Logger } from "../src";
 
 describe("Default logging suite", function () {
   const error = vi.spyOn(console, "error").mockImplementation(() => {});

test/options.test.ts

@@ -2,7 +2,7 @@
 import { describe, it, expect, beforeAll, afterAll } from "vitest";
 import { Agent } from "undici";
 
-import { SenderOptions } from "../src/options";
+import { SenderOptions } from "../src";
 import { MockHttp } from "./util/mockhttp";
 import { readFileSync } from "fs";