From ea62e1ecb7c424dcb0e72cc579da77164519fbee Mon Sep 17 00:00:00 2001
From: chenxudong <chenxudong@cqida.cn>
Date: Thu, 17 Apr 2025 11:03:02 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E9=80=9A=E4=BA=86=E6=95=B0=E6=8D=AE?=
 =?UTF-8?q?=E5=BA=93=E5=AF=BC=E5=85=A5=E7=9A=84=E9=83=A8=E5=88=86=E5=8A=9F?=
 =?UTF-8?q?=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../serviceimpl/EdFileInfoServiceImpl.java      | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

diff --git a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/EdFileInfoServiceImpl.java b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/EdFileInfoServiceImpl.java
index fb2595a..cfb421d 100644
--- a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/EdFileInfoServiceImpl.java
+++ b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/EdFileInfoServiceImpl.java
@@ -474,7 +474,7 @@ public class EdFileInfoServiceImpl extends ServiceImpl<EdFileInfoMapper, EdFileI
         ZipUtil.unzip(zipDirPath, tmpDir);
         update2Database(tmpDir, dataOwnCode);
         fileSystemService.deleteFile(zipDirPath, destColibPath);
-        UserThreadLocal.setSuccessInfo("", "", "导入数据库成功");
+        UserThreadLocal.setSuccessInfo("", "", "导入数据库成功，文件名称为 " + mainName);
         return ElectromagneticResultUtil.success(true);
     }
 
@@ -494,13 +494,13 @@ public class EdFileInfoServiceImpl extends ServiceImpl<EdFileInfoMapper, EdFileI
     }
 
     private void updateFileRelationInfo(String prjDirPath) {
-        String path = prjDirPath + File.separator + ED_TAG_RELATIONS + ".json";
+        String path = prjDirPath + File.separator + ED_FILE_RELATION + ".json";
         List<EdFileRelation> relations = JSONUtil.toList(FileUtil.readString(path, Charset.defaultCharset()), EdFileRelation.class);
         edFileRelationService.saveOrUpdateBatch(relations);
     }
 
     private void updateFileTageInfo(String prjDirPath) {
-        String path = prjDirPath + File.separator + USER_ACCESS_LOG + ".json";
+        String path = prjDirPath + File.separator + ED_TAG_RELATIONS + ".json";
         List<FileTagRelation> relations = JSONUtil.toList(FileUtil.readString(path, Charset.defaultCharset()), FileTagRelation.class);
         fileTagRelationService.saveOrUpdateBatch(relations);
     }
@@ -692,7 +692,7 @@ public class EdFileInfoServiceImpl extends ServiceImpl<EdFileInfoMapper, EdFileI
 
     private void update2FileSystem(Set<EdFileInfo> needMove2FileSystemFiles, String prjDirPath, int dataOwnCode) {
         List<EdFileInfo> files = needMove2FileSystemFiles.stream().filter(e -> e.getDataType().equals(EleDataTypeEnum.FILE.code)).toList();
-        List<File> files1 = FileUtil.loopFiles(new File(prjDirPath), file -> file.isDirectory() && file.getName().startsWith(EXPORT_PRJ_NAME));
+        List<File> files1 = Arrays.stream(Objects.requireNonNull(new File(prjDirPath).listFiles())).filter(e -> e.isDirectory() && e.getName().startsWith(EXPORT_PRJ_NAME)).toList();
         String prjFilePath = files1.get(0).getAbsolutePath();
         for (EdFileInfo edFileInfo : files) {
             String id = edFileInfo.getId();
@@ -748,7 +748,7 @@ public class EdFileInfoServiceImpl extends ServiceImpl<EdFileInfoMapper, EdFileI
         // 导出标签相关信息
         exportFileTagInfo(nowTimeStr, exportFileIds, userDownloadDataDir);
         // 导出操作记录相关
-        exportLogInfo(nowTimeStr, exportFileIds, userDownloadDataDir);
+        exportLogInfo(nowTimeStr, userDownloadDataDir);
 
         String prjDirPath = userDownloadDataDir + File.separator + EXPORT_PRJ_NAME + "_" + nowTimeStr;
         String exportZipFile = userDownloadDataDir + File.separator + EXPORT_PRJ_NAME + "_" + nowTimeStr + ".zip";
@@ -772,11 +772,12 @@ public class EdFileInfoServiceImpl extends ServiceImpl<EdFileInfoMapper, EdFileI
         FileSystemResource fileSystemResource = new FileSystemResource(file);
         HttpHeaders headers = new HttpHeaders();
         headers.add("Cache-Control", "no-cache, no-store, must-revalidate");
-        String fileName = Base64.encode(fileSystemResource.getFilename() + "_" + nowTimeStr);
+        String tmpFileName = fileSystemResource.getFilename() + "_" + nowTimeStr;
+        String fileName = Base64.encode(tmpFileName);
         headers.add("Pragma", "no-cache");
         headers.add("Expires", "0");
         response.setHeader("content-disposition", "attachment;filename=" + fileName);
-        UserThreadLocal.setSuccessInfo("", "", "导出数据库成功");
+        UserThreadLocal.setSuccessInfo("", "", "导出数据库成功，文件名称为" + tmpFileName);
         // 构建响应实体(可以返回<byte[]或Resource，返回类型取决body入参类型)
         return ResponseEntity
                 .ok()
@@ -786,7 +787,7 @@ public class EdFileInfoServiceImpl extends ServiceImpl<EdFileInfoMapper, EdFileI
                 .body(new InputStreamResource(fileSystemResource.getInputStream()));
     }
 
-    private void exportLogInfo(String nowTimeStr, List<String> exportFileIds, String userDownloadDataDir) {
+    private void exportLogInfo(String nowTimeStr, String userDownloadDataDir) {
         List<UserAccessLog> userAccessLogs = userAccessLogService.getBaseMapper().selectList(null);
         String json = JSONUtil.toJsonStr(userAccessLogs);
         String path = userDownloadDataDir + File.separator + EXPORT_PRJ_NAME + "_" + nowTimeStr + File.separator + USER_ACCESS_LOG + ".json";