mod: OreumResponse 오름 유형 enum 타입으로 변경

Song-EunJu · Nov 17, 2022 · 6abde55 · 6abde55
1 parent 3e84068
commit 6abde55
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/main/java/com/example/goorm/oreum/dto/OreumResponse.java b/src/main/java/com/example/goorm/oreum/dto/OreumResponse.java
@@ -2,6 +2,7 @@
 
 import com.example.goorm.oreum.domain.MyOreum;
 import com.example.goorm.oreum.domain.Oreum;
+import com.example.goorm.oreum.domain.OreumType;
 import lombok.Builder;
 import lombok.Getter;
 import lombok.Setter;
@@ -12,7 +13,7 @@
 public class OreumResponse {
     private String nickname;
     private String name;
-    private String type;
+    private OreumType type;
     private double xPos;
     private double yPos;
     private double zPos;
@@ -24,7 +25,7 @@ public static OreumResponse of(Oreum oreum, MyOreum myOreum, BirthdayRequest req
         return OreumResponse.builder()
                 .nickname(request.getNickname())
                 .name(oreum.getName())
-                .type(oreum.getOreumType().getTitle())
+                .type(oreum.getOreumType())
                 .xPos(oreum.getXPos())
                 .yPos(oreum.getYPos())
                 .zPos(oreum.getZPos())
@@ -38,7 +39,7 @@ public static OreumResponse ofOthers(Oreum oreum, MyOreum myOreum){
         return OreumResponse.builder()
                 .nickname(myOreum.getNickname())
                 .name(oreum.getName())
-                .type(oreum.getOreumType().getTitle())
+                .type(oreum.getOreumType())
                 .xPos(oreum.getXPos())
                 .yPos(oreum.getYPos())
                 .zPos(oreum.getZPos())