mirror of
https://github.com/apache/superset.git
synced 2026-04-20 08:34:37 +00:00
105 lines
3.0 KiB
TypeScript
105 lines
3.0 KiB
TypeScript
/**
|
|
* Licensed to the Apache Software Foundation (ASF) under one
|
|
* or more contributor license agreements. See the NOTICE file
|
|
* distributed with this work for additional information
|
|
* regarding copyright ownership. The ASF licenses this file
|
|
* to you under the Apache License, Version 2.0 (the
|
|
* "License"); you may not use this file except in compliance
|
|
* with the License. You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing,
|
|
* software distributed under the License is distributed on an
|
|
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
* KIND, either express or implied. See the License for the
|
|
* specific language governing permissions and limitations
|
|
* under the License.
|
|
*/
|
|
import { useDispatch } from 'react-redux';
|
|
import { t } from '@apache-superset/core';
|
|
import { Dropdown, Button } from '@superset-ui/core/components';
|
|
import { Menu } from '@superset-ui/core/components/Menu';
|
|
import { Icons } from '@superset-ui/core/components/Icons';
|
|
import { queryEditorSetQueryLimit } from 'src/SqlLab/actions/sqlLab';
|
|
import useQueryEditor from 'src/SqlLab/hooks/useQueryEditor';
|
|
|
|
export interface QueryLimitSelectProps {
|
|
queryEditorId: string;
|
|
maxRow: number;
|
|
defaultQueryLimit: number;
|
|
}
|
|
|
|
export function convertToNumWithSpaces(num: number) {
|
|
return num.toString().replace(/(\d)(?=(\d{3})+(?!\d))/g, '$1 ');
|
|
}
|
|
|
|
export function convertToShortNum(num: number) {
|
|
if (num < 1000) {
|
|
return num;
|
|
}
|
|
if (num < 1_000_000) {
|
|
return `${num / 1000}K`;
|
|
}
|
|
if (num < 1_000_000_000) {
|
|
return `${num / 1000_000}M`;
|
|
}
|
|
return num;
|
|
}
|
|
|
|
function renderQueryLimit(
|
|
maxRow: number,
|
|
setQueryLimit: (limit: number) => void,
|
|
) {
|
|
const limitDropdown = [];
|
|
|
|
// Construct limit dropdown as increasing powers of ten until we reach SQL_MAX_ROW
|
|
for (let i = 10; i < maxRow; i *= 10) {
|
|
limitDropdown.push(i);
|
|
}
|
|
limitDropdown.push(maxRow);
|
|
|
|
return (
|
|
<Menu
|
|
items={[...new Set(limitDropdown)].map(limit => ({
|
|
key: `${limit}`,
|
|
onClick: () => setQueryLimit(limit),
|
|
label: `${convertToNumWithSpaces(limit)} `,
|
|
}))}
|
|
/>
|
|
);
|
|
}
|
|
|
|
const QueryLimitSelect = ({
|
|
queryEditorId,
|
|
maxRow,
|
|
defaultQueryLimit,
|
|
}: QueryLimitSelectProps) => {
|
|
const dispatch = useDispatch();
|
|
|
|
const queryEditor = useQueryEditor(queryEditorId, ['id', 'queryLimit']);
|
|
const queryLimit = queryEditor.queryLimit || defaultQueryLimit;
|
|
const setQueryLimit = (updatedQueryLimit: number) =>
|
|
dispatch(queryEditorSetQueryLimit(queryEditor, updatedQueryLimit));
|
|
|
|
return (
|
|
<Dropdown
|
|
popupRender={() => renderQueryLimit(maxRow, setQueryLimit)}
|
|
trigger={['click']}
|
|
>
|
|
<Button
|
|
size="small"
|
|
color="primary"
|
|
variant="text"
|
|
showMarginRight={false}
|
|
>
|
|
<span>{t('Limit')}</span>
|
|
<span className="limitDropdown">{convertToShortNum(queryLimit)}</span>
|
|
<Icons.DownOutlined iconSize="m" />
|
|
</Button>
|
|
</Dropdown>
|
|
);
|
|
};
|
|
|
|
export default QueryLimitSelect;
|