UDT request

Adds support for converting JS UDT type into Rust UDT type.

Author: PiotrJunior

lib/types/cql-utils.js

@@ -117,6 +117,24 @@ function encodeTuple(value, type) {
     return res;
 }
 
+/**
+ * Encodes UDT into QueryParameterWrapper
+ * @param {Object} value
+ * @param {rust.ComplexType} complexType
+ * @returns {Array}
+ */
+function encodeUdt(value, complexType) {
+    let names = [];
+    let values = [];
+    while (complexType) {
+        let name = complexType.getFieldName();
+        names.push(name);
+        values.push(getWrapped(complexType.getFirstSupportType(), value[name]));
+        complexType = complexType.getSecondSupportType();
+    }
+    return [names, values];
+}
+
 /**
  * Wrap value into MaybeUnsetQueryParameterWrapper based on the type
  * @param {rust.ComplexType} type
@@ -143,7 +161,7 @@ function getWrapped(type, value) {
  * @returns {rust.QueryParameterWrapper}
  */
 function wrapValueBasedOnType(type, value) {
-    let encodedElement;
+    let encodedElement, name, keyspace, names, values;
     let tmpElement;
     // To increase clarity of the error messages, in case value is of different type then expected,
     // when we call some methods on value variable, type is checked explicitly.
@@ -198,6 +216,18 @@ function wrapValueBasedOnType(type, value) {
             } */
             encodedElement = encodeListLike(value, type.getFirstSupportType());
             return rust.QueryParameterWrapper.fromSet(encodedElement);
+        case rust.CqlType.UserDefinedType:
+            name = type.getUdtName();
+            keyspace = type.getUdtKeyspace();
+            encodedElement = encodeUdt(value, type);
+            names = encodedElement[0];
+            values = encodedElement[1];
+            return rust.QueryParameterWrapper.fromUdt(
+                name,
+                keyspace,
+                names,
+                values,
+            );
         case rust.CqlType.Text:
             return rust.QueryParameterWrapper.fromText(value);
         case rust.CqlType.Timestamp:
@@ -293,7 +323,7 @@ function wrapValueBasedOnType(type, value) {
         default:
             // Or not yet implemented type
             throw new ReferenceError(
-                `[INTERNAL DRIVER ERROR] Unknown type: ${type}`,
+                `[INTERNAL DRIVER ERROR] Unknown type: ${type.baseType}`,
             );
     }
 }

src/requests/parameter_wrappers.rs

@@ -140,6 +140,26 @@ impl QueryParameterWrapper {
         }
     }
 
+    #[napi]
+    pub fn from_udt(
+        name: String,
+        keyspace: String,
+        names: Vec<String>,
+        values: Vec<&QueryParameterWrapper>,
+    ) -> QueryParameterWrapper {
+        QueryParameterWrapper {
+            parameter: CqlValue::UserDefinedType {
+                keyspace,
+                name,
+                fields: names
+                    .iter()
+                    .zip(values.iter())
+                    .map(|(name, value)| (name.clone(), Some(value.parameter.clone())))
+                    .collect(),
+            },
+        }
+    }
+
     #[napi]
     pub fn from_map(
         val: Vec<(&QueryParameterWrapper, &QueryParameterWrapper)>,