dev check violation for lawcase,complaint

2026-05-19 15:45:40 +07:00
parent e633d43a74
commit 4dc479e31b
2 changed files with 11 additions and 2 deletions
@@ -245,6 +245,7 @@ public class PaymentService {
            tariffInfo.setPaymentMethodId(paymentMethodId);

            tariffInfoService.linkUserTariffInfo(user, tariffInfoService.addTariffInfo(tariffInfo));
+            tariffInfoService.linkCompanyTariffInfo(user, tariffInfoService.addTariffInfo(tariffInfo));
        }
    }
 }
@@ -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) {