refactor code, add interfaces for services

This commit is contained in:
mirrerror
2023-10-24 19:59:23 +03:00
parent d402514ecb
commit 65c5c3f2df
10 changed files with 89 additions and 30 deletions

View File

@@ -35,7 +35,7 @@ public class ExpenseController {
@GetMapping() @GetMapping()
@PreAuthorize("hasRole('ADMIN')") @PreAuthorize("hasRole('ADMIN')")
public ResponseEntity<List<ExpenseDTO>> getAllExpenses() { public ResponseEntity<List<ExpenseDTO>> getAllExpenses() {
List<ExpenseDTO> expenses = expenseService.getExpenses().stream().map(expenseMapper::toDto).collect(Collectors.toList()); List<ExpenseDTO> expenses = expenseService.getTransactions().stream().map(expenseMapper::toDto).collect(Collectors.toList());
if (!expenses.isEmpty()) return ResponseEntity.ok(expenses); if (!expenses.isEmpty()) return ResponseEntity.ok(expenses);
else return ResponseEntity.notFound().build(); else return ResponseEntity.notFound().build();
} }
@@ -53,7 +53,7 @@ public class ExpenseController {
User user = userService.getUserByEmail(email); User user = userService.getUserByEmail(email);
expense.setUser(user); expense.setUser(user);
expenseService.createOrUpdateExpense(expense); expenseService.createOrUpdate(expense);
ExpenseDTO createdExpenseDTO = expenseMapper.toDto(expense); ExpenseDTO createdExpenseDTO = expenseMapper.toDto(expense);
return ResponseEntity.status(HttpStatus.CREATED).body(createdExpenseDTO); return ResponseEntity.status(HttpStatus.CREATED).body(createdExpenseDTO);
} }
@@ -67,7 +67,7 @@ public class ExpenseController {
BindingResult bindingResult) { BindingResult bindingResult) {
Expense expense = expenseMapper.toExpense(expenseDTO); Expense expense = expenseMapper.toExpense(expenseDTO);
if (!bindingResult.hasErrors()) { if (!bindingResult.hasErrors()) {
expenseService.createOrUpdateExpense(expense); expenseService.createOrUpdate(expense);
return ResponseEntity.ok(expenseMapper.toDto(expense)); return ResponseEntity.ok(expenseMapper.toDto(expense));
} else { } else {
return ResponseEntity.notFound().build(); return ResponseEntity.notFound().build();
@@ -82,7 +82,7 @@ public class ExpenseController {
if (authentication != null && authentication.getPrincipal() instanceof UserDetails userDetails) { if (authentication != null && authentication.getPrincipal() instanceof UserDetails userDetails) {
String email = userDetails.getUsername(); String email = userDetails.getUsername();
List<ExpenseDTO> expenses = expenseService.getExpensesByEmail(email).stream().map(expenseMapper::toDto).collect(Collectors.toList()); List<ExpenseDTO> expenses = expenseService.getTransactionsByEmail(email).stream().map(expenseMapper::toDto).collect(Collectors.toList());
if (!expenses.isEmpty()) { if (!expenses.isEmpty()) {
return ResponseEntity.ok(expenses); return ResponseEntity.ok(expenses);
@@ -91,4 +91,11 @@ public class ExpenseController {
return ResponseEntity.notFound().build(); return ResponseEntity.notFound().build();
} }
@GetMapping("/categories")
public ResponseEntity<List<ExpenseCategory>> getAllCategories() {
List<ExpenseCategory> categories = expenseCategoryService.getAllCategories();
if (!categories.isEmpty()) return ResponseEntity.ok(categories);
else return ResponseEntity.notFound().build();
}
} }

View File

@@ -1,11 +1,12 @@
package com.faf223.expensetrackerfaf.controller; package com.faf223.expensetrackerfaf.controller;
import com.faf223.expensetrackerfaf.dto.ExpenseDTO;
import com.faf223.expensetrackerfaf.dto.IncomeCreationDTO; import com.faf223.expensetrackerfaf.dto.IncomeCreationDTO;
import com.faf223.expensetrackerfaf.dto.IncomeDTO; import com.faf223.expensetrackerfaf.dto.IncomeDTO;
import com.faf223.expensetrackerfaf.dto.mappers.IncomeMapper; import com.faf223.expensetrackerfaf.dto.mappers.IncomeMapper;
import com.faf223.expensetrackerfaf.model.Income; import com.faf223.expensetrackerfaf.model.Income;
import com.faf223.expensetrackerfaf.model.IncomeCategory;
import com.faf223.expensetrackerfaf.model.User; import com.faf223.expensetrackerfaf.model.User;
import com.faf223.expensetrackerfaf.service.IncomeCategoryService;
import com.faf223.expensetrackerfaf.service.IncomeService; import com.faf223.expensetrackerfaf.service.IncomeService;
import com.faf223.expensetrackerfaf.service.UserService; import com.faf223.expensetrackerfaf.service.UserService;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
@@ -29,11 +30,12 @@ public class IncomeController {
private final IncomeService incomeService; private final IncomeService incomeService;
private final UserService userService; private final UserService userService;
private final IncomeMapper incomeMapper; private final IncomeMapper incomeMapper;
private final IncomeCategoryService incomeCategoryService;
@GetMapping() @GetMapping()
@PreAuthorize("hasRole('ADMIN')") @PreAuthorize("hasRole('ADMIN')")
public ResponseEntity<List<IncomeDTO>> getAllIncomes() { public ResponseEntity<List<IncomeDTO>> getAllIncomes() {
List<IncomeDTO> incomes = incomeService.getIncomes().stream().map(incomeMapper::toDto).collect(Collectors.toList()); List<IncomeDTO> incomes = incomeService.getTransactions().stream().map(incomeMapper::toDto).collect(Collectors.toList());
if (!incomes.isEmpty()) return ResponseEntity.ok(incomes); if (!incomes.isEmpty()) return ResponseEntity.ok(incomes);
else return ResponseEntity.notFound().build(); else return ResponseEntity.notFound().build();
} }
@@ -51,7 +53,7 @@ public class IncomeController {
income.setUser(user); income.setUser(user);
System.out.println(income); System.out.println(income);
incomeService.createOrUpdateIncome(income); incomeService.createOrUpdate(income);
IncomeDTO createdIncomeDTO = incomeMapper.toDto(income); IncomeDTO createdIncomeDTO = incomeMapper.toDto(income);
return ResponseEntity.status(HttpStatus.CREATED).body(createdIncomeDTO); return ResponseEntity.status(HttpStatus.CREATED).body(createdIncomeDTO);
} }
@@ -64,7 +66,7 @@ public class IncomeController {
BindingResult bindingResult) { BindingResult bindingResult) {
Income income = incomeMapper.toIncome(incomeDTO); Income income = incomeMapper.toIncome(incomeDTO);
if (!bindingResult.hasErrors()) { if (!bindingResult.hasErrors()) {
incomeService.createOrUpdateIncome(income); incomeService.createOrUpdate(income);
return ResponseEntity.ok(incomeMapper.toDto(income)); return ResponseEntity.ok(incomeMapper.toDto(income));
} else { } else {
return ResponseEntity.notFound().build(); return ResponseEntity.notFound().build();
@@ -79,7 +81,7 @@ public class IncomeController {
if (authentication != null && authentication.getPrincipal() instanceof UserDetails userDetails) { if (authentication != null && authentication.getPrincipal() instanceof UserDetails userDetails) {
String email = userDetails.getUsername(); String email = userDetails.getUsername();
List<IncomeDTO> incomes = incomeService.getIncomesByEmail(email).stream().map(incomeMapper::toDto).collect(Collectors.toList()); List<IncomeDTO> incomes = incomeService.getTransactionsByEmail(email).stream().map(incomeMapper::toDto).collect(Collectors.toList());
if (!incomes.isEmpty()) { if (!incomes.isEmpty()) {
return ResponseEntity.ok(incomes); return ResponseEntity.ok(incomes);
@@ -88,4 +90,11 @@ public class IncomeController {
return ResponseEntity.notFound().build(); return ResponseEntity.notFound().build();
} }
@GetMapping("/categories")
public ResponseEntity<List<IncomeCategory>> getAllCategories() {
List<IncomeCategory> categories = incomeCategoryService.getAllCategories();
if (!categories.isEmpty()) return ResponseEntity.ok(categories);
else return ResponseEntity.notFound().build();
}
} }

View File

@@ -31,7 +31,7 @@ public class ExpenseMapper {
public Expense toExpense(ExpenseCreationDTO expenseDTO) { public Expense toExpense(ExpenseCreationDTO expenseDTO) {
return new Expense(expenseCategoryService.getExpenseCategory(expenseDTO.getExpenseCategory()), LocalDate.now(), expenseDTO.getAmount()); return new Expense(expenseCategoryService.getCategoryById(expenseDTO.getExpenseCategory()), LocalDate.now(), expenseDTO.getAmount());
} }
} }

View File

@@ -31,8 +31,7 @@ public class IncomeMapper {
} }
public Income toIncome(IncomeCreationDTO incomeDTO) { public Income toIncome(IncomeCreationDTO incomeDTO) {
return new Income(incomeCategoryService.getCategoryById(incomeDTO.getIncomeCategory()), LocalDate.now(), incomeDTO.getAmount());
return new Income(incomeCategoryService.getExpenseCategory(incomeDTO.getIncomeCategory()), LocalDate.now(), incomeDTO.getAmount());
} }
} }

View File

@@ -5,14 +5,22 @@ import com.faf223.expensetrackerfaf.repository.ExpenseCategoryRepository;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import java.util.List;
@Service @Service
@RequiredArgsConstructor @RequiredArgsConstructor
public class ExpenseCategoryService { public class ExpenseCategoryService implements ICategoryService {
private final ExpenseCategoryRepository expenseCategoryRepository; private final ExpenseCategoryRepository expenseCategoryRepository;
public ExpenseCategory getExpenseCategory(long category) { @Override
return expenseCategoryRepository.getReferenceById(category); public List<ExpenseCategory> getAllCategories() {
return expenseCategoryRepository.findAll();
}
@Override
public ExpenseCategory getCategoryById(long id) {
return expenseCategoryRepository.getReferenceById(id);
} }
} }

View File

@@ -2,6 +2,7 @@ package com.faf223.expensetrackerfaf.service;
import com.faf223.expensetrackerfaf.model.Credential; import com.faf223.expensetrackerfaf.model.Credential;
import com.faf223.expensetrackerfaf.model.Expense; import com.faf223.expensetrackerfaf.model.Expense;
import com.faf223.expensetrackerfaf.model.IMoneyTransaction;
import com.faf223.expensetrackerfaf.repository.CredentialRepository; import com.faf223.expensetrackerfaf.repository.CredentialRepository;
import com.faf223.expensetrackerfaf.repository.ExpenseRepository; import com.faf223.expensetrackerfaf.repository.ExpenseRepository;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
@@ -13,16 +14,16 @@ import java.util.Optional;
@Service @Service
@RequiredArgsConstructor @RequiredArgsConstructor
public class ExpenseService { public class ExpenseService implements ITransactionService {
private final ExpenseRepository expenseRepository; private final ExpenseRepository expenseRepository;
private final CredentialRepository credentialRepository; private final CredentialRepository credentialRepository;
public void createOrUpdateExpense(Expense expense) { public void createOrUpdate(IMoneyTransaction expense) {
expenseRepository.save(expense); expenseRepository.save((Expense) expense);
} }
public List<Expense> getExpensesByEmail(String email) { public List<Expense> getTransactionsByEmail(String email) {
Optional<Credential> credential = credentialRepository.findByEmail(email); Optional<Credential> credential = credentialRepository.findByEmail(email);
if (credential.isPresent()) { if (credential.isPresent()) {
@@ -32,11 +33,11 @@ public class ExpenseService {
return new ArrayList<>(); return new ArrayList<>();
} }
public List<Expense> getExpenses() { public List<Expense> getTransactions() {
return expenseRepository.findAll(); return expenseRepository.findAll();
} }
public Expense getExpenseById(long id) { public Expense getTransactionById(long id) {
return expenseRepository.findById(id).orElse(null); return expenseRepository.findById(id).orElse(null);
} }
} }

View File

@@ -0,0 +1,12 @@
package com.faf223.expensetrackerfaf.service;
import com.faf223.expensetrackerfaf.model.IMoneyTransactionCategory;
import java.util.List;
public interface ICategoryService {
List<? extends IMoneyTransactionCategory> getAllCategories();
IMoneyTransactionCategory getCategoryById(long id);
}

View File

@@ -0,0 +1,14 @@
package com.faf223.expensetrackerfaf.service;
import com.faf223.expensetrackerfaf.model.IMoneyTransaction;
import java.util.List;
public interface ITransactionService {
void createOrUpdate(IMoneyTransaction transaction);
List<? extends IMoneyTransaction> getTransactions();
List<? extends IMoneyTransaction> getTransactionsByEmail(String email);
IMoneyTransaction getTransactionById(long id);
}

View File

@@ -5,13 +5,21 @@ import com.faf223.expensetrackerfaf.repository.IncomeCategoryRepository;
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import java.util.List;
@Service @Service
@RequiredArgsConstructor @RequiredArgsConstructor
public class IncomeCategoryService { public class IncomeCategoryService implements ICategoryService {
private final IncomeCategoryRepository incomeCategoryRepository; private final IncomeCategoryRepository incomeCategoryRepository;
public IncomeCategory getExpenseCategory(long category) { @Override
return incomeCategoryRepository.getReferenceById(category); public List<IncomeCategory> getAllCategories() {
return incomeCategoryRepository.findAll();
}
@Override
public IncomeCategory getCategoryById(long id) {
return incomeCategoryRepository.getReferenceById(id);
} }
} }

View File

@@ -1,6 +1,7 @@
package com.faf223.expensetrackerfaf.service; package com.faf223.expensetrackerfaf.service;
import com.faf223.expensetrackerfaf.model.Credential; import com.faf223.expensetrackerfaf.model.Credential;
import com.faf223.expensetrackerfaf.model.IMoneyTransaction;
import com.faf223.expensetrackerfaf.model.Income; import com.faf223.expensetrackerfaf.model.Income;
import com.faf223.expensetrackerfaf.repository.CredentialRepository; import com.faf223.expensetrackerfaf.repository.CredentialRepository;
import com.faf223.expensetrackerfaf.repository.IncomeRepository; import com.faf223.expensetrackerfaf.repository.IncomeRepository;
@@ -13,20 +14,20 @@ import java.util.Optional;
@Service @Service
@RequiredArgsConstructor @RequiredArgsConstructor
public class IncomeService { public class IncomeService implements ITransactionService {
private final IncomeRepository incomeRepository; private final IncomeRepository incomeRepository;
private final CredentialRepository credentialRepository; private final CredentialRepository credentialRepository;
public void createOrUpdateIncome(Income income) { public void createOrUpdate(IMoneyTransaction income) {
incomeRepository.save(income); incomeRepository.save((Income) income);
} }
public List<Income> getIncomes() { public List<Income> getTransactions() {
return incomeRepository.findAll(); return incomeRepository.findAll();
} }
public List<Income> getIncomesByEmail(String email) { public List<Income> getTransactionsByEmail(String email) {
Optional<Credential> credential = credentialRepository.findByEmail(email); Optional<Credential> credential = credentialRepository.findByEmail(email);
if (credential.isPresent()) { if (credential.isPresent()) {
@@ -36,7 +37,7 @@ public class IncomeService {
return new ArrayList<>(); return new ArrayList<>();
} }
public Income getIncomeById(long id) { public Income getTransactionById(long id) {
return incomeRepository.findById(id).orElse(null); return incomeRepository.findById(id).orElse(null);
} }
} }