fix load collection dialog

Signed-off-by: ryjiang <jiangruiyi@gmail.com>

client/src/pages/collections/Collections.tsx

@@ -145,7 +145,7 @@ const Collections = () => {
           params: {
             component: (
               <LoadCollectionDialog
-                collection={selectedCollections[0].collection_name}
+                collection={selectedCollections[0]}
                 onLoad={async () => {
                   setSelectedCollections([]);
                 }}
@@ -375,9 +375,7 @@ const Collections = () => {
                 params: {
                   component:
                     v.status === LOADING_STATE.UNLOADED ? (
-                      <LoadCollectionDialog
-                        collectionName={v.collection_name}
-                      />
+                      <LoadCollectionDialog collection={v} />
                     ) : (
                       <ReleaseCollectionDialog collection={v} />
                     ),

client/src/pages/dialogs/LoadCollectionDialog.tsx

@@ -17,6 +17,7 @@ import { ITextfieldConfig } from '@/components/customInput/Types';
 import DialogTemplate from '@/components/customDialog/DialogTemplate';
 import CustomToolTip from '@/components/customToolTip/CustomToolTip';
 import icons from '@/components/icons/Icons';
+import { CollectionObject } from '@server/types';
 
 const useStyles = makeStyles((theme: Theme) => ({
   desc: {
@@ -36,10 +37,13 @@ const useStyles = makeStyles((theme: Theme) => ({
   },
 }));
 
-const LoadCollectionDialog = (props: any) => {
+const LoadCollectionDialog = (props: {
+  collection: CollectionObject;
+  onLoad?: (collection: CollectionObject) => void;
+}) => {
   const { loadCollection } = useContext(dataContext);
   const classes = useStyles();
-  const { collectionName, onLoad } = props;
+  const { collection, onLoad } = props;
   const { t: dialogTrans } = useTranslation('dialog');
   const { t: collectionTrans } = useTranslation('collection');
   const { t: successTrans } = useTranslation('success');
@@ -102,7 +106,7 @@ const LoadCollectionDialog = (props: any) => {
     }
 
     // load collection request
-    await loadCollection(collectionName, params);
+    await loadCollection(collection.collection_name, params);
 
     // show success message
     openSnackBar(
@@ -113,7 +117,7 @@ const LoadCollectionDialog = (props: any) => {
 
     // callback
     if (onLoad) {
-      await onLoad(collectionName);
+      await onLoad(collection);
     }
     // close dialog
     handleCloseDialog();
@@ -175,7 +179,7 @@ const LoadCollectionDialog = (props: any) => {
   return (
     <DialogTemplate
       title={dialogTrans('loadTitle', {
-        type: collectionName,
+        type: collection.collection_name,
       })}
       handleClose={handleCloseDialog}
       children={

client/src/pages/schema/Schema.tsx

@@ -266,13 +266,9 @@ const Schema = () => {
                   params: {
                     component:
                       collection.status === LOADING_STATE.UNLOADED ? (
-                        <LoadCollectionDialog
-                          collectionName={collection.collection_name}
-                        />
+                        <LoadCollectionDialog collection={collection} />
                       ) : (
-                        <ReleaseCollectionDialog
-                          collectionName={collection.collection_name}
-                        />
+                        <ReleaseCollectionDialog collection={collection} />
                       ),
                   },
                 });