(feature)(chat-sdk) add unit (#2067)

webapp/packages/supersonic-fe/src/pages/SemanticModel/View/components/DataSetTable.tsx

@@ -12,6 +12,7 @@ import { ISemantic } from '../../data';
 import { ColumnsConfig } from '../../components/TableColumnRender';
 import ViewSearchFormModal from './ViewSearchFormModal';
 import { toDatasetEditPage } from '@/pages/SemanticModel/utils';
+import UploadFile from './UploadFile';
 
 type Props = {
   // dataSetList: ISemantic.IDatasetItem[];
@@ -92,9 +93,6 @@ const DataSetTable: React.FC<Props> = ({ disabledEdit = false }) => {
           <a
             onClick={() => {
               toDatasetEditPage(record.domainId, record.id, 'relation');
-              // setEditFormStep(1);
-              // setViewItem(record);
-              // setCreateDataSourceModalOpen(true);
             }}
           >
             {name}
@@ -146,9 +144,6 @@ const DataSetTable: React.FC<Props> = ({ disabledEdit = false }) => {
               key="metricEditBtn"
               onClick={() => {
                 toDatasetEditPage(record.domainId, record.id);
-                // setEditFormStep(0);
-                // setViewItem(record);
-                // setCreateDataSourceModalOpen(true);
               }}
             >
               编辑
@@ -189,6 +184,12 @@ const DataSetTable: React.FC<Props> = ({ disabledEdit = false }) => {
                 启用
               </Button>
             )}
+            <UploadFile
+              key="uploadFile"
+              buttonType="link"
+              domainId={record.domainId}
+              datasetId={record.id}
+            />
             <Popconfirm
               title="确认删除？"
               okText="是"
@@ -229,6 +230,13 @@ const DataSetTable: React.FC<Props> = ({ disabledEdit = false }) => {
           disabledEdit
             ? [<></>]
             : [
+                <UploadFile
+                  key="uploadFile"
+                  domainId={selectDomainId}
+                  onFileUploaded={() => {
+                    queryDataSetList();
+                  }}
+                />,
                 <Button
                   key="create"
                   type="primary"

webapp/packages/supersonic-fe/src/pages/SemanticModel/View/components/UploadFile.tsx

@@ -0,0 +1,44 @@
+import { getToken } from '@/utils/utils';
+import { UploadOutlined } from '@ant-design/icons';
+import type { UploadProps } from 'antd';
+import { Button, message, Upload } from 'antd';
+
+type Props = {
+  buttonType?: string;
+  domainId?: number;
+  datasetId?: string;
+  onFileUploaded?: () => void;
+};
+
+const UploadFile = ({ buttonType, domainId, datasetId, onFileUploaded }: Props) => {
+  const props: UploadProps = {
+    name: 'multipartFile',
+    action: `/aibi/api/data/file/uploadFileNew?type=DATASET&domainId=${domainId}${
+      datasetId ? `&dataSetId=${datasetId}` : ''
+    }`,
+    showUploadList: false,
+    onChange(info) {
+      if (info.file.status !== 'uploading') {
+        console.log(info.file, info.fileList);
+      }
+      if (info.file.status === 'done') {
+        message.success('导入成功');
+        onFileUploaded?.();
+      } else if (info.file.status === 'error') {
+        message.error('导入失败');
+      }
+    },
+  };
+
+  return (
+    <Upload {...props}>
+      {buttonType === 'link' ? (
+        <a>导入文件</a>
+      ) : (
+        <Button icon={<UploadOutlined />}>导入文件</Button>
+      )}
+    </Upload>
+  );
+};
+
+export default UploadFile;