diff --git a/src/main/java/com/fav/daengnyang/domain/account/controller/AccountController.java b/src/main/java/com/fav/daengnyang/domain/account/controller/AccountController.java index 434ff7d..f1bdb35 100644 --- a/src/main/java/com/fav/daengnyang/domain/account/controller/AccountController.java +++ b/src/main/java/com/fav/daengnyang/domain/account/controller/AccountController.java @@ -41,9 +41,9 @@ public ResponseEntity> inquireAccount( @PatchMapping("/update-color") public ResponseEntity> updateCustomColor( - @RequestParam String accountNumber, + @AuthenticationPrincipal MemberPrincipal memberPrincipal, @RequestBody ColorUpdateRequest request) { - AccountResponse response = accountService.updateAccountColor(accountNumber, request.getNewColor()); + AccountResponse response = accountService.updateAccountColor(memberPrincipal.getMemberId(), request.getNewColor()); return ResponseEntity.ok(SuccessResponse.ok(response)); } diff --git a/src/main/java/com/fav/daengnyang/domain/account/repository/AccountRepository.java b/src/main/java/com/fav/daengnyang/domain/account/repository/AccountRepository.java index 47518e2..4b7cd3f 100644 --- a/src/main/java/com/fav/daengnyang/domain/account/repository/AccountRepository.java +++ b/src/main/java/com/fav/daengnyang/domain/account/repository/AccountRepository.java @@ -8,10 +8,7 @@ import java.util.Optional; public interface AccountRepository extends JpaRepository { - Optional findByAccountNumber(String bankbookNumber); - // memberId로 Account 조회 @Query("SELECT b FROM Account b WHERE b.member.memberId = :memberId") Optional findByMemberId(@Param("memberId")Long memberId); - } diff --git a/src/main/java/com/fav/daengnyang/domain/account/service/AccountService.java b/src/main/java/com/fav/daengnyang/domain/account/service/AccountService.java index f75a2ec..34122fa 100644 --- a/src/main/java/com/fav/daengnyang/domain/account/service/AccountService.java +++ b/src/main/java/com/fav/daengnyang/domain/account/service/AccountService.java @@ -95,8 +95,9 @@ public AccountResponse inquireAccount(Long memberId, String userKey) throws Json } // 커스텀 색상 업데이트 - public AccountResponse updateAccountColor(String accountNumber, String newColor) { - Optional optionalAccount = accountRepository.findByAccountNumber(accountNumber); + public AccountResponse updateAccountColor(Long memberId, String newColor) { + + Optional optionalAccount = accountRepository.findByMemberId(memberId); if (!optionalAccount.isPresent()) { throw new RuntimeException("해당 계좌 번호를 가진 계좌를 찾을 수 없습니다.");