From c81efce3de008dfa1e2d67046bcb49985038b804 Mon Sep 17 00:00:00 2001
From: lanyuanxiaoyao <lanyuanxiaoyao@gmail.com>
Date: Fri, 1 Mar 2024 13:55:34 +0800
Subject: [PATCH] =?UTF-8?q?fix(uploader):=20=E9=81=BF=E5=85=8D=E8=B7=AF?=
 =?UTF-8?q?=E5=BE=84=E4=B8=8D=E5=AD=98=E5=9C=A8=E5=AF=BC=E8=87=B4=E9=94=99?=
 =?UTF-8?q?=E8=AF=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/filesystem/HdfsUploadAndDownloadService.java  | 8 +++++++-
 .../service/filesystem/LocalUploadAndDownloadService.java | 5 ++++-
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/service-uploader/src/main/java/com/lanyuanxiaoyao/service/uploader/service/filesystem/HdfsUploadAndDownloadService.java b/service-uploader/src/main/java/com/lanyuanxiaoyao/service/uploader/service/filesystem/HdfsUploadAndDownloadService.java
index 81338ed..8b6681a 100644
--- a/service-uploader/src/main/java/com/lanyuanxiaoyao/service/uploader/service/filesystem/HdfsUploadAndDownloadService.java
+++ b/service-uploader/src/main/java/com/lanyuanxiaoyao/service/uploader/service/filesystem/HdfsUploadAndDownloadService.java
@@ -2,6 +2,7 @@ package com.lanyuanxiaoyao.service.uploader.service.filesystem;
 
 import cn.hutool.core.io.IoUtil;
 import com.lanyuanxiaoyao.service.uploader.configuration.UploaderConfiguration;
+import java.io.IOException;
 import java.io.InputStream;
 import java.io.OutputStream;
 import org.apache.hadoop.conf.Configuration;
@@ -25,9 +26,14 @@ public class HdfsUploadAndDownloadService implements UploadAndDownloadService {
     private final UploaderConfiguration uploaderConfiguration;
     private final Configuration configuration;
 
-    public HdfsUploadAndDownloadService(UploaderConfiguration uploaderConfiguration) {
+    public HdfsUploadAndDownloadService(UploaderConfiguration uploaderConfiguration) throws IOException {
         this.uploaderConfiguration = uploaderConfiguration;
         this.configuration = new Configuration();
+
+        logger.info("Create tmp dir: {}", uploaderConfiguration.getTmpDir());
+        try (FileSystem fileSystem = FileSystem.get(configuration)) {
+            fileSystem.mkdirs(new Path(uploaderConfiguration.getTmpDir()));
+        }
     }
 
     @Override
diff --git a/service-uploader/src/main/java/com/lanyuanxiaoyao/service/uploader/service/filesystem/LocalUploadAndDownloadService.java b/service-uploader/src/main/java/com/lanyuanxiaoyao/service/uploader/service/filesystem/LocalUploadAndDownloadService.java
index fb53627..96b4b45 100644
--- a/service-uploader/src/main/java/com/lanyuanxiaoyao/service/uploader/service/filesystem/LocalUploadAndDownloadService.java
+++ b/service-uploader/src/main/java/com/lanyuanxiaoyao/service/uploader/service/filesystem/LocalUploadAndDownloadService.java
@@ -2,6 +2,7 @@ package com.lanyuanxiaoyao.service.uploader.service.filesystem;
 
 import cn.hutool.core.io.IoUtil;
 import com.lanyuanxiaoyao.service.uploader.configuration.UploaderConfiguration;
+import java.io.IOException;
 import java.io.InputStream;
 import java.io.OutputStream;
 import java.nio.file.Files;
@@ -21,8 +22,10 @@ public class LocalUploadAndDownloadService implements UploadAndDownloadService {
     private static final Logger logger = LoggerFactory.getLogger(LocalUploadAndDownloadService.class);
     private final UploaderConfiguration uploaderConfiguration;
 
-    public LocalUploadAndDownloadService(UploaderConfiguration uploaderConfiguration) {
+    public LocalUploadAndDownloadService(UploaderConfiguration uploaderConfiguration) throws IOException {
         this.uploaderConfiguration = uploaderConfiguration;
+        logger.info("Create tmp dir: {}", uploaderConfiguration.getTmpDir());
+        Files.createDirectories(Paths.get(uploaderConfiguration.getTmpDir()));
     }
 
     @Override