updateById 수정하기

entity 자체를 수정해주었다. 그 결과, Service의 addVisit, setStarRating 메소드 둘 다 하나의 updateById 메소드를 통해 기능 수행이 가능해졌다. Related to: #37
ahyoon99 · Sep 3, 2024 · 5935346 · 5935346
1 parent bce12e8
commit 5935346
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/src/main/java/com/example/naverrestaurant/db/H2DbRepositoryAbstract.java b/src/main/java/com/example/naverrestaurant/db/H2DbRepositoryAbstract.java
@@ -70,15 +70,17 @@ public List<T> listAll() {
 
     @Override
     public void updateById(int index, T entity) {
-        String updateByIdQuery = "update restaurant set isvisit=?, visitcount=? where index=?";
+        String updateByIdQuery = "update restaurant " +
+                "set title=?, category=?, address=?, roadAddress=?, homepageLink=?, imageLink=?, isvisit=?, visitCount=?, lastVisitDate=?, starRating=? where index=?";
 
         WishListEntity wishListEntity = (WishListEntity) entity;
-        jdbcTemplate.update(updateByIdQuery, wishListEntity.isVisit(), wishListEntity.getVisitCount(), index);
-    }
+        Object []updateParams = new Object[] {wishListEntity.getTitle(),
+                wishListEntity.getCategory(), wishListEntity.getAddress(), wishListEntity.getRoadAddress(),
+                wishListEntity.getHomePageLink(), wishListEntity.getImageLink(), wishListEntity.isVisit(),
+                wishListEntity.getVisitCount(), wishListEntity.getLastVisitDate(), wishListEntity.getStarRating(), index};
+
+        jdbcTemplate.update(updateByIdQuery, updateParams);
 
-    public void updateStarRatingById(int index, int starRating){
-        String updateStarRatingByIdQuery = "update restaurant set starRating = ? where index = ?";
-        jdbcTemplate.update(updateStarRatingByIdQuery, starRating, index);
     }
 
     private RowMapper<WishListEntity> wishListEntityRowMapper(){

diff --git a/src/main/java/com/example/naverrestaurant/wishlist/service/WishListService.java b/src/main/java/com/example/naverrestaurant/wishlist/service/WishListService.java
@@ -123,7 +123,7 @@ public void setStarRating(int index, int starRating){
         if(restaurant.isPresent()){
             var restaurantEntity = restaurant.get();
             restaurantEntity.setStarRating(starRating);
-            wishListRepository.updateStarRatingById(index, starRating);
+            wishListRepository.updateById(index, restaurantEntity);
 
             var result = wishListRepository.findById(index);
             System.out.println(result.toString());