ImagesService refactoring

1 files changed, 5 insertions, 1 deletions

diff --git a/juick-common/src/main/java/com/juick/service/ImagesServiceImpl.java b/juick-common/src/main/java/com/juick/service/ImagesServiceImpl.java
index dbc3f3a9..67c8360e 100644
--- a/juick-common/src/main/java/com/juick/service/ImagesServiceImpl.java
+++ b/juick-common/src/main/java/com/juick/service/ImagesServiceImpl.java
@@ -12,11 +12,15 @@ import java.nio.file.Paths;
 
 public class ImagesServiceImpl implements ImagesService {
     private ImageUtils imageUtils;
+    private String imgDir;
+    private String tmpDir;
     public ImagesServiceImpl(String imgDir, String tmpDir) {
+        this.imgDir = imgDir;
+        this.tmpDir = tmpDir;
         imageUtils = new ImageUtils(imgDir, tmpDir);
     }
     @Override
-    public void setAttachmentMetadata(String imgDir, String baseUrl, Message msg) throws Exception {
+    public void setAttachmentMetadata(String baseUrl, Message msg) throws Exception {
         if (!StringUtils.isEmpty(msg.getAttachmentType())) {
             Photo photo = new Photo();
             if (msg.getRid()> 0) {