add cubeorchestrator/parse_cubestore_ws_result_message export

Author: KSDaemon

packages/cubejs-backend-native/Cargo.lock

@@ -16,7 +16,8 @@ opt-level = 1
 crate-type = ["cdylib", "lib"]
 
 [dependencies]
-cubesqlplanner = { path = "../../rust/cubesqlplanner/cubesqlplanner/" }
+cubesqlplanner = { path = "../../rust/cubesqlplanner/cubesqlplanner" }
+cubeorchestrator = { path = "../../rust/cubeorchestrator" }
 cubenativeutils = { path = "../../rust/cubenativeutils" }
 async-channel = { version = "2" }
 async-trait = "0.1.36"

packages/cubejs-backend-native/Cargo.toml

@@ -348,6 +348,14 @@ export const buildSqlAndParams = (cubeEvaluator: any): String => {
   return native.buildSqlAndParams(cubeEvaluator);
 };
 
+export type ResultRow = Record<string, string>;
+
+export const parseCubestoreResultMessage = (message: ArrayBuffer): ResultRow[] => {
+  const native = loadNative();
+
+  return native.parseCubestoreResultMessage(message);
+};
+
 export interface PyConfiguration {
   repositoryFactory?: (ctx: unknown) => Promise<unknown>,
   logger?: (msg: string, params: Record<string, any>) => void,

packages/cubejs-backend-native/js/index.ts

@@ -32,9 +32,12 @@ use cubenativeutils::wrappers::NativeContextHolder;
 use cubesqlplanner::cube_bridge::base_query_options::NativeBaseQueryOptions;
 use cubesqlplanner::planner::base_query::BaseQuery;
 
+use cubeorchestrator::cubestore_message_parser::parse_cubestore_ws_result;
+
 use cubesql::{telemetry::ReportingLogger, CubeError};
 
 use neon::prelude::*;
+use neon::types::buffer::TypedArray;
 
 struct SQLInterface {
     services: Arc<NodeCubeServices>,
@@ -503,6 +506,30 @@ fn debug_js_to_clrepr_to_js(mut cx: FunctionContext) -> JsResult<JsValue> {
     arg_clrep.into_js(&mut cx)
 }
 
+//============ sql orchestrator ===================
+
+fn parse_cubestore_ws_result_message(mut cx: FunctionContext) -> JsResult<JsValue> {
+    let msg = cx.argument::<JsBuffer>(0)?;
+    let msg_data = msg.as_slice(&cx);
+    match parse_cubestore_ws_result(msg_data.to_vec()) {
+        Ok(result) => {
+            let js_array = JsArray::new(&mut cx, result.len());
+            for (i, row) in result.into_iter().enumerate() {
+                let js_row = JsObject::new(&mut cx);
+                for (key, value) in row {
+                    let js_key = cx.string(key);
+                    let js_value = cx.string(value);
+                    js_row.set(&mut cx, js_key, js_value)?;
+                }
+                js_array.set(&mut cx, i as u32, js_row)?;
+            }
+            Ok(js_array.upcast())
+        }
+        Err(err) => cx.throw_error(err.to_string()),
+    }
+}
+
+
 pub fn register_module_exports<C: NodeConfiguration + 'static>(
     mut cx: ModuleContext,
 ) -> NeonResult<()> {
@@ -513,8 +540,12 @@ pub fn register_module_exports<C: NodeConfiguration + 'static>(
     cx.export_function("isFallbackBuild", is_fallback_build)?;
     cx.export_function("__js_to_clrepr_to_js", debug_js_to_clrepr_to_js)?;
 
+    //============ sql planner exports ===================
     cx.export_function("buildSqlAndParams", build_sql_and_params)?;
 
+    //========= sql orchestrator exports =================
+    cx.export_function("parseCubestoreResultMessage", parse_cubestore_ws_result_message)?;
+
     crate::template::template_register_module(&mut cx)?;
 
     #[cfg(feature = "python")]