refactor: 이미지 업로드 로직 수정

woowacourse-teams · Oct 19, 2023 · 4dc0804 · 4dc0804
1 parent 4fb352a
commit 4dc0804
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/backend/src/main/java/hanglog/image/infrastructure/ImageUploader.java b/backend/src/main/java/hanglog/image/infrastructure/ImageUploader.java
@@ -8,11 +8,13 @@
 import com.amazonaws.services.s3.model.ObjectMetadata;
 import hanglog.global.exception.ImageException;
 import hanglog.image.domain.ImageFile;
+import hanglog.image.domain.S3ImageEvent;
 import java.io.IOException;
 import java.io.InputStream;
 import java.util.List;
 import lombok.RequiredArgsConstructor;
 import org.springframework.beans.factory.annotation.Value;
+import org.springframework.context.ApplicationEventPublisher;
 import org.springframework.stereotype.Component;
 
 @Component
@@ -22,6 +24,7 @@ public class ImageUploader {
     private static final String CACHE_CONTROL_VALUE = "max-age=3153600";
 
     private final AmazonS3 s3Client;
+    private final ApplicationEventPublisher publisher;
 
     @Value("${cloud.aws.s3.bucket}")
     private String bucket;
@@ -30,9 +33,14 @@ public class ImageUploader {
     private String folder;
 
     public List<String> uploadImages(final List<ImageFile> imageFiles) {
-        return imageFiles.stream()
-                .map(this::uploadImage)
-                .toList();
+        try {
+            return imageFiles.stream()
+                    .map(this::uploadImage)
+                    .toList();
+        } catch (final ImageException e) {
+            imageFiles.forEach(imageFile -> publisher.publishEvent(new S3ImageEvent(imageFile.getHashedName())));
+            throw e;
+        }
     }
 
     private String uploadImage(final ImageFile imageFile) {