점포 북마크 기능 개선

src/main/java/com/dongyang/dongpo/controller/bookmark/BookmarkController.java

@@ -2,15 +2,14 @@
 
 import com.dongyang.dongpo.apiresponse.ApiResponse;
 import com.dongyang.dongpo.domain.member.Member;
-import com.dongyang.dongpo.dto.bookmark.BookmarkDto;
 import com.dongyang.dongpo.service.bookmark.BookmarkService;
 import io.swagger.v3.oas.annotations.Operation;
 import lombok.RequiredArgsConstructor;
 import org.springframework.http.ResponseEntity;
 import org.springframework.security.core.annotation.AuthenticationPrincipal;
 import org.springframework.web.bind.annotation.*;
 
-import java.util.List;
+import java.util.Map;
 
 @RestController
 @RequestMapping("/api/bookmark")
@@ -21,15 +20,15 @@ public class BookmarkController {
 
     @PostMapping("")
     @Operation(summary = "북마크 추가")
-    public ResponseEntity<ApiResponse<String>> addBookmark(@RequestBody Long storeId,
-                                                           @AuthenticationPrincipal Member member) throws Exception {
-        bookmarkService.addBookmark(member, storeId);
+    public ResponseEntity<ApiResponse<String>> addBookmark(@RequestBody Map<String, Long> request,
+                                                           @AuthenticationPrincipal Member member) {
+        bookmarkService.addBookmark(member, request.get("storeId"));
         return ResponseEntity.ok(new ApiResponse<>("success"));
     }
 
     @DeleteMapping("/{id}")
     public ResponseEntity<ApiResponse<String>> deleteBookmark(@PathVariable Long id,
-                                                              @AuthenticationPrincipal Member member) throws Exception {
+                                                              @AuthenticationPrincipal Member member) {
         bookmarkService.deleteBookmark(id, member);
         return ResponseEntity.ok(new ApiResponse<>("success"));
     }

src/main/java/com/dongyang/dongpo/domain/store/StoreBookmark.java

@@ -11,7 +11,9 @@
 @Builder
 @NoArgsConstructor(access = AccessLevel.PROTECTED)
 @AllArgsConstructor
-@Table(name = "store_bookmark")
+@Table(name = "store_bookmark", uniqueConstraints = {
+        @UniqueConstraint(columnNames = {"member_id", "store_id"})
+})
 public class StoreBookmark {
     @Id
     @GeneratedValue(strategy = GenerationType.IDENTITY)

src/main/java/com/dongyang/dongpo/exception/ErrorCode.java

@@ -19,6 +19,7 @@ public enum ErrorCode {
 
     STORE_REGISTRATION_NOT_VALID(400, "위치 정보가 오차를 벗어났습니다."),
     ARGUMENT_NOT_SATISFIED(400, "요청이 잘못되었습니다."),
+    BOOKMARK_ALREADY_EXISTS(400, "해당 점포의 북마크가 이미 존재합니다."),
     ;
 
     private final int code;

src/main/java/com/dongyang/dongpo/service/bookmark/BookmarkService.java

@@ -10,6 +10,7 @@
 import com.dongyang.dongpo.repository.store.StoreRepository;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
+import org.springframework.dao.DataIntegrityViolationException;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
@@ -35,7 +36,11 @@ public void addBookmark(Member member, Long storeId) {
                 .member(member)
                 .build();
 
-        bookmarkRepository.save(bookmark);
+        try {
+            bookmarkRepository.save(bookmark);
+        } catch (DataIntegrityViolationException ignore) {
+            throw new CustomException(ErrorCode.BOOKMARK_ALREADY_EXISTS);
+        }
 
         log.info("Member Id : {} is Add Bookmark Store Id : {}", member.getId(), storeId);
     }