|
@@ -0,0 +1,270 @@
|
|
|
+import { makeStyles, Theme } from '@material-ui/core';
|
|
|
+import { FC, useContext, useEffect, useState } from 'react';
|
|
|
+import {
|
|
|
+ PartitionManageParam,
|
|
|
+ // PartitionParam,
|
|
|
+ PartitionView,
|
|
|
+} from './Types';
|
|
|
+import MilvusGrid from '../../components/grid';
|
|
|
+import { ColDefinitionsType, ToolBarConfig } from '../../components/grid/Types';
|
|
|
+import { useTranslation } from 'react-i18next';
|
|
|
+import { usePaginationHook } from '../../hooks/Pagination';
|
|
|
+import icons from '../../components/icons/Icons';
|
|
|
+import CustomToolTip from '../../components/customToolTip/CustomToolTip';
|
|
|
+import { rootContext } from '../../context/Root';
|
|
|
+import CreatePartition from './Create';
|
|
|
+import { PartitionHttp } from '../../http/Partition';
|
|
|
+import Status from '../../components/status/Status';
|
|
|
+import { ManageRequestMethods } from '../../types/Common';
|
|
|
+// import { StatusEnum } from '../../components/status/Types';
|
|
|
+// import { useDialogHook } from '../../hooks/Dialog';
|
|
|
+import DeleteTemplate from '../../components/customDialog/DeleteDialogTemplate';
|
|
|
+
|
|
|
+const useStyles = makeStyles((theme: Theme) => ({
|
|
|
+ wrapper: {
|
|
|
+ height: '100%',
|
|
|
+ },
|
|
|
+ icon: {
|
|
|
+ fontSize: '20px',
|
|
|
+ marginLeft: theme.spacing(0.5),
|
|
|
+ },
|
|
|
+}));
|
|
|
+
|
|
|
+const Partitions: FC<{
|
|
|
+ collectionName: string;
|
|
|
+}> = ({ collectionName }) => {
|
|
|
+ const classes = useStyles();
|
|
|
+ const { t } = useTranslation('partition');
|
|
|
+ const { t: successTrans } = useTranslation('success');
|
|
|
+ const { t: btnTrans } = useTranslation('btn');
|
|
|
+ const { t: dialogTrans } = useTranslation('dialog');
|
|
|
+ const InfoIcon = icons.info;
|
|
|
+ // const LoadIcon = icons.load;
|
|
|
+ // const ReleaseIcon = icons.release;
|
|
|
+
|
|
|
+ // const { handleAction } = useDialogHook({ type: 'partition' });
|
|
|
+ const [selectedPartitions, setSelectedPartitions] = useState<PartitionView[]>(
|
|
|
+ []
|
|
|
+ );
|
|
|
+ const [partitions, setPartitions] = useState<PartitionView[]>([]);
|
|
|
+ const {
|
|
|
+ pageSize,
|
|
|
+ handlePageSize,
|
|
|
+ currentPage,
|
|
|
+ handleCurrentPage,
|
|
|
+ total,
|
|
|
+ data: partitionList,
|
|
|
+ } = usePaginationHook(partitions);
|
|
|
+ const [loading, setLoading] = useState<boolean>(true);
|
|
|
+ const { setDialog, handleCloseDialog, openSnackBar } =
|
|
|
+ useContext(rootContext);
|
|
|
+
|
|
|
+ useEffect(() => {
|
|
|
+ fetchPartitions(collectionName);
|
|
|
+ }, [collectionName]);
|
|
|
+
|
|
|
+ const fetchPartitions = async (collectionName: string) => {
|
|
|
+ try {
|
|
|
+ const res = await PartitionHttp.getPartitions(collectionName);
|
|
|
+
|
|
|
+ const partitons: PartitionView[] = res.map(p =>
|
|
|
+ Object.assign(p, { _statusElement: <Status status={p._status} /> })
|
|
|
+ );
|
|
|
+ setLoading(false);
|
|
|
+ setPartitions(partitons);
|
|
|
+ } catch (err) {
|
|
|
+ setLoading(false);
|
|
|
+ }
|
|
|
+ };
|
|
|
+
|
|
|
+ const handleDelete = async () => {
|
|
|
+ for (const partition of selectedPartitions) {
|
|
|
+ const param: PartitionManageParam = {
|
|
|
+ partitionName: partition._name,
|
|
|
+ collectionName,
|
|
|
+ type: ManageRequestMethods.DELETE,
|
|
|
+ };
|
|
|
+ await PartitionHttp.managePartition(param);
|
|
|
+ }
|
|
|
+
|
|
|
+ openSnackBar(successTrans('delete', { name: t('partition') }));
|
|
|
+ fetchPartitions(collectionName);
|
|
|
+ handleCloseDialog();
|
|
|
+ };
|
|
|
+
|
|
|
+ // const handleRelease = async (data: PartitionView) => {
|
|
|
+ // const param: PartitionParam = {
|
|
|
+ // collectionName,
|
|
|
+ // partitionNames: [data._name],
|
|
|
+ // };
|
|
|
+ // const res = await PartitionHttp.releasePartition(param);
|
|
|
+ // openSnackBar(successTrans('release', { name: t('partition') }));
|
|
|
+ // fetchPartitions(collectionName);
|
|
|
+ // return res;
|
|
|
+ // };
|
|
|
+
|
|
|
+ // const handleLoad = async (data: PartitionView) => {
|
|
|
+ // const param: PartitionParam = {
|
|
|
+ // collectionName,
|
|
|
+ // partitionNames: [data._name!],
|
|
|
+ // };
|
|
|
+ // const res = await PartitionHttp.loadPartition(param);
|
|
|
+ // openSnackBar(successTrans('load', { name: t('partition') }));
|
|
|
+ // fetchPartitions(collectionName);
|
|
|
+ // return res;
|
|
|
+ // };
|
|
|
+
|
|
|
+ const toolbarConfigs: ToolBarConfig[] = [
|
|
|
+ {
|
|
|
+ label: t('create'),
|
|
|
+ onClick: () => {
|
|
|
+ setDialog({
|
|
|
+ open: true,
|
|
|
+ type: 'custom',
|
|
|
+ params: {
|
|
|
+ component: (
|
|
|
+ <CreatePartition
|
|
|
+ handleCreate={handleCreatePartition}
|
|
|
+ handleClose={handleCloseDialog}
|
|
|
+ />
|
|
|
+ ),
|
|
|
+ },
|
|
|
+ });
|
|
|
+ },
|
|
|
+ icon: 'add',
|
|
|
+ },
|
|
|
+ {
|
|
|
+ type: 'iconBtn',
|
|
|
+ onClick: () => {
|
|
|
+ setDialog({
|
|
|
+ open: true,
|
|
|
+ type: 'custom',
|
|
|
+ params: {
|
|
|
+ component: (
|
|
|
+ <DeleteTemplate
|
|
|
+ label={btnTrans('delete')}
|
|
|
+ title={dialogTrans('deleteTitle', { type: t('partition') })}
|
|
|
+ text={t('deleteWarning')}
|
|
|
+ handleDelete={handleDelete}
|
|
|
+ />
|
|
|
+ ),
|
|
|
+ },
|
|
|
+ });
|
|
|
+ },
|
|
|
+ label: '',
|
|
|
+ icon: 'delete',
|
|
|
+ // can't delete default partition
|
|
|
+ disabled: () =>
|
|
|
+ selectedPartitions.length === 0 ||
|
|
|
+ selectedPartitions.some(p => p._name === '_default'),
|
|
|
+ tooltip: selectedPartitions.some(p => p._name === '_default')
|
|
|
+ ? t('deletePartitionError')
|
|
|
+ : '',
|
|
|
+ },
|
|
|
+ ];
|
|
|
+
|
|
|
+ const colDefinitions: ColDefinitionsType[] = [
|
|
|
+ {
|
|
|
+ id: '_id',
|
|
|
+ align: 'left',
|
|
|
+ disablePadding: true,
|
|
|
+ label: t('id'),
|
|
|
+ },
|
|
|
+ {
|
|
|
+ id: '_formatName',
|
|
|
+ align: 'left',
|
|
|
+ disablePadding: false,
|
|
|
+ label: t('name'),
|
|
|
+ },
|
|
|
+ // {
|
|
|
+ // id: '_statusElement',
|
|
|
+ // align: 'left',
|
|
|
+ // disablePadding: false,
|
|
|
+ // label: t('status'),
|
|
|
+ // },
|
|
|
+ {
|
|
|
+ id: '_rowCount',
|
|
|
+ align: 'left',
|
|
|
+ disablePadding: false,
|
|
|
+ label: (
|
|
|
+ <span className="flex-center">
|
|
|
+ {t('rowCount')}
|
|
|
+ <CustomToolTip title={t('tooltip')}>
|
|
|
+ <InfoIcon classes={{ root: classes.icon }} />
|
|
|
+ </CustomToolTip>
|
|
|
+ </span>
|
|
|
+ ),
|
|
|
+ },
|
|
|
+ // {
|
|
|
+ // id: 'action',
|
|
|
+ // align: 'center',
|
|
|
+ // disablePadding: false,
|
|
|
+ // label: '',
|
|
|
+ // showActionCell: true,
|
|
|
+ // isHoverAction: true,
|
|
|
+ // actionBarConfigs: [
|
|
|
+ // {
|
|
|
+ // onClick: (e: React.MouseEvent, row: PartitionView) => {
|
|
|
+ // const cb =
|
|
|
+ // row._status === StatusEnum.unloaded ? handleLoad : handleRelease;
|
|
|
+ // handleAction(row, cb);
|
|
|
+ // },
|
|
|
+ // icon: 'load',
|
|
|
+ // label: 'load',
|
|
|
+ // showIconMethod: 'renderFn',
|
|
|
+ // getLabel: (row: PartitionView) =>
|
|
|
+ // row._status === StatusEnum.loaded ? 'release' : 'load',
|
|
|
+ // renderIconFn: (row: PartitionView) =>
|
|
|
+ // row._status === StatusEnum.loaded ? <ReleaseIcon /> : <LoadIcon />,
|
|
|
+ // },
|
|
|
+ // ],
|
|
|
+ // },
|
|
|
+ ];
|
|
|
+
|
|
|
+ const handleSelectChange = (value: PartitionView[]) => {
|
|
|
+ setSelectedPartitions(value);
|
|
|
+ };
|
|
|
+
|
|
|
+ const handlePageChange = (e: any, page: number) => {
|
|
|
+ handleCurrentPage(page);
|
|
|
+ setSelectedPartitions([]);
|
|
|
+ };
|
|
|
+
|
|
|
+ const handleCreatePartition = async (name: string) => {
|
|
|
+ const param: PartitionManageParam = {
|
|
|
+ partitionName: name,
|
|
|
+ collectionName,
|
|
|
+ type: ManageRequestMethods.CREATE,
|
|
|
+ };
|
|
|
+
|
|
|
+ await PartitionHttp.managePartition(param);
|
|
|
+
|
|
|
+ openSnackBar(successTrans('create', { name: t('partition') }));
|
|
|
+ handleCloseDialog();
|
|
|
+ // refresh partitions
|
|
|
+ fetchPartitions(collectionName);
|
|
|
+ };
|
|
|
+
|
|
|
+ return (
|
|
|
+ <section className={classes.wrapper}>
|
|
|
+ <MilvusGrid
|
|
|
+ toolbarConfigs={toolbarConfigs}
|
|
|
+ colDefinitions={colDefinitions}
|
|
|
+ rows={partitionList}
|
|
|
+ rowCount={total}
|
|
|
+ primaryKey="id"
|
|
|
+ openCheckBox={true}
|
|
|
+ showHoverStyle={true}
|
|
|
+ selected={selectedPartitions}
|
|
|
+ setSelected={handleSelectChange}
|
|
|
+ page={currentPage}
|
|
|
+ onChangePage={handlePageChange}
|
|
|
+ rowsPerPage={pageSize}
|
|
|
+ setRowsPerPage={handlePageSize}
|
|
|
+ isLoading={loading}
|
|
|
+ />
|
|
|
+ </section>
|
|
|
+ );
|
|
|
+};
|
|
|
+
|
|
|
+export default Partitions;
|