Added nightly account value snapshots.
This commit is contained in:
		
							parent
							
								
									f2f1df5b42
								
							
						
					
					
						commit
						945b13524e
					
				| 
						 | 
				
			
			@ -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);
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			@ -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);
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -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);
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in New Issue