fix result processing when streaming

cube-js · Dec 18, 2024 · 1a39a49 · 1a39a49
1 parent 070c1c9
commit 1a39a49
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/packages/cubejs-api-gateway/src/gateway.ts b/packages/cubejs-api-gateway/src/gateway.ts
@@ -1953,7 +1953,14 @@ class ApiGateway {
           })
         );
 
-        if (!request.streaming) {
+        if (request.streaming) {
+          const r = results[0];
+          res(await getFinalQueryResult(
+            r.transformDataParams,
+            r.rawData.isNative ? r.rawData.getNativeRef() : r.rawData,
+            cleanupResult(r),
+          ));
+        } else {
           // We prepare the final json result on native side
           const [transformDataJson, rawData, resultDataJson] = (results as {
             transformDataParams: any;
@@ -1969,8 +1976,6 @@ class ApiGateway {
           );
 
           res(await getFinalQueryResultArray(transformDataJson, rawData, resultDataJson));
-        } else {
-          res(results[0]);
         }
       }
     } catch (e: any) {