Added nightly account value snapshots.

This commit is contained in:
Andrew Lalis 2022-02-23 21:48:43 +01:00
parent f2f1df5b42
commit 945b13524e
3 changed files with 39 additions and 6 deletions

View File

@ -0,0 +1,12 @@
package nl.andrewl.coyotecredit.dao;
import nl.andrewl.coyotecredit.model.Account;
import nl.andrewl.coyotecredit.model.AccountValueSnapshot;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.stereotype.Repository;
@Repository
public interface AccountValueSnapshotRepository extends JpaRepository<AccountValueSnapshot, Long> {
void deleteAllByAccount(Account account);
}

View File

@ -2,20 +2,23 @@ package nl.andrewl.coyotecredit.service;
import lombok.RequiredArgsConstructor;
import nl.andrewl.coyotecredit.ctl.dto.*;
import nl.andrewl.coyotecredit.dao.AccountRepository;
import nl.andrewl.coyotecredit.dao.TradeableRepository;
import nl.andrewl.coyotecredit.dao.TransactionRepository;
import nl.andrewl.coyotecredit.dao.TransferRepository;
import nl.andrewl.coyotecredit.dao.*;
import nl.andrewl.coyotecredit.model.*;
import nl.andrewl.coyotecredit.util.AccountNumberUtils;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Pageable;
import org.springframework.http.HttpStatus;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.MultiValueMap;
import org.springframework.web.server.ResponseStatusException;
import java.math.BigDecimal;
import java.time.LocalDateTime;
import java.time.ZoneOffset;
import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
@ -26,6 +29,7 @@ public class AccountService {
private final TransactionRepository transactionRepository;
private final TradeableRepository tradeableRepository;
private final TransferRepository transferRepository;
private final AccountValueSnapshotRepository valueSnapshotRepository;
@Transactional(readOnly = true)
public List<BalanceData> getTransferData(long accountId, User user) {
@ -169,4 +173,21 @@ public class AccountService {
}
accountRepository.save(account);
}
@Scheduled(cron = "@midnight")
@Transactional
public void doAccountValueSnapshots() {
Pageable pageable = Pageable.ofSize(10);
LocalDateTime timestamp = LocalDateTime.now(ZoneOffset.UTC);
Page<Account> page = accountRepository.findAll(pageable);
while (!page.isEmpty()) {
List<AccountValueSnapshot> snapshots = new ArrayList<>();
for (var account : page.getContent()) {
snapshots.add(new AccountValueSnapshot(account, timestamp, account.getTotalBalanceUsd()));
}
valueSnapshotRepository.saveAll(snapshots);
pageable = pageable.next();
page = accountRepository.findAll(pageable);
}
}
}

View File

@ -10,7 +10,6 @@ import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.HttpStatus;
import org.springframework.mail.javamail.JavaMailSender;
import org.springframework.mail.javamail.MimeMessageHelper;
import org.springframework.security.crypto.password.PasswordEncoder;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.server.ResponseStatusException;
@ -31,10 +30,10 @@ public class ExchangeService {
private final AccountRepository accountRepository;
private final TransactionRepository transactionRepository;
private final TradeableRepository tradeableRepository;
private final AccountValueSnapshotRepository accountValueSnapshotRepository;
private final UserRepository userRepository;
private final ExchangeInvitationRepository invitationRepository;
private final JavaMailSender mailSender;
private final PasswordEncoder passwordEncoder;
@Value("${coyote-credit.base-url}")
private String baseUrl;
@ -122,6 +121,7 @@ public class ExchangeService {
if (!userAccount.isAdmin()) {
throw new ResponseStatusException(HttpStatus.NOT_FOUND);
}
accountValueSnapshotRepository.deleteAllByAccount(account);
accountRepository.delete(account);
}