dev check violation for lawcase,complaint

2026-05-19 16:22:22 +07:00
parent 24bae18311
commit 3a1f4ea9d9
2 changed files with 7 additions and 14 deletions
@@ -165,10 +165,11 @@ public class TariffInfoService {
    }

    @Transactional
-    public void linkCompanyTariffInfo(User user, TariffInfo tariffInfo) {
-        Company company = user.getCompany();
+    public void linkCompanyTariffInfo(Long userId, TariffInfo tariffInfo) {
+        User user = userRepository.findById(userId).orElse(null);

-        if (company != null) {
+        if (user != null && user.getCompany() != null) {
+            Company company = user.getCompany();
            company.setTariffInfo(tariffInfo);

            companyRepository.save(company);