src/utils/hooks/useQueryExecutionSettings.ts - main - githubmirror/ydb-embedded-ui

src/utils/hooks/useQueryExecutionSettings.ts
68 строк3 KB
Copilot
feat: add snapshot read-write transaction mode (#3763)
07 апр 2026, 15:57
07 апр 2026, 15:57bec4acc
import React from 'react';

import {
    useSnapshotReadWriteAvailable,
    useTracingLevelOptionAvailable,
} from '../../store/reducers/capabilities/hooks';
import {SETTING_KEYS} from '../../store/reducers/settings/constants';
import type {QuerySettings, StatisticsMode} from '../../types/store/query';
import {
    DEFAULT_QUERY_SETTINGS,
    STATISTICS_MODES,
    STATISTICS_MODES_WITH_SVG,
    TRANSACTION_MODES,
    isStreamingSupportedForMode,
    querySettingsRestoreSchema,
} from '../query';

import {useQueryStreamingSetting} from './useQueryStreamingSetting';
import {useSetting} from './useSetting';

function getSvgStatisticsMode(mode: StatisticsMode): StatisticsMode {
    return STATISTICS_MODES_WITH_SVG.includes(mode) ? mode : STATISTICS_MODES.full;
}

export const useQueryExecutionSettings = () => {
    const enableTracingLevel = useTracingLevelOptionAvailable();
    const enableSnapshotReadWrite = useSnapshotReadWriteAvailable();
    const [storageSettings, setSettings] = useSetting<QuerySettings>(
        SETTING_KEYS.QUERY_EXECUTION_SETTINGS,
    );

    const validatedSettings = querySettingsRestoreSchema.parse(storageSettings);
    const [useShowPlanToSvg] = useSetting<boolean>(SETTING_KEYS.USE_SHOW_PLAN_SVG);
    const [enableQueryStreaming] = useQueryStreamingSetting();

    const setQueryExecutionSettings = React.useCallback(
        (settings: QuerySettings) => {
            setSettings({
                ...settings,
                statisticsMode: useShowPlanToSvg
                    ? getSvgStatisticsMode(settings.statisticsMode)
                    : settings.statisticsMode,
            });
        },
        [setSettings, useShowPlanToSvg],
    );

    const settings: QuerySettings = {
        ...validatedSettings,
        timeout:
            enableQueryStreaming && isStreamingSupportedForMode(validatedSettings.queryMode)
                ? validatedSettings.timeout || null
                : validatedSettings.timeout || undefined,
        statisticsMode: useShowPlanToSvg
            ? getSvgStatisticsMode(validatedSettings.statisticsMode)
            : validatedSettings.statisticsMode,
        tracingLevel: enableTracingLevel
            ? validatedSettings.tracingLevel
            : DEFAULT_QUERY_SETTINGS.tracingLevel,
        transactionMode:
            !enableSnapshotReadWrite &&
            validatedSettings.transactionMode === TRANSACTION_MODES.snapshotrw
                ? DEFAULT_QUERY_SETTINGS.transactionMode
                : validatedSettings.transactionMode,
    };

    return [settings, setQueryExecutionSettings] as const;
};