updateById 작성하기

Related to: #37
ahyoon99 · Aug 21, 2024 · bc6c648 · bc6c648
1 parent 2f3f067
commit bc6c648
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/src/main/java/com/example/naverrestaurant/db/H2DbRepositoryAbstract.java b/src/main/java/com/example/naverrestaurant/db/H2DbRepositoryAbstract.java
@@ -68,6 +68,14 @@ public List<T> listAll() {
         return (List<T>) result;
     }
 
+    @Override
+    public void updateById(int index, T entity) {
+        String updateByIdQuery = "update restaurant set isvisit=?, visitcount=? where index=?";
+
+        WishListEntity wishListEntity = (WishListEntity) entity;
+        jdbcTemplate.update(updateByIdQuery, wishListEntity.isVisit(), wishListEntity.getVisitCount(), index);
+    }
+
     private RowMapper<WishListEntity> wishListEntityRowMapper(){
         return ((rs, rowNum) -> {
             WishListEntity wishListEntity = new WishListEntity();

diff --git a/src/main/java/com/example/naverrestaurant/db/MemoryDbRepositoryIfs.java b/src/main/java/com/example/naverrestaurant/db/MemoryDbRepositoryIfs.java
@@ -9,5 +9,6 @@ public interface MemoryDbRepositoryIfs<T> {
     T save(T entity);
     void deleteById(int index);
     List<T> listAll();
+    void updateById(int index, T entity);
 
 }
diff --git a/src/main/java/com/example/naverrestaurant/wishlist/service/WishListService.java b/src/main/java/com/example/naverrestaurant/wishlist/service/WishListService.java
@@ -112,6 +112,7 @@ public void addVisit(int index) {
             var restaurnatEntity = restaurant.get();
             restaurnatEntity.setVisit(true);
             restaurnatEntity.setVisitCount(restaurnatEntity.getVisitCount()+1);
+            wishListRepository.updateById(index, restaurnatEntity);
         }
     }