diff --git a/src/main/java/zero/eight/donut/domain/FcmToken.java b/src/main/java/zero/eight/donut/domain/FcmToken.java index 4b107a0..9e2e6b0 100644 --- a/src/main/java/zero/eight/donut/domain/FcmToken.java +++ b/src/main/java/zero/eight/donut/domain/FcmToken.java @@ -31,8 +31,8 @@ public FcmToken(String token, Long memberId, Role role) { this.role = role; } - public void updateToken(String token, Role role) { + public FcmToken updateToken(String token) { this.token = token; - this.role = role; + return this; } } \ No newline at end of file diff --git a/src/main/java/zero/eight/donut/service/FcmService.java b/src/main/java/zero/eight/donut/service/FcmService.java index 6a81b33..19e5d4c 100644 --- a/src/main/java/zero/eight/donut/service/FcmService.java +++ b/src/main/java/zero/eight/donut/service/FcmService.java @@ -34,7 +34,7 @@ public ApiResponse registerFcmToken(FcmTokenRequestDto requestDto) throws Exc fcmTokenRepository.findByMemberId(member.getId()) .ifPresentOrElse( - it -> it.updateToken(token, member.getRole()), + it -> fcmTokenRepository.save(it.updateToken(token)), () -> fcmTokenRepository.save(FcmToken.builder() .token(token) .memberId(member.getId())