资产目录功能统一

src/view/Manage/AssetManage/Component/AssetTree.jsx

@@ -387,7 +387,7 @@ const AssetTree = (props) => {
             payload: {data: [ currentRightClickDir.nodeId ]},
             callback: () => {
               showMessage("success","删除成功");
-              getAllDirectoryAsTree(true);
+              getAllDirectoryAsTree(true, currentDirId);
             }
           })
         }
@@ -516,9 +516,15 @@ const AssetTree = (props) => {
     refresh && getAllDirectoryAsTree(resetCurrentDirId);
   }
 
-  const onCustomDirectoryCancel = (refresh=false) => {
+  const onCustomDirectoryCancel = (refresh=false, id='') => {
     setCustomDirectoryModalVisible(false);
-    refresh && getAllDirectoryAsTree(false);
+    if (refresh) {
+      if (customDirectoryAction === 'add') {
+        getAllDirectoryAsTree(true, id);
+      } else {
+        getAllDirectoryAsTree(false);
+      }
+    }
   }
 
   const displayMenu = (e) => {

src/view/Manage/AssetManage/Component/CustomDirectoryModal.jsx

@@ -130,10 +130,10 @@ const CustomDirectoryModal = (props) => {
       dispatch({
         type: 'assetmanage.saveTreeByCustomElements',
         payload,
-        callback: () => {
+        callback: data => {
           setConfirmLoading(false);
           reset();
-          onCancel && onCancel(true);
+          onCancel && onCancel(true, data?.id);
         },
         error: () => {
           setConfirmLoading(false);