From 153f8ebbef26eba0854077bcf670b60a791704b4 Mon Sep 17 00:00:00 2001 From: Ryan Wang Date: Thu, 12 Mar 2020 15:04:35 +0800 Subject: [PATCH] refactor: attachment month path. (#636) --- .../java/run/halo/app/handler/file/LocalFileHandler.java | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/main/java/run/halo/app/handler/file/LocalFileHandler.java b/src/main/java/run/halo/app/handler/file/LocalFileHandler.java index c75e62d0a..06cbbfbc2 100644 --- a/src/main/java/run/halo/app/handler/file/LocalFileHandler.java +++ b/src/main/java/run/halo/app/handler/file/LocalFileHandler.java @@ -97,8 +97,10 @@ public class LocalFileHandler implements FileHandler { int year = current.get(Calendar.YEAR); int month = current.get(Calendar.MONTH) + 1; + String monthString = month < 10 ? "0" + month : String.valueOf(month); + // Build directory - String subDir = UPLOAD_SUB_DIR + year + FILE_SEPARATOR + month + FILE_SEPARATOR; + String subDir = UPLOAD_SUB_DIR + year + FILE_SEPARATOR + monthString + FILE_SEPARATOR; String originalBasename = FilenameUtils.getBasename(Objects.requireNonNull(file.getOriginalFilename())); @@ -137,7 +139,7 @@ public class LocalFileHandler implements FileHandler { uploadResult.setSize(file.getSize()); // TODO refactor this: if image is svg ext. extension - boolean isSvg = "svg".equals(extension); + boolean isSvg = "svg" .equals(extension); // Check file type if (FileHandler.isImageType(uploadResult.getMediaType()) && !isSvg) {