From 675c93c0c6509006381da1c7ff138ee97269fab7 Mon Sep 17 00:00:00 2001
From: backdev-1 <backdev-1@xxxxxxxxx.su>
Date: Wed, 13 May 2026 16:31:55 +0700
Subject: [PATCH] dev add context for notifcations

---
 .../soune/nocopy/service/tariff/TariffInfoService.java   | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

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 6e91ed5..7076cb7 100644
--- a/src/main/java/ru/soune/nocopy/service/tariff/TariffInfoService.java
+++ b/src/main/java/ru/soune/nocopy/service/tariff/TariffInfoService.java
@@ -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);