99 lines
3.5 KiB
Java
99 lines
3.5 KiB
Java
|
package ru.soune.nocopy.service.tariff;
|
||
|
|
|
||
|
|
import lombok.RequiredArgsConstructor;
|
||
|
|
import org.springframework.boot.context.event.ApplicationReadyEvent;
|
||
|
|
import org.springframework.context.event.EventListener;
|
||
|
|
import org.springframework.stereotype.Service;
|
||
|
|
import org.springframework.transaction.annotation.Transactional;
|
||
|
|
import ru.soune.nocopy.entity.tarif.Tariff;
|
||
|
|
import ru.soune.nocopy.entity.tarif.TariffType;
|
||
|
|
import ru.soune.nocopy.repository.TariffRepository;
|
||
|
|
|
||
|
|
import java.util.Arrays;
|
||
|
|
import java.util.HashMap;
|
||
|
|
import java.util.List;
|
||
|
|
import java.util.Map;
|
||
|
|
|
||
|
|
@Service
|
||
|
|
@RequiredArgsConstructor
|
||
|
|
public class TariffInitializationService {
|
||
|
|
|
||
|
|
private final TariffRepository tariffRepository;
|
||
|
|
|
||
|
|
@EventListener(ApplicationReadyEvent.class)
|
||
|
|
@Transactional
|
||
|
|
public void initializeDefaultTariffs() {
|
||
|
|
List<Tariff> existingTariffs = tariffRepository.findAll();
|
||
|
|
|
||
|
|
if (existingTariffs.isEmpty()) {
|
||
|
|
createDefaultTariffs();
|
||
|
|
} else {
|
||
|
|
Map<TariffType, Boolean> existingTypes = new HashMap<>();
|
||
|
|
existingTariffs.forEach(tariff -> existingTypes.put(TariffType.valueOf(tariff.getType()), true));
|
||
|
|
|
||
|
|
Arrays.stream(TariffType.values())
|
||
|
|
.filter(type -> !existingTypes.containsKey(type))
|
||
|
|
.forEach(this::createDefaultTariff);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private void createDefaultTariffs() {
|
||
|
|
Arrays.stream(TariffType.values())
|
||
|
|
.forEach(this::createDefaultTariff);
|
||
|
|
}
|
||
|
|
|
||
|
|
private void createDefaultTariff(TariffType type) {
|
||
|
|
Tariff tariff = new Tariff();
|
||
|
|
tariff.setType(type.name());
|
||
|
|
|
||
|
|
switch (type) {
|
||
|
|
case DEMO:
|
||
|
|
tariff.setName("Демо");
|
||
|
|
tariff.setPrice(TariffConstants.DEMO_PRICE);
|
||
|
|
tariff.setTokens(TariffConstants.DEMO_TOKENS);
|
||
|
|
tariff.setMaxFilesCount(10);
|
||
|
|
tariff.setDiskSize(1024L * 1024 * 100);
|
||
|
|
tariff.setMaxUsers(1L);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case START:
|
||
|
|
tariff.setName("Стартовый");
|
||
|
|
tariff.setPrice(TariffConstants.START_PRICE);
|
||
|
|
tariff.setTokens(TariffConstants.START_TOKENS);
|
||
|
|
tariff.setMaxFilesCount(50);
|
||
|
|
tariff.setDiskSize(1024L * 1024 * 500);
|
||
|
|
tariff.setMaxUsers(3L);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case BASIC:
|
||
|
|
tariff.setName("Базовый");
|
||
|
|
tariff.setPrice(TariffConstants.BASIC_PRICE);
|
||
|
|
tariff.setTokens(TariffConstants.BASIC_TOKENS);
|
||
|
|
tariff.setMaxFilesCount(200);
|
||
|
|
tariff.setDiskSize(1024L * 1024 * 1024 * 2L);
|
||
|
|
tariff.setMaxUsers(10L);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case PRO:
|
||
|
|
tariff.setName("Профессиональный");
|
||
|
|
tariff.setPrice(TariffConstants.PRO_PRICE);
|
||
|
|
tariff.setTokens(TariffConstants.PRO_TOKENS);
|
||
|
|
tariff.setMaxFilesCount(1000);
|
||
|
|
tariff.setDiskSize(1024L * 1024 * 1024 * 10L);
|
||
|
|
tariff.setMaxUsers(25L);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case ENTERPRISE:
|
||
|
|
tariff.setName("Корпоративный");
|
||
|
|
tariff.setPrice(TariffConstants.ENTERPRISE_PRICE);
|
||
|
|
tariff.setTokens(TariffConstants.ENTERPRISE_TOKENS);
|
||
|
|
tariff.setMaxFilesCount(5000);
|
||
|
|
tariff.setDiskSize(1024L * 1024 * 1024 * 50L);
|
||
|
|
tariff.setMaxUsers(100L);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
tariffRepository.save(tariff);
|
||
|
|
}
|
||
|
|
}
|