addVisit 수정하기

- 별점 변수를 int 타입에서 double 타입으로 수정하기 Related to: #37
ahyoon99 · Sep 6, 2024 · 1c1d74f · 1c1d74f
1 parent 865b769
commit 1c1d74f
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 6 deletions.
diff --git a/src/main/java/com/example/naverrestaurant/db/H2DbRepositoryAbstract.java b/src/main/java/com/example/naverrestaurant/db/H2DbRepositoryAbstract.java
@@ -111,7 +111,7 @@ private RowMapper<WishListEntity> wishListEntityRowMapper(){
             if(rs.getTimestamp("lastVisitDate") != null){   // lastVisitDate가 null이 아닐 때만 set 해주기
                 wishListEntity.setLastVisitDate(rs.getTimestamp("lastVisitDate").toLocalDateTime());
             }
-            wishListEntity.setStarRating((rs.getInt("starRating")));
+            wishListEntity.setStarRating((rs.getDouble("starRating")));
             return wishListEntity;
         });
     }

diff --git a/src/main/java/com/example/naverrestaurant/wishlist/dto/WishListDto.java b/src/main/java/com/example/naverrestaurant/wishlist/dto/WishListDto.java
@@ -20,5 +20,5 @@ public class WishListDto {
     private boolean isVisit;    // 방문 여부
     private int visitCount; // 방문 횟수
     private LocalDateTime lastVisitDate;    // 마지막 방문 날짜
-    private int starRating;     // 별점
+    private double starRating;     // 별점
 }
diff --git a/src/main/java/com/example/naverrestaurant/wishlist/entity/WishListEntity.java b/src/main/java/com/example/naverrestaurant/wishlist/entity/WishListEntity.java
@@ -22,5 +22,5 @@ public class WishListEntity extends MemoryDbEntity {
     private boolean isVisit;    // 방문 여부
     private int visitCount; // 방문 횟수
     private LocalDateTime lastVisitDate;    // 마지막 방문 날짜
-    private int starRating;         // 별점
+    private double starRating;         // 별점
 }
diff --git a/src/main/java/com/example/naverrestaurant/wishlist/service/WishListService.java b/src/main/java/com/example/naverrestaurant/wishlist/service/WishListService.java
@@ -121,10 +121,10 @@ public void addVisit(int index, int starRating) {
             int restaurantEntityVisitCount = restaurantEntity.getVisitCount();
             double restaurantEntityStarRating = restaurantEntity.getStarRating();
 
-            starRating = (int) (restaurantEntityStarRating*restaurantEntityVisitCount+starRating)/(restaurantEntityVisitCount+1);
+            restaurantEntityStarRating = (Math.round(((restaurantEntityStarRating*restaurantEntityVisitCount+starRating)/(restaurantEntityVisitCount+1))*100)/100.0);
 
             restaurantEntity.setVisitCount(restaurantEntityVisitCount+1);
-            restaurantEntity.setStarRating(starRating);
+            restaurantEntity.setStarRating(restaurantEntityStarRating);
             restaurantEntity.setLastVisitDate(LocalDateTime.now());
             wishListRepository.updateById(index, restaurantEntity);
 

diff --git a/src/main/resources/h2/schema.sql b/src/main/resources/h2/schema.sql
@@ -13,5 +13,5 @@ CREATE TABLE RESTAURANT
     isVisit BOOLEAN NOT NULL,           -- 방문 여부
     visitCount INTEGER NOT NULL,         -- 방문 횟수
     lastVisitDate TIMESTAMP,    -- 마지막 방문 날짜
-    starRating INTEGER          -- 별점
+    starRating DOUBLE          -- 별점
 );