diff --git a/src/main/java/ru/soune/nocopy/service/payment/PaymentService.java b/src/main/java/ru/soune/nocopy/service/payment/PaymentService.java index 4aecb1b..195e984 100644 --- a/src/main/java/ru/soune/nocopy/service/payment/PaymentService.java +++ b/src/main/java/ru/soune/nocopy/service/payment/PaymentService.java @@ -245,6 +245,7 @@ public class PaymentService { tariffInfo.setPaymentMethodId(paymentMethodId); tariffInfoService.linkUserTariffInfo(user, tariffInfoService.addTariffInfo(tariffInfo)); + tariffInfoService.linkCompanyTariffInfo(user, tariffInfoService.addTariffInfo(tariffInfo)); } } } diff --git a/src/main/java/ru/soune/nocopy/service/tariff/TariffInfoService.java b/src/main/java/ru/soune/nocopy/service/tariff/TariffInfoService.java index f5a29ad..997b479 100644 --- a/src/main/java/ru/soune/nocopy/service/tariff/TariffInfoService.java +++ b/src/main/java/ru/soune/nocopy/service/tariff/TariffInfoService.java @@ -14,6 +14,7 @@ import ru.soune.nocopy.entity.tarif.TariffType; import ru.soune.nocopy.entity.tokenoperation.OperationType; import ru.soune.nocopy.entity.user.User; import ru.soune.nocopy.exception.TariffNotFoundException; +import ru.soune.nocopy.repository.CompanyRepository; import ru.soune.nocopy.repository.TariffInfoRepository; import ru.soune.nocopy.repository.TariffRepository; import ru.soune.nocopy.repository.UserRepository; @@ -40,6 +41,8 @@ public class TariffInfoService { private TokenOperationService tokenOperationService; + private CompanyRepository companyRepository; + public TariffInfo createTariffInfo(LocalDateTime startTariff, LocalDateTime endTariff, TariffStatus tariffStatus, Integer tokens) { TariffInfo tariffInfo = new TariffInfo(); @@ -157,13 +160,18 @@ public class TariffInfoService { @Transactional public void linkUserTariffInfo(User user, TariffInfo tariffInfo) { user.setPersonalTariffInfo(tariffInfo); + + userRepository.save(user); + } + + @Transactional + public void linkCompanyTariffInfo(User user, TariffInfo tariffInfo) { Company company = user.getCompany(); if (company != null) { company.setTariffInfo(tariffInfo); + companyRepository.save(company); } - - userRepository.save(user); } private void updateTariffInfoFromDTO(TariffInfo tariffInfo, TariffInfoDTO dto) {