Change getting expenses/incomes logic

This commit is contained in:
Dmitrii Cravcenco
2023-10-23 18:52:38 +03:00
parent a3bbe4433e
commit 4fed3dba9e
4 changed files with 52 additions and 26 deletions

View File

@@ -1,11 +1,10 @@
package com.faf223.expensetrackerfaf.service;
import com.faf223.expensetrackerfaf.model.Credential;
import com.faf223.expensetrackerfaf.model.Expense;
import com.faf223.expensetrackerfaf.model.User;
import com.faf223.expensetrackerfaf.repository.CredentialRepository;
import com.faf223.expensetrackerfaf.repository.ExpenseRepository;
import com.faf223.expensetrackerfaf.repository.UserRepository;
import lombok.RequiredArgsConstructor;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import java.util.ArrayList;
@@ -17,17 +16,17 @@ import java.util.Optional;
public class ExpenseService {
private final ExpenseRepository expenseRepository;
private final UserRepository userRepository;
private final CredentialRepository credentialRepository;
public void createOrUpdateExpense(Expense expense) {
expenseRepository.save(expense);
}
public List<Expense> getExpensesByUserId(String userUuid) {
public List<Expense> getExpensesByEmail(String email) {
Optional<User> user = userRepository.getUserByUserUuid(userUuid);
if (user.isPresent()) {
return expenseRepository.findByUser(user.get());
Optional<Credential> credential = credentialRepository.findByEmail(email);
if (credential.isPresent()) {
return expenseRepository.findByUser(credential.get().getUser());
}
return new ArrayList<>();