prepare transformQueryData native wrapper

Author: KSDaemon

packages/cubejs-backend-native/Cargo.lock

@@ -38,6 +38,9 @@ use cubesql::{telemetry::ReportingLogger, CubeError};
 
 use neon::prelude::*;
 use neon::types::buffer::TypedArray;
+use serde::Deserialize;
+use cubeorchestrator::cubestore_result_transform::transform_data;
+use cubeorchestrator::types::{TransformDataRequest};
 
 struct SQLInterface {
     services: Arc<NodeCubeServices>,
@@ -543,6 +546,56 @@ fn get_cubestore_result(mut cx: FunctionContext) -> JsResult<JsValue> {
     Ok(js_array.upcast())
 }
 
+fn js_object_to_struct<T>(cx: &mut FunctionContext) -> Result<T, neon::result::Throw>
+where
+    T: for<'de> Deserialize<'de>,
+{
+    let json_str = cx
+        .argument::<JsString>(0)?
+        .value(cx);
+
+    let result: Result<T, serde_json::Error> = serde_json::from_str(&json_str);
+    match result {
+        Ok(value) => Ok(value),
+        Err(err) => cx.throw_error(err.to_string()),
+    }
+}
+
+fn transform_query_data(mut cx: FunctionContext) -> JsResult<JsObject> {
+    let request_data: TransformDataRequest = js_object_to_struct(&mut cx)?;
+
+    let alias_to_member_name_map = &request_data.alias_to_member_name_map;
+    let annotation = &request_data.annotation;
+    let data = &request_data.data;
+    let query = &request_data.query;
+    let query_type = &request_data.query_type;
+    let res_type = &request_data.res_type;
+
+    let transformed = match transform_data(
+        alias_to_member_name_map,
+        annotation,
+        data.clone(),
+        query,
+        query_type,
+        res_type.clone(),
+    ) {
+        Ok(data) => data,
+        Err(err) => return cx.throw_error(err.to_string())
+    };
+
+    let json_data = match serde_json::to_string(&transformed) {
+        Ok(data) => data,
+        Err(e) => return cx.throw_error(format!("Serialization error: {}", e)),
+    };
+
+    let js_string = cx.string(json_data);
+
+    let js_result = cx.empty_object();
+    js_result.set(&mut cx, "result", js_string)?;
+
+    Ok(js_result)
+}
+
 pub fn register_module_exports<C: NodeConfiguration + 'static>(
     mut cx: ModuleContext,
 ) -> NeonResult<()> {
@@ -562,6 +615,7 @@ pub fn register_module_exports<C: NodeConfiguration + 'static>(
         parse_cubestore_ws_result_message,
     )?;
     cx.export_function("getCubestoreResult", get_cubestore_result)?;
+    cx.export_function("transformQueryData", transform_query_data)?;
 
     crate::template::template_register_module(&mut cx)?;
 

packages/cubejs-backend-native/src/node_export.rs

@@ -4,7 +4,7 @@ version = "0.1.0"
 edition = "2021"
 
 [dependencies]
-chrono = "0.4.31"
+chrono = { version = "0.4.31", features = ["serde"] }
 cubeshared = { path = "../cubeshared" }
 serde = { version = "1.0.215", features = ["derive"] }
 serde_json = "1.0.133"

rust/cubeorchestrator/Cargo.lock

@@ -17,7 +17,7 @@ pub enum DBResponsePrimitive {
     String(String),
 }
 
-#[derive(Debug, Clone)]
+#[derive(Debug, Clone, Deserialize)]
 pub enum DBResponseValue {
     DateTime(DateTime<Utc>),
     Primitive(DBResponsePrimitive),
@@ -148,7 +148,7 @@ pub type AliasToMemberMap = HashMap<String, String>;
 
 pub type MembersMap = HashMap<String, String>;
 
-#[derive(Debug, Serialize, Deserialize)]
+#[derive(Debug, Clone, Serialize, Deserialize)]
 pub struct GranularityMeta {
     pub name: String,
     pub title: String,
@@ -159,7 +159,7 @@ pub struct GranularityMeta {
     pub origin: Option<String>,
 }
 
-#[derive(Debug, Serialize, Deserialize)]
+#[derive(Debug, Clone, Serialize, Deserialize)]
 pub struct ConfigItem {
     pub title: String,
     pub short_title: String,
@@ -278,3 +278,16 @@ pub enum TransformedData {
     },
     Vanilla(Vec<HashMap<String, DBResponsePrimitive>>),
 }
+
+#[derive(Debug, Clone, Deserialize)]
+pub struct TransformDataRequest {
+    #[serde(rename = "aliasToMemberNameMap")]
+    pub alias_to_member_name_map: HashMap<String, String>,
+    pub annotation: HashMap<String, ConfigItem>,
+    pub data: Vec<HashMap<String, DBResponseValue>>,
+    pub query: NormalizedQuery,
+    #[serde(rename = "queryType")]
+    pub query_type: QueryType,
+    #[serde(rename = "resType")]
+    pub res_type: Option<ResultType>,
+}