fix: fetch the file list after uploading the file by @tanstack/react-query #1306 (#1654)

### What problem does this PR solve?
fix: fetch the file list after uploading the file by
@tanstack/react-query #1306

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
balibabu 2024-07-23 17:53:15 +08:00 committed by GitHub
parent 022afbb39d
commit 80d703f9c2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
7 changed files with 136 additions and 405 deletions

View File

@ -1,13 +1,12 @@
import { ResponseType } from '@/interfaces/database/base'; import { ResponseType } from '@/interfaces/database/base';
import { import { IFolder } from '@/interfaces/database/file-manager';
IConnectRequestBody, import { IConnectRequestBody } from '@/interfaces/request/file-manager';
IFileListRequestBody,
} from '@/interfaces/request/file-manager';
import fileManagerService from '@/services/file-manager-service'; import fileManagerService from '@/services/file-manager-service';
import { useMutation, useQuery } from '@tanstack/react-query'; import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
import { PaginationProps, UploadFile } from 'antd'; import { PaginationProps, UploadFile, message } from 'antd';
import React, { useCallback } from 'react'; import React, { useCallback } from 'react';
import { useDispatch, useSearchParams, useSelector } from 'umi'; import { useTranslation } from 'react-i18next';
import { useSearchParams } from 'umi';
import { useGetNextPagination, useHandleSearchChange } from './logic-hooks'; import { useGetNextPagination, useHandleSearchChange } from './logic-hooks';
import { useSetPaginationParams } from './route-hook'; import { useSetPaginationParams } from './route-hook';
@ -18,41 +17,22 @@ export const useGetFolderId = () => {
return id ?? ''; return id ?? '';
}; };
export const useFetchFileList = () => {
const dispatch = useDispatch();
const fetchFileList = useCallback(
(payload: IFileListRequestBody) => {
return dispatch<any>({
type: 'fileManager/listFile',
payload,
});
},
[dispatch],
);
return fetchFileList;
};
export interface IListResult { export interface IListResult {
searchString: string; searchString: string;
handleInputChange: React.ChangeEventHandler<HTMLInputElement>; handleInputChange: React.ChangeEventHandler<HTMLInputElement>;
pagination: PaginationProps; pagination: PaginationProps;
setPagination: (pagination: { page: number; pageSize: number }) => void; setPagination: (pagination: { page: number; pageSize: number }) => void;
loading: boolean;
} }
export const useFetchNextFileList = (): ResponseType<any> & IListResult => { export const useFetchFileList = (): ResponseType<any> & IListResult => {
const { searchString, handleInputChange } = useHandleSearchChange(); const { searchString, handleInputChange } = useHandleSearchChange();
const { pagination, setPagination } = useGetNextPagination(); const { pagination, setPagination } = useGetNextPagination();
const id = useGetFolderId(); const id = useGetFolderId();
const { data } = useQuery({ const { data, isFetching: loading } = useQuery({
queryKey: [ queryKey: [
'fetchFileList', 'fetchFileList',
// pagination.current,
// id,
// pagination.pageSize,
// searchString,
{ {
id, id,
searchString, searchString,
@ -87,27 +67,14 @@ export const useFetchNextFileList = (): ResponseType<any> & IListResult => {
handleInputChange: onInputChange, handleInputChange: onInputChange,
pagination: { ...pagination, total: data?.data?.total }, pagination: { ...pagination, total: data?.data?.total },
setPagination, setPagination,
loading,
}; };
}; };
export const useRemoveFile = () => {
const dispatch = useDispatch();
const removeFile = useCallback(
(fileIds: string[], parentId: string) => {
return dispatch<any>({
type: 'fileManager/removeFile',
payload: { fileIds, parentId },
});
},
[dispatch],
);
return removeFile;
};
export const useDeleteFile = () => { export const useDeleteFile = () => {
const { setPaginationParams } = useSetPaginationParams(); const { setPaginationParams } = useSetPaginationParams();
const queryClient = useQueryClient();
const { const {
data, data,
isPending: loading, isPending: loading,
@ -117,9 +84,10 @@ export const useDeleteFile = () => {
mutationFn: async (params: { fileIds: string[]; parentId: string }) => { mutationFn: async (params: { fileIds: string[]; parentId: string }) => {
const { data } = await fileManagerService.removeFile(params); const { data } = await fileManagerService.removeFile(params);
if (data.retcode === 0) { if (data.retcode === 0) {
setPaginationParams(1); setPaginationParams(1); // TODO: There should be a better way to paginate the request list
queryClient.invalidateQueries({ queryKey: ['fetchFileList'] });
} }
return data?.data ?? {}; return data.retcode;
}, },
}); });
@ -127,106 +95,129 @@ export const useDeleteFile = () => {
}; };
export const useRenameFile = () => { export const useRenameFile = () => {
const dispatch = useDispatch(); const queryClient = useQueryClient();
const { t } = useTranslation();
const renameFile = useCallback( const {
(fileId: string, name: string, parentId: string) => { data,
return dispatch<any>({ isPending: loading,
type: 'fileManager/renameFile', mutateAsync,
payload: { fileId, name, parentId }, } = useMutation({
}); mutationKey: ['renameFile'],
mutationFn: async (params: { fileId: string; name: string }) => {
const { data } = await fileManagerService.renameFile(params);
if (data.retcode === 0) {
message.success(t('message.renamed'));
queryClient.invalidateQueries({ queryKey: ['fetchFileList'] });
}
return data.retcode;
}, },
[dispatch], });
);
return renameFile; return { data, loading, renameFile: mutateAsync };
}; };
export const useFetchParentFolderList = () => { export const useFetchParentFolderList = (): IFolder[] => {
const dispatch = useDispatch(); const id = useGetFolderId();
const { data } = useQuery({
const fetchParentFolderList = useCallback( queryKey: ['fetchParentFolderList', id],
(fileId: string) => { initialData: [],
return dispatch<any>({ enabled: !!id,
type: 'fileManager/getAllParentFolder', queryFn: async () => {
payload: { fileId }, const { data } = await fileManagerService.getAllParentFolder({
fileId: id,
}); });
},
[dispatch],
);
return fetchParentFolderList; return data?.data?.parent_folders?.toReversed() ?? [];
},
});
return data;
}; };
export const useCreateFolder = () => { export const useCreateFolder = () => {
const dispatch = useDispatch(); const { setPaginationParams } = useSetPaginationParams();
const queryClient = useQueryClient();
const { t } = useTranslation();
const createFolder = useCallback( const {
(parentId: string, name: string) => { data,
return dispatch<any>({ isPending: loading,
type: 'fileManager/createFolder', mutateAsync,
payload: { parentId, name, type: 'folder' }, } = useMutation({
mutationKey: ['createFolder'],
mutationFn: async (params: { parentId: string; name: string }) => {
const { data } = await fileManagerService.createFolder({
...params,
type: 'folder',
}); });
if (data.retcode === 0) {
message.success(t('message.created'));
setPaginationParams(1);
queryClient.invalidateQueries({ queryKey: ['fetchFileList'] });
}
return data.retcode;
}, },
[dispatch], });
);
return createFolder; return { data, loading, createFolder: mutateAsync };
};
export const useSelectFileList = () => {
const fileList = useSelector((state) => state.fileManager.fileList);
return fileList;
};
export const useSelectParentFolderList = () => {
const parentFolderList = useSelector(
(state) => state.fileManager.parentFolderList,
);
return parentFolderList.toReversed();
}; };
export const useUploadFile = () => { export const useUploadFile = () => {
const dispatch = useDispatch(); const { setPaginationParams } = useSetPaginationParams();
const { t } = useTranslation();
const uploadFile = useCallback( const queryClient = useQueryClient();
(fileList: UploadFile[], parentId: string) => { const {
try { data,
return dispatch<any>({ isPending: loading,
type: 'fileManager/uploadFile', mutateAsync,
payload: { } = useMutation({
file: fileList, mutationKey: ['uploadFile'],
parentId, mutationFn: async (params: {
path: fileList.map((file) => (file as any).webkitRelativePath), fileList: UploadFile[];
}, parentId: string;
}); }) => {
} catch (errorInfo) { const fileList = params.fileList;
console.log('Failed:', errorInfo); const pathList = params.fileList.map(
(file) => (file as any).webkitRelativePath,
);
const formData = new FormData();
formData.append('parent_id', params.parentId);
fileList.forEach((file: any, index: number) => {
formData.append('file', file);
formData.append('path', pathList[index]);
});
const { data } = await fileManagerService.uploadFile(formData);
if (data.retcode === 0) {
message.success(t('message.uploaded'));
setPaginationParams(1);
queryClient.invalidateQueries({ queryKey: ['fetchFileList'] });
} }
return data.retcode;
}, },
[dispatch], });
);
return uploadFile; return { data, loading, uploadFile: mutateAsync };
}; };
export const useConnectToKnowledge = () => { export const useConnectToKnowledge = () => {
const dispatch = useDispatch(); const queryClient = useQueryClient();
const { t } = useTranslation();
const uploadFile = useCallback( const {
(payload: IConnectRequestBody) => { data,
try { isPending: loading,
return dispatch<any>({ mutateAsync,
type: 'fileManager/connectFileToKnowledge', } = useMutation({
payload, mutationKey: ['connectFileToKnowledge'],
}); mutationFn: async (params: IConnectRequestBody) => {
} catch (errorInfo) { const { data } = await fileManagerService.connectFileToKnowledge(params);
console.log('Failed:', errorInfo); if (data.retcode === 0) {
message.success(t('message.operated'));
queryClient.invalidateQueries({ queryKey: ['fetchFileList'] });
} }
return data.retcode;
}, },
[dispatch], });
);
return uploadFile; return { data, loading, connectFileToKnowledge: mutateAsync };
}; };

View File

@ -8,7 +8,7 @@ interface BaseRequestBody {
parentId: string; parentId: string;
} }
export interface IConnectRequestBody extends BaseRequestBody { export interface IConnectRequestBody {
fileIds: string[]; fileIds: string[];
kbIds: string[]; kbIds: string[];
} }

View File

@ -26,7 +26,7 @@ import {
import { import {
IListResult, IListResult,
useSelectParentFolderList, useFetchParentFolderList,
} from '@/hooks/file-manager-hooks'; } from '@/hooks/file-manager-hooks';
import styles from './index.less'; import styles from './index.less';
@ -49,7 +49,7 @@ const FileToolbar = ({
const { t } = useTranslate('knowledgeDetails'); const { t } = useTranslate('knowledgeDetails');
const breadcrumbItems = useSelectBreadcrumbItems(); const breadcrumbItems = useSelectBreadcrumbItems();
const { handleBreadcrumbClick } = useHandleBreadcrumbClick(); const { handleBreadcrumbClick } = useHandleBreadcrumbClick();
const parentFolderList = useSelectParentFolderList(); const parentFolderList = useFetchParentFolderList();
const isKnowledgeBase = const isKnowledgeBase =
parentFolderList.at(-1)?.source_type === 'knowledgebase'; parentFolderList.at(-1)?.source_type === 'knowledgebase';

View File

@ -3,21 +3,15 @@ import {
useConnectToKnowledge, useConnectToKnowledge,
useCreateFolder, useCreateFolder,
useDeleteFile, useDeleteFile,
useFetchFileList,
useFetchParentFolderList, useFetchParentFolderList,
useRenameFile, useRenameFile,
useSelectFileList,
useSelectParentFolderList,
useUploadFile, useUploadFile,
} from '@/hooks/file-manager-hooks'; } from '@/hooks/file-manager-hooks';
import { useGetPagination, useSetPagination } from '@/hooks/logic-hooks';
import { useOneNamespaceEffectsLoading } from '@/hooks/store-hooks';
import { IFile } from '@/interfaces/database/file-manager'; import { IFile } from '@/interfaces/database/file-manager';
import { PaginationProps } from 'antd';
import { TableRowSelection } from 'antd/es/table/interface'; import { TableRowSelection } from 'antd/es/table/interface';
import { UploadFile } from 'antd/lib'; import { UploadFile } from 'antd/lib';
import { useCallback, useEffect, useMemo, useState } from 'react'; import { useCallback, useMemo, useState } from 'react';
import { useDispatch, useNavigate, useSearchParams, useSelector } from 'umi'; import { useNavigate, useSearchParams } from 'umi';
export const useGetFolderId = () => { export const useGetFolderId = () => {
const [searchParams] = useSearchParams(); const [searchParams] = useSearchParams();
@ -26,71 +20,6 @@ export const useGetFolderId = () => {
return id ?? ''; return id ?? '';
}; };
export const useFetchDocumentListOnMount = () => {
const fetchDocumentList = useFetchFileList();
const fileList = useSelectFileList();
const id = useGetFolderId();
const { searchString, pagination } = useSelector(
(state) => state.fileManager,
);
const { pageSize, current } = pagination;
const dispatch = useDispatch();
useEffect(() => {
fetchDocumentList({
parent_id: id,
keywords: searchString,
page_size: pageSize,
page: current,
});
}, [dispatch, fetchDocumentList, id, current, pageSize, searchString]);
return { fetchDocumentList, fileList };
};
export const useGetFilesPagination = () => {
const { pagination } = useSelector((state) => state.fileManager);
const setPagination = useSetPagination('fileManager');
const onPageChange: PaginationProps['onChange'] = useCallback(
(pageNumber: number, pageSize: number) => {
setPagination(pageNumber, pageSize);
},
[setPagination],
);
const { pagination: paginationInfo } = useGetPagination(
pagination.total,
pagination.current,
pagination.pageSize,
onPageChange,
);
return {
pagination: paginationInfo,
setPagination,
};
};
export const useHandleSearchChange = () => {
const dispatch = useDispatch();
const { searchString } = useSelector((state) => state.fileManager);
const setPagination = useSetPagination('fileManager');
const handleInputChange = useCallback(
(e: React.ChangeEvent<HTMLInputElement | HTMLTextAreaElement>) => {
const value = e.target.value;
dispatch({ type: 'fileManager/setSearchString', payload: value });
setPagination();
},
[setPagination, dispatch],
);
return { handleInputChange, searchString };
};
export const useGetRowSelection = () => { export const useGetRowSelection = () => {
const [selectedRowKeys, setSelectedRowKeys] = useState<React.Key[]>([]); const [selectedRowKeys, setSelectedRowKeys] = useState<React.Key[]>([]);
@ -126,11 +55,14 @@ export const useRenameCurrentFile = () => {
hideModal: hideFileRenameModal, hideModal: hideFileRenameModal,
showModal: showFileRenameModal, showModal: showFileRenameModal,
} = useSetModalState(); } = useSetModalState();
const renameFile = useRenameFile(); const { renameFile, loading } = useRenameFile();
const onFileRenameOk = useCallback( const onFileRenameOk = useCallback(
async (name: string) => { async (name: string) => {
const ret = await renameFile(file.id, name, file.parent_id); const ret = await renameFile({
fileId: file.id,
name,
});
if (ret === 0) { if (ret === 0) {
hideFileRenameModal(); hideFileRenameModal();
@ -139,8 +71,6 @@ export const useRenameCurrentFile = () => {
[renameFile, file, hideFileRenameModal], [renameFile, file, hideFileRenameModal],
); );
const loading = useOneNamespaceEffectsLoading('fileManager', ['renameFile']);
const handleShowFileRenameModal = useCallback( const handleShowFileRenameModal = useCallback(
async (record: IFile) => { async (record: IFile) => {
setFile(record); setFile(record);
@ -160,15 +90,7 @@ export const useRenameCurrentFile = () => {
}; };
export const useSelectBreadcrumbItems = () => { export const useSelectBreadcrumbItems = () => {
const parentFolderList = useSelectParentFolderList(); const parentFolderList = useFetchParentFolderList();
const id = useGetFolderId();
const fetchParentFolderList = useFetchParentFolderList();
useEffect(() => {
if (id) {
fetchParentFolderList(id);
}
}, [id, fetchParentFolderList]);
return parentFolderList.length === 1 return parentFolderList.length === 1
? [] ? []
@ -184,12 +106,12 @@ export const useHandleCreateFolder = () => {
hideModal: hideFolderCreateModal, hideModal: hideFolderCreateModal,
showModal: showFolderCreateModal, showModal: showFolderCreateModal,
} = useSetModalState(); } = useSetModalState();
const createFolder = useCreateFolder(); const { createFolder, loading } = useCreateFolder();
const id = useGetFolderId(); const id = useGetFolderId();
const onFolderCreateOk = useCallback( const onFolderCreateOk = useCallback(
async (name: string) => { async (name: string) => {
const ret = await createFolder(id, name); const ret = await createFolder({ parentId: id, name });
if (ret === 0) { if (ret === 0) {
hideFolderCreateModal(); hideFolderCreateModal();
@ -198,10 +120,6 @@ export const useHandleCreateFolder = () => {
[createFolder, hideFolderCreateModal, id], [createFolder, hideFolderCreateModal, id],
); );
const loading = useOneNamespaceEffectsLoading('fileManager', [
'createFolder',
]);
return { return {
folderCreateLoading: loading, folderCreateLoading: loading,
onFolderCreateOk, onFolderCreateOk,
@ -234,24 +152,19 @@ export const useHandleDeleteFile = (
return { handleRemoveFile }; return { handleRemoveFile };
}; };
export const useSelectFileListLoading = () => {
return useOneNamespaceEffectsLoading('fileManager', ['listFile']);
};
export const useHandleUploadFile = () => { export const useHandleUploadFile = () => {
const { const {
visible: fileUploadVisible, visible: fileUploadVisible,
hideModal: hideFileUploadModal, hideModal: hideFileUploadModal,
showModal: showFileUploadModal, showModal: showFileUploadModal,
} = useSetModalState(); } = useSetModalState();
const uploadFile = useUploadFile(); const { uploadFile, loading } = useUploadFile();
const id = useGetFolderId(); const id = useGetFolderId();
const onFileUploadOk = useCallback( const onFileUploadOk = useCallback(
async (fileList: UploadFile[]): Promise<number | undefined> => { async (fileList: UploadFile[]): Promise<number | undefined> => {
if (fileList.length > 0) { if (fileList.length > 0) {
const ret: number = await uploadFile(fileList, id); const ret: number = await uploadFile({ fileList, parentId: id });
console.info(ret);
if (ret === 0) { if (ret === 0) {
hideFileUploadModal(); hideFileUploadModal();
} }
@ -261,8 +174,6 @@ export const useHandleUploadFile = () => {
[uploadFile, hideFileUploadModal, id], [uploadFile, hideFileUploadModal, id],
); );
const loading = useOneNamespaceEffectsLoading('fileManager', ['uploadFile']);
return { return {
fileUploadLoading: loading, fileUploadLoading: loading,
onFileUploadOk, onFileUploadOk,
@ -278,8 +189,8 @@ export const useHandleConnectToKnowledge = () => {
hideModal: hideConnectToKnowledgeModal, hideModal: hideConnectToKnowledgeModal,
showModal: showConnectToKnowledgeModal, showModal: showConnectToKnowledgeModal,
} = useSetModalState(); } = useSetModalState();
const connectToKnowledge = useConnectToKnowledge(); const { connectFileToKnowledge: connectToKnowledge, loading } =
const id = useGetFolderId(); useConnectToKnowledge();
const [record, setRecord] = useState<IFile>({} as IFile); const [record, setRecord] = useState<IFile>({} as IFile);
const initialValue = useMemo(() => { const initialValue = useMemo(() => {
@ -291,7 +202,6 @@ export const useHandleConnectToKnowledge = () => {
const onConnectToKnowledgeOk = useCallback( const onConnectToKnowledgeOk = useCallback(
async (knowledgeIds: string[]) => { async (knowledgeIds: string[]) => {
const ret = await connectToKnowledge({ const ret = await connectToKnowledge({
parentId: id,
fileIds: [record.id], fileIds: [record.id],
kbIds: knowledgeIds, kbIds: knowledgeIds,
}); });
@ -301,13 +211,9 @@ export const useHandleConnectToKnowledge = () => {
} }
return ret; return ret;
}, },
[connectToKnowledge, hideConnectToKnowledgeModal, id, record.id], [connectToKnowledge, hideConnectToKnowledgeModal, record.id],
); );
const loading = useOneNamespaceEffectsLoading('fileManager', [
'connectFileToKnowledge',
]);
const handleShowConnectToKnowledgeModal = useCallback( const handleShowConnectToKnowledgeModal = useCallback(
(record: IFile) => { (record: IFile) => {
setRecord(record); setRecord(record);
@ -328,16 +234,14 @@ export const useHandleConnectToKnowledge = () => {
export const useHandleBreadcrumbClick = () => { export const useHandleBreadcrumbClick = () => {
const navigate = useNavigate(); const navigate = useNavigate();
const setPagination = useSetPagination('fileManager');
const handleBreadcrumbClick = useCallback( const handleBreadcrumbClick = useCallback(
(path?: string) => { (path?: string) => {
if (path) { if (path) {
setPagination();
navigate(path); navigate(path);
} }
}, },
[setPagination, navigate], [navigate],
); );
return { handleBreadcrumbClick }; return { handleBreadcrumbClick };

View File

@ -1,4 +1,4 @@
import { useFetchNextFileList } from '@/hooks/file-manager-hooks'; import { useFetchFileList } from '@/hooks/file-manager-hooks';
import { IFile } from '@/interfaces/database/file-manager'; import { IFile } from '@/interfaces/database/file-manager';
import { formatDate } from '@/utils/date'; import { formatDate } from '@/utils/date';
import { Button, Flex, Space, Table, Tag, Typography } from 'antd'; import { Button, Flex, Space, Table, Tag, Typography } from 'antd';
@ -12,7 +12,6 @@ import {
useHandleUploadFile, useHandleUploadFile,
useNavigateToOtherFolder, useNavigateToOtherFolder,
useRenameCurrentFile, useRenameCurrentFile,
useSelectFileListLoading,
} from './hooks'; } from './hooks';
import FileUploadModal from '@/components/file-upload-modal'; import FileUploadModal from '@/components/file-upload-modal';
@ -31,7 +30,6 @@ const FileManager = () => {
const { t } = useTranslate('fileManager'); const { t } = useTranslate('fileManager');
// const fileList = useSelectFileList(); // const fileList = useSelectFileList();
const { rowSelection, setSelectedRowKeys } = useGetRowSelection(); const { rowSelection, setSelectedRowKeys } = useGetRowSelection();
const loading = useSelectFileListLoading();
const navigateToOtherFolder = useNavigateToOtherFolder(); const navigateToOtherFolder = useNavigateToOtherFolder();
const { const {
fileRenameVisible, fileRenameVisible,
@ -64,8 +62,8 @@ const FileManager = () => {
connectToKnowledgeLoading, connectToKnowledgeLoading,
} = useHandleConnectToKnowledge(); } = useHandleConnectToKnowledge();
// const { pagination } = useGetFilesPagination(); // const { pagination } = useGetFilesPagination();
const { pagination, data, searchString, handleInputChange } = const { pagination, data, searchString, handleInputChange, loading } =
useFetchNextFileList(); useFetchFileList();
const columns: ColumnsType<IFile> = [ const columns: ColumnsType<IFile> = [
{ {
title: t('name'), title: t('name'),

View File

@ -1,159 +0,0 @@
import { paginationModel } from '@/base';
import { BaseState } from '@/interfaces/common';
import { IFile, IFolder } from '@/interfaces/database/file-manager';
import i18n from '@/locales/config';
import fileManagerService, {
getDocumentFile,
} from '@/services/file-manager-service';
import { message } from 'antd';
import omit from 'lodash/omit';
import { DvaModel } from 'umi';
export interface FileManagerModelState extends BaseState {
fileList: IFile[];
parentFolderList: IFolder[];
}
const model: DvaModel<FileManagerModelState> = {
namespace: 'fileManager',
state: {
fileList: [],
parentFolderList: [],
...(paginationModel.state as BaseState),
},
reducers: {
setFileList(state, { payload }) {
return { ...state, fileList: payload };
},
setParentFolderList(state, { payload }) {
return { ...state, parentFolderList: payload };
},
...paginationModel.reducers,
},
effects: {
*removeFile({ payload = {} }, { call, put }) {
const { data } = yield call(fileManagerService.removeFile, {
fileIds: payload.fileIds,
});
const { retcode } = data;
if (retcode === 0) {
message.success(i18n.t('message.deleted'));
yield put({
type: 'listFile',
payload: { parentId: payload.parentId },
});
}
return retcode;
},
*listFile({ payload = {} }, { call, put, select }) {
const { searchString, pagination } = yield select(
(state: any) => state.fileManager,
);
const { current, pageSize } = pagination;
const { data } = yield call(fileManagerService.listFile, {
...payload,
keywords: searchString.trim(),
page: current,
pageSize,
});
const { retcode, data: res } = data;
if (retcode === 0 && Array.isArray(res.files)) {
yield put({
type: 'setFileList',
payload: res.files,
});
yield put({
type: 'setPagination',
payload: { total: res.total },
});
}
},
*renameFile({ payload = {} }, { call, put }) {
const { data } = yield call(
fileManagerService.renameFile,
omit(payload, ['parentId']),
);
if (data.retcode === 0) {
message.success(i18n.t('message.renamed'));
yield put({
type: 'listFile',
payload: { parentId: payload.parentId },
});
}
return data.retcode;
},
*uploadFile({ payload = {} }, { call, put }) {
const fileList = payload.file;
const pathList = payload.path;
const formData = new FormData();
formData.append('parent_id', payload.parentId);
fileList.forEach((file: any, index: number) => {
formData.append('file', file);
formData.append('path', pathList[index]);
});
const { data } = yield call(fileManagerService.uploadFile, formData);
if (data.retcode === 0) {
message.success(i18n.t('message.uploaded'));
yield put({
type: 'listFile',
payload: { parentId: payload.parentId },
});
}
return data.retcode;
},
*createFolder({ payload = {} }, { call, put }) {
const { data } = yield call(fileManagerService.createFolder, payload);
if (data.retcode === 0) {
message.success(i18n.t('message.created'));
yield put({
type: 'listFile',
payload: { parentId: payload.parentId },
});
}
return data.retcode;
},
*getAllParentFolder({ payload = {} }, { call, put }) {
const { data } = yield call(
fileManagerService.getAllParentFolder,
payload,
);
if (data.retcode === 0) {
yield put({
type: 'setParentFolderList',
payload: data.data?.parent_folders ?? [],
});
}
return data.retcode;
},
*connectFileToKnowledge({ payload = {} }, { call, put }) {
const { data } = yield call(
fileManagerService.connectFileToKnowledge,
omit(payload, 'parentId'),
);
if (data.retcode === 0) {
message.success(i18n.t('message.operated'));
yield put({
type: 'listFile',
payload: { parentId: payload.parentId },
});
}
return data.retcode;
},
*getDocumentFile({ payload = {} }, { call }) {
const ret = yield call(getDocumentFile, payload);
return ret;
},
},
};
// const finalModel = modelExtend<DvaModel<FileManagerModelState & BaseState>>(
// paginationModel,
// model,
// );
// console.info(finalModel);
export default model;

3
web/typings.d.ts vendored
View File

@ -3,7 +3,6 @@ import { KFModelState } from '@/pages/add-knowledge/components/knowledge-file/mo
import { TestingModelState } from '@/pages/add-knowledge/components/knowledge-testing/model'; import { TestingModelState } from '@/pages/add-knowledge/components/knowledge-testing/model';
import { kAModelState } from '@/pages/add-knowledge/model'; import { kAModelState } from '@/pages/add-knowledge/model';
import { ChatModelState } from '@/pages/chat/model'; import { ChatModelState } from '@/pages/chat/model';
import { FileManagerModelState } from '@/pages/file-manager/model';
import { LoginModelState } from '@/pages/login/model'; import { LoginModelState } from '@/pages/login/model';
import { SettingModelState } from '@/pages/user-setting/model'; import { SettingModelState } from '@/pages/user-setting/model';
@ -15,8 +14,6 @@ function useSelector<TState = RootState, TSelected = unknown>(
): TSelected; ): TSelected;
export interface RootState { export interface RootState {
// loading: Loading;
fileManager: FileManagerModelState;
chatModel: ChatModelState; chatModel: ChatModelState;
loginModel: LoginModelState; loginModel: LoginModelState;
settingModel: SettingModelState; settingModel: SettingModelState;