github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-12 20:53:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat[Datasource]:add summary refresh (#1433)

Browse Source 
Co-authored-by: hzh97 <2976151305@qq.com>
This commit is contained in:
Aries-ckt
2024-04-20 10:04:12 +08:00
committed by GitHub
parent 57be1ece18
commit b49b07f011
175 changed files with 61020 additions and 1494 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
web/pages/database/index.tsx
View File

@@ -1,9 +1,9 @@
import React, { useMemo, useState } from 'react';
import { useAsyncEffect } from 'ahooks';
import { Badge, Button, Card, Drawer, Empty, Modal, message } from 'antd';
import { Badge, Button, Card, Drawer, Empty, Modal, Spin, message } from 'antd';
import FormDialog from '@/components/database/form-dialog';
import { apiInterceptors, getDbList, getDbSupportType, postDbDelete } from '@/client/api';
import { DeleteFilled, EditFilled, PlusOutlined } from '@ant-design/icons';
import { apiInterceptors, getDbList, getDbSupportType, postDbDelete, postDbRefresh } from '@/client/api';
import { DeleteFilled, EditFilled, PlusOutlined, RedoOutlined } from '@ant-design/icons';
import { DBOption, DBType, DbListResponse, DbSupportTypeResponse } from '@/types/db';
import MuiLoading from '@/components/common/loading';
import { dbMapper } from '@/utils';
@@ -24,6 +24,7 @@ function Database() {
const [loading, setLoading] = useState(false);
const [modal, setModal] = useState<{ open: boolean; info?: DBItem; dbType?: DBType }>({ open: false });
const [draw, setDraw] = useState<{ open: boolean; dbList?: DbListResponse; name?: string; type?: DBType }>({ open: false });
const [refreshLoading, setRefreshLoading] = useState(false);
const getDbSupportList = async () => {
const [, data] = await apiInterceptors(getDbSupportType());
@@ -77,6 +78,13 @@ function Database() {
});
};
const onRefresh = async (item: DBItem) => {
setRefreshLoading(true);
const [, res] = await apiInterceptors(postDbRefresh({ db_name: item.db_name, db_type: item.db_type }));
if (res) message.success(t('refreshSuccess'));
setRefreshLoading(false);
};
const dbListByType = useMemo(() => {
const mapper = dbTypeList.reduce((acc, item) => {
acc[item.value] = dbList.filter((dbConn) => dbConn.db_type === item.value);
@@ -150,7 +158,7 @@ function Database() {
open={draw.open}
>
{draw.type && dbListByType[draw.type] && dbListByType[draw.type].length ? (
<>
<Spin spinning={refreshLoading}>
<Button
type="primary"
className="mb-4 flex items-center"
@@ -166,9 +174,14 @@ function Database() {
key={item.db_name}
title={item.db_name}
extra={
<>
<div className="flex items-center gap-3">
<RedoOutlined
style={{ color: 'gray' }}
onClick={() => {
onRefresh(item);
}}
/>
<EditFilled
className="mr-2"
style={{ color: '#1b7eff' }}
onClick={() => {
onModify(item);
@@ -180,7 +193,7 @@ function Database() {
onDelete(item);
}}
/>
</>
</div>
}
className="mb-4"
>
@@ -196,7 +209,7 @@ function Database() {
<p>remark: {item.comment}</p>
</Card>
))}
</>
</Spin>
) : (
<Empty image={Empty.PRESENTED_IMAGE_DEFAULT}>
<Button