feat: 이미지 업로드 안하면, 스파크 장소 이미지 리턴하도록 구현 (#14)

KakaoONna · Oct 17, 2023 · 0b34d15 · 0b34d15
1 parent 9ec45c1
commit 0b34d15
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/main/java/com/onna/onnaback/domain/spark/adapter/in/web/SparkController.java b/src/main/java/com/onna/onnaback/domain/spark/adapter/in/web/SparkController.java
@@ -26,6 +26,7 @@
 import com.onna.onnaback.domain.spark.domain.CapacityType;
 import com.onna.onnaback.domain.spark.domain.DurationHour;
 import com.onna.onnaback.domain.spark.domain.SortType;
+import com.onna.onnaback.domain.spark.domain.Spark;
 import com.onna.onnaback.domain.spark.domain.SparkType;
 
 import io.swagger.v3.oas.annotations.Operation;
@@ -75,8 +76,10 @@ public ResponseEntity<SparkHostResponse> host(@AuthenticationPrincipal CustomUse
                                       capacity, durationHour, hostDetail, chatUrl);
 
         Member host = customUserDetails.getMember();
+        Spark spark = this.sparkUseCase.uploadSpark(host, hostDto);
         return ResponseEntity.ok().body(
-                new SparkHostResponse(this.sparkUseCase.uploadSpark(host, hostDto).getSparkId())
+                new SparkHostResponse(spark.getSparkId(),
+                                      img == null ? spark.getPlace().getImg() : spark.getImg())
         );
     }
 

diff --git a/src/main/java/com/onna/onnaback/domain/spark/adapter/in/web/response/SparkHostResponse.java b/src/main/java/com/onna/onnaback/domain/spark/adapter/in/web/response/SparkHostResponse.java
@@ -6,7 +6,10 @@
 public class SparkHostResponse {
     Long sparkId;
 
-    public SparkHostResponse(Long sparkId) {
+    String img;
+
+    public SparkHostResponse(Long sparkId, String img) {
         this.sparkId = sparkId;
+        this.img = img;
     }
 }