Add support for advanced decoding CQL BigInt

Allows to decode CQL BigInt into either BigInt type or Long type,
according to provided configuration option.

This still requires the configuration option to be correctly passed
from the option provided by the client into the rust part of the code.

Author: adespawn

lib/types/result-set.js

@@ -81,6 +81,8 @@ class ResultSet {
             }
             return;
         }
+
+        let decodingOptions = result.getDecodingOptions();
         /**
          * Gets an array rows returned by the query.
          * When the result set represents a response from a write query, this property will be <code>undefined</code>.
@@ -89,7 +91,10 @@ class ResultSet {
          * following pages of results.
          * @type {Array<Row>|undefined}
          */
-        this.rows = resultsWrapper.getRowsFromResultsWrapper(result);
+        this.rows = resultsWrapper.getRowsFromResultsWrapper(
+            result,
+            decodingOptions,
+        );
 
         /**
          * Gets the row length of the result, regardless if the result has been buffered or not

lib/types/results-wrapper.js

@@ -13,16 +13,20 @@ const Row = require("./row");
 /**
  * Checks the type of value wrapper object and gets it from the underlying value
  * @param {rust.CqlValueWrapper} field
+ * @param {rust.DecodingOptions | null} decodingOptions
  * @returns {any}
  */
-function getCqlObject(field) {
+function getCqlObject(field, decodingOptions) {
     if (field == null) return null;
     let type = field.getType();
     let res, fields;
     switch (type) {
         case rust.CqlType.Ascii:
             return field.getAscii();
         case rust.CqlType.BigInt:
+            if (decodingOptions && decodingOptions.useBigIntAsLong) {
+                return field.getBigint();
+            }
             return bigintToLong(field.getBigint());
         case rust.CqlType.Blob:
             return field.getBlob();
@@ -87,9 +91,10 @@ function getCqlObject(field) {
  * Simple way of getting results from rust driver.
  * Call the driver O(columns * rows) times
  * @param {rust.QueryResultWrapper} result
+ * @param {rust.DecodingOptions} decodingOptions
  * @returns {Array<Row> | undefined} Returns array of rows if ResultWrapper has any, and undefined otherwise
  */
-function getRowsFromResultsWrapper(result) {
+function getRowsFromResultsWrapper(result, decodingOptions) {
     let rustRows = result.getRows();
     if (rustRows == null) {
         // Empty results are treated as undefined
@@ -105,7 +110,7 @@ function getRowsFromResultsWrapper(result) {
         for (let j = 0; j < cols.length; j++) {
             // By default driver returns row as a map:
             //   column name -> column value
-            collectedRow[colNames[j]] = getCqlObject(cols[j]);
+            collectedRow[colNames[j]] = getCqlObject(cols[j], decodingOptions);
         }
         rows.push(collectedRow);
     }

src/result.rs

@@ -29,9 +29,17 @@ enum QueryResultVariant {
     RowsResult(QueryRowsResult),
 }
 
+#[napi]
+#[derive(Clone)]
+pub struct EncodingOptions {
+    pub use_big_int_as_long: bool,
+    pub use_big_int_as_varint: bool,
+}
+
 #[napi]
 pub struct QueryResultWrapper {
     internal: QueryResultVariant,
+    decoding_options: EncodingOptions,
 }
 
 #[napi]
@@ -62,7 +70,14 @@ impl QueryResultWrapper {
                 return Err(err_to_napi(e));
             }
         };
-        Ok(QueryResultWrapper { internal: value })
+        let empty_decoding_options = EncodingOptions {
+            use_big_int_as_long: false,
+            use_big_int_as_varint: false,
+        };
+        Ok(QueryResultWrapper {
+            internal: value,
+            decoding_options: empty_decoding_options,
+        })
     }
 
     #[napi]
@@ -134,6 +149,11 @@ impl QueryResultWrapper {
             QueryResultVariant::EmptyResult(v) => v.tracing_id().map(UuidWrapper::from_cql_uuid),
         }
     }
+
+    #[napi]
+    pub fn get_decoding_options(&self) -> EncodingOptions {
+        self.decoding_options.clone()
+    }
 }
 
 #[napi]

test/unit/cql-value-wrapper-tests.js

@@ -34,6 +34,16 @@ describe("Cql value wrapper", function () {
         assert.strictEqual(Long.fromString("69").equals(value), true);
     });
 
+    it("should get bigInt type correctly from napi, as bigint", function () {
+        let element = rust.testsGetCqlWrapperBigint();
+        let type = element.getType();
+        assert.strictEqual(type, rust.CqlType.BigInt);
+        let value = getCqlObject(element, { useBigIntAsLong: true });
+        /* Corresponding value:
+        let element = CqlValue::BigInt(69); */
+        assert.strictEqual(BigInt(69), value);
+    });
+
     it("should get boolean type correctly from napi", function () {
         let element = rust.testsGetCqlWrapperBoolean();
         let type = element.getType();