diff --git a/src/main/java/ru/soune/nocopy/handler/RegRequestHandler.java b/src/main/java/ru/soune/nocopy/handler/RegRequestHandler.java index 77efad5..d9a54bd 100644 --- a/src/main/java/ru/soune/nocopy/handler/RegRequestHandler.java +++ b/src/main/java/ru/soune/nocopy/handler/RegRequestHandler.java @@ -127,7 +127,7 @@ public class RegRequestHandler implements RequestHandler { referralService.onRegister(authToken.getUser().getId(), regRequest.getReferralLink()); } - additionalInfoService.additionalInfo(regRequest.getIp(), regRequest.getUserAgent()); + additionalInfoService.additionalInfo(regRequest.getIp(), regRequest.getUserAgent(), authToken.getUser()); return new BaseResponse(request.getMsgId(), MessageCode.SUCCESS.getCode(), MessageCode.SUCCESS.getDescription(), regAnswer); diff --git a/src/main/java/ru/soune/nocopy/service/user/AdditionalInfoService.java b/src/main/java/ru/soune/nocopy/service/user/AdditionalInfoService.java index f0016cc..3ca53d2 100644 --- a/src/main/java/ru/soune/nocopy/service/user/AdditionalInfoService.java +++ b/src/main/java/ru/soune/nocopy/service/user/AdditionalInfoService.java @@ -3,6 +3,7 @@ package ru.soune.nocopy.service.user; import lombok.RequiredArgsConstructor; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; +import ru.soune.nocopy.entity.user.User; import ru.soune.nocopy.entity.user.UserAdditionalInfo; import ru.soune.nocopy.exception.RegistrationAdditionalInfoNullException; import ru.soune.nocopy.repository.UserAdditionalInfoRepository; @@ -17,7 +18,7 @@ public class AdditionalInfoService { private final UserAdditionalInfoRepository additionalInfoRepository; @Transactional - public void additionalInfo(String ip, String userAgent) { + public void additionalInfo(String ip, String userAgent, User user) { if (ip == null || userAgent == null) { throw new RegistrationAdditionalInfoNullException("Ip address or user agent is null"); } @@ -25,6 +26,7 @@ public class AdditionalInfoService { UserAdditionalInfo userAdditionalInfo = UserAdditionalInfo.builder() .ipAddress(ip) .userAgent(userAgent) + .user(user) .build(); additionalInfoRepository.save(userAdditionalInfo);