dev add notification entity

src/main/java/ru/soune/nocopy/service/referral/ReferralRepoImpl.java

@@ -11,9 +11,11 @@ import org.springframework.transaction.annotation.Transactional;
 import ru.soune.Referral;
 import ru.soune.ReferralInvitee;
 import ru.soune.ReferralRepo;
+import ru.soune.nocopy.entity.notification.NotificationType;
 import ru.soune.nocopy.entity.user.User;
 import ru.soune.nocopy.repository.ReferralJpaRepository;
 import ru.soune.nocopy.repository.UserRepository;
+import ru.soune.nocopy.service.notification.NotificationService;
 
 
 import java.lang.reflect.Constructor;
@@ -31,6 +33,8 @@ public class ReferralRepoImpl implements ReferralRepo {
 
     private final UserRepository userRepository;
 
+    private final NotificationService notificationService;
+
     @Override
     public @NotNull String getUserReferralLink(long userId) {
         ru.soune.nocopy.entity.referral.Referral referral =
@@ -150,6 +154,8 @@ public class ReferralRepoImpl implements ReferralRepo {
     @Override
     public void createReferralEntity(@NotNull Referral entity) {
         ru.soune.nocopy.entity.referral.Referral referral = new ru.soune.nocopy.entity.referral.Referral();
+        Long inviter = entity.getInviter();
+
         referral.setUserId(entity.getUserId());
         referral.setReferralLink(entity.getReferralLink());
         referral.setInviterId(entity.getInviter());
@@ -160,6 +166,10 @@ public class ReferralRepoImpl implements ReferralRepo {
         referral.setActive(entity.getActive());
 
         referralRepository.save(referral);
+
+        if (inviter != null){
+            notificationService.addNotification(NotificationType.REFERRAL_REGISTERED, inviter);
+        }
     }
 
     @Override