dev add context for notifcations

2026-05-13 16:31:55 +07:00
parent cc77f481dc
commit 675c93c0c6
1 changed files with 5 additions and 4 deletions
@@ -6,6 +6,7 @@ import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 import ru.soune.nocopy.dto.tarriff.TariffInfoDTO;
+import ru.soune.nocopy.entity.company.Company;
 import ru.soune.nocopy.entity.tarif.Tariff;
 import ru.soune.nocopy.entity.tarif.TariffInfo;
 import ru.soune.nocopy.entity.tarif.TariffStatus;
@@ -51,16 +52,16 @@ public class TariffInfoService {
        return tariffInfoRepository.save(tariffInfo);
    }

-//    @Transactional
+    @Transactional
    public void writeOffTokens(long userId, Integer value, OperationType operationType) throws MessagingException, IOException {
        User user = userRepository.findById(userId).orElseThrow();
-
+        Company company = user.getCompany();
        TariffInfo activeTariffInfo;

-        if (user.getCompany() == null) {
+        if (company == null) {
            activeTariffInfo = user.getPersonalTariffInfo();
        } else {
-            activeTariffInfo = user.getCompany().getTariffInfo();
+            activeTariffInfo = company.getTariffInfo();
        }

        writeOffTokensForTariff(activeTariffInfo, value);