fix(cubesql): Continue wait error is being thrown to users

cube-js · Feb 27, 2024 · cb2376c · cb2376c
1 parent 917004e
commit cb2376c
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/packages/cubejs-backend-native/src/transport.rs b/packages/cubejs-backend-native/src/transport.rs
@@ -382,12 +382,18 @@ impl TransportService for NodeBridgeTransport {
                 streaming: false,
             })?;
 
-            let response: serde_json::Value = call_js_with_channel_as_callback(
+            let result = call_js_with_channel_as_callback(
                 self.channel.clone(),
                 self.on_sql_api_load.clone(),
                 Some(extra),
             )
-            .await?;
+            .await;
+            if let Err(e) = &result {
+                if e.message.to_lowercase().contains("continue wait") {
+                    continue;
+                }
+            }
+            let response: serde_json::Value = result?;
             #[cfg(debug_assertions)]
             trace!("[transport] Request <- {:?}", response);
             #[cfg(not(debug_assertions))]