This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
package ru.soune.nocopy.service.tariff;
|
||||
|
||||
import jakarta.mail.MessagingException;
|
||||
import lombok.AllArgsConstructor;
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.springframework.transaction.annotation.Transactional;
|
||||
@@ -13,7 +14,9 @@ import ru.soune.nocopy.exception.TariffNotFoundException;
|
||||
import ru.soune.nocopy.repository.TariffInfoRepository;
|
||||
import ru.soune.nocopy.repository.TariffRepository;
|
||||
import ru.soune.nocopy.repository.UserRepository;
|
||||
import ru.soune.nocopy.service.mail.EmailService;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.time.LocalDateTime;
|
||||
import java.util.List;
|
||||
import java.util.stream.Collectors;
|
||||
@@ -28,6 +31,8 @@ public class TariffInfoService {
|
||||
|
||||
private UserRepository userRepository;
|
||||
|
||||
private EmailService emailService;
|
||||
|
||||
public TariffInfo createTariffInfo(LocalDateTime startTariff, LocalDateTime endTariff, TariffStatus tariffStatus,
|
||||
Integer tokens) {
|
||||
TariffInfo tariffInfo = new TariffInfo();
|
||||
@@ -41,7 +46,7 @@ public class TariffInfoService {
|
||||
}
|
||||
|
||||
@Transactional
|
||||
public void writeOffTokens(long userId, Integer value) {
|
||||
public void writeOffTokens(long userId, Integer value) throws MessagingException, IOException {
|
||||
User user = userRepository.findById(userId).orElseThrow();
|
||||
|
||||
TariffInfo activeTariffInfo;
|
||||
@@ -55,7 +60,7 @@ public class TariffInfoService {
|
||||
writeOffTokensForTariff(activeTariffInfo, value);
|
||||
}
|
||||
|
||||
public void writeOffTokensForTariff(TariffInfo activeTariffInfo, Integer value) {
|
||||
public void writeOffTokensForTariff(TariffInfo activeTariffInfo, Integer value) throws MessagingException, IOException {
|
||||
Integer tokens = activeTariffInfo.getTokens();
|
||||
Integer boughtTokens = activeTariffInfo.getBoughtTokens();
|
||||
|
||||
@@ -68,6 +73,11 @@ public class TariffInfoService {
|
||||
|
||||
tariffInfoRepository.save(activeTariffInfo);
|
||||
} else {
|
||||
int needTokens = boughtTokens - value;
|
||||
User user = userRepository.findByPersonalTariffInfo(activeTariffInfo);
|
||||
|
||||
emailService.sendTokensNotFoundEmail(user, tokens, Math.abs(needTokens));
|
||||
|
||||
throw new TariffNotFoundException("User not have tokens for protect");
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user