fix: save progress

Author: astandrik

src/containers/Tenant/Query/QueryResult/QueryResultViewer.tsx

@@ -328,7 +328,7 @@ export function QueryResultViewer({
                 {renderRightControls()}
             </div>
             {isLoading || isQueryCancelledError(error) ? null : <QuerySettingsBanner />}
-            <LoaderWrapper loading={isLoading}>
+            <LoaderWrapper loading={isLoading && !streaming}>
                 <Fullscreen className={b('result')}>{renderResultSection()}</Fullscreen>
             </LoaderWrapper>
         </React.Fragment>

src/containers/Tenant/Query/QueryResult/components/ResultSetsViewer/ResultSetsViewer.tsx

@@ -28,31 +28,6 @@ interface StreamingResultSetsProps extends BaseResultSetsViewerProps {
 type ResultSetsViewerProps = RegularResultSetsProps | StreamingResultSetsProps;
 
 export function ResultSetsViewer(props: ResultSetsViewerProps) {
-    console.log('[ResultSetsViewer] Rendering with props:', {
-        type: props.type,
-        selectedResultSet: props.selectedResultSet,
-        isStreaming: props.type === 'streaming',
-        streamingState:
-            props.type === 'streaming'
-                ? {
-                      hasColumns: Boolean(props.streaming.columns),
-                      rowCount: props.streaming.rows?.length,
-                      active: props.streaming.active,
-                      progress: props.streaming.progress,
-                  }
-                : undefined,
-        regularState:
-            props.type === 'regular'
-                ? {
-                      resultSetsCount: props.resultSets.length,
-                      currentSetColumns: Boolean(
-                          props.resultSets[props.selectedResultSet]?.columns,
-                      ),
-                      currentSetRows: props.resultSets[props.selectedResultSet]?.result?.length,
-                  }
-                : undefined,
-    });
-
     const {selectedResultSet, setSelectedResultSet} = props;
 
     const resultsSetsCount = props.type === 'streaming' ? 1 : props.resultSets.length;
@@ -66,13 +41,6 @@ export function ResultSetsViewer(props: ResultSetsViewerProps) {
               }
             : props.resultSets[selectedResultSet];
 
-    console.log('[ResultSetsViewer] Current result:', {
-        hasResult: Boolean(currentResult),
-        hasColumns: Boolean(currentResult?.columns),
-        rowCount: currentResult?.result?.length,
-        truncated: currentResult?.truncated,
-    });
-
     const renderTabs = () => {
         if (resultsSetsCount > 1) {
             const tabsItems = getArray(resultsSetsCount).map((item) => ({

src/services/api/viewer.ts

@@ -35,7 +35,7 @@ import type {TUserToken} from '../../types/api/whoami';
 import type {QuerySyntax, TransactionMode} from '../../types/store/query';
 import {BINARY_DATA_IN_PLAIN_TEXT_DISPLAY} from '../../utils/constants';
 import type {Nullable} from '../../utils/typecheckers';
-import {parseMultipart, resetMultipartState} from '../parsers/parseMultipart';
+import {parseMultipart} from '../parsers/parseMultipart';
 import {settingsManager} from '../settings';
 
 import {BaseYdbAPI} from './base';
@@ -389,8 +389,11 @@ export class ViewerAPI extends BaseYdbAPI {
             true,
         );
 
-        // Create parser state in closure
-        let parserState = resetMultipartState();
+        // Create parser state and tracking in closure
+        let parserState = {
+            lastProcessedLength: 0,
+            buffer: '',
+        };
 
         return this.get<string>(
             this.getPath('/viewer/query'),
@@ -413,14 +416,15 @@ export class ViewerAPI extends BaseYdbAPI {
                     Accept: 'multipart/x-mixed-replace',
                 },
                 onDownloadProgress: (progressEvent) => {
-                    console.log('progressEvent', progressEvent.event.target);
                     const response = progressEvent.event.target as XMLHttpRequest;
-                    const {chunks, state} = parseMultipart<QueryAPIResponse<Action>>({
+                    const {chunks, state} = parseMultipart({
                         responseText: response.responseText,
                         state: parserState,
                     });
                     parserState = state;
-                    chunks.forEach((chunk) => onChunk?.(chunk.content));
+                    for (const chunk of chunks) {
+                        onChunk?.(chunk);
+                    }
                 },
             },
         );