fix value

src/main/java/ru/soune/nocopy/handler/RegRequestHandler.java

@@ -10,10 +10,13 @@ import ru.soune.ReferralService;
 import ru.soune.nocopy.dto.*;
 import ru.soune.nocopy.dto.register.RegAnswer;
 import ru.soune.nocopy.dto.register.RegRequest;
+import ru.soune.nocopy.entity.referral.Referral;
 import ru.soune.nocopy.entity.user.AuthToken;
 import ru.soune.nocopy.exception.NotValidFieldException;
 import ru.soune.nocopy.exception.ValidationException;
 import ru.soune.nocopy.handler.validator.RegRequestValidator;
+import ru.soune.nocopy.repository.ReferralJpaRepository;
+import ru.soune.nocopy.repository.ReferralLevelRepository;
 import ru.soune.nocopy.repository.UserRepository;
 import ru.soune.nocopy.service.register.AuthService;
 import ru.soune.nocopy.service.mail.EmailService;
@@ -37,6 +40,8 @@ public class RegRequestHandler implements RequestHandler {
 
     private final ReferralService referralService;
 
+    private final ReferralJpaRepository referralJpaRepository;
+
     @Override
     public BaseResponse handle(BaseRequest request) throws ValidationException {
         RegRequest regRequest = objectMapper.convertValue(request.getMessageBody(), RegRequest.class);
@@ -52,6 +57,19 @@ public class RegRequestHandler implements RequestHandler {
                     MessageCode.REG_EMAIL_OR_PHONE_EXISTS.getDescription(), regAnswer));
         }
 
+        String referralLink = regRequest.getReferralLink();
+
+        Referral referral = referralJpaRepository.findByReferralLink(referralLink);
+
+        if (referral == null) {
+            throw new NotValidFieldException("Referral link is not found:" + referralLink +
+                    referralLink, new BaseResponse(request.getMsgId(),
+                    MessageCode.REFERRAL_LINK_IS_NOT_EXIST.getCode(),
+                    MessageCode.REFERRAL_LINK_IS_NOT_EXIST.getDescription(), Map.of(
+                            "referralLink", referralLink
+            )));
+        }
+
         BindingResult bindingResult = new BeanPropertyBindingResult(regRequest, "regRequest");
 
         regRequestValidator.validate(regRequest, bindingResult);
@@ -79,7 +97,6 @@ public class RegRequestHandler implements RequestHandler {
 
         RegAnswer regAnswer = new RegAnswer();
 
-
         regAnswer.setUserId(authToken.getUser().getId());
         regAnswer.setToken(token);
         regAnswer.setVerified(true);