Просмотр исходного кода

update file-service FileService

reghao 3 месяцев назад
Родитель
Сommit
a2e9afd879

+ 13 - 0
file/file-service/src/main/java/cn/reghao/tnb/file/app/service/FileService.java

@@ -13,6 +13,7 @@ import cn.reghao.tnb.file.app.model.po.LocalFile;
 import cn.reghao.tnb.file.app.model.vo.FileQuery;
 import cn.reghao.tnb.file.app.model.vo.LocalFileInfo;
 import cn.reghao.tnb.file.app.model.vo.LocalFileUrl;
+import jakarta.annotation.PostConstruct;
 import jakarta.servlet.http.HttpServletResponse;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.io.FileUtils;
@@ -48,6 +49,18 @@ public class FileService {
         this.baseDir = appProperties.getBaseDir();
     }
 
+    @PostConstruct
+    public void createDir() {
+        File dir = new File(this.baseDir);
+        if (!dir.exists()) {
+            try {
+                FileUtils.forceMkdirParent(dir);
+            } catch (IOException e) {
+                log.error("create dir {} failed: {}", this.baseDir, e.getMessage());
+            }
+        }
+    }
+
     public void getFile(String objectName) throws IOException {
         LocalFile diskFile = getByObjectName(objectName);
         if (diskFile == null) {