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