dev split tariff by account types

2026-02-09 17:16:08 +07:00
parent 5a70c091f8
commit 7cc744829b
8 changed files with 82 additions and 23 deletions
@@ -6,6 +6,7 @@ import org.springframework.transaction.annotation.Transactional;
 import ru.soune.nocopy.dto.tarriff.TariffDTO;
 import ru.soune.nocopy.entity.tarif.Tariff;
 import ru.soune.nocopy.entity.tarif.TariffType;
+import ru.soune.nocopy.entity.user.User;
 import ru.soune.nocopy.exception.TariffNotFoundException;
 import ru.soune.nocopy.repository.TariffRepository;

@@ -32,6 +33,15 @@ public class TariffService {
        tariffRepository.save(tariff);
    }

+    public List<TariffDTO> getTariffByAccountType(User user) {
+        String typeAccount = user.canManageCompanySettings() ? "b2b": "b2c";
+
+        return tariffRepository.findByTariffAccountType(typeAccount)
+                .stream()
+                .map(this::convertToDTO)
+                .collect(Collectors.toList());
+    }
+
    public List<TariffDTO> getAllTariffs() {
        return tariffRepository.findAllByOrderByPriceAsc()
                .stream()