diff --git a/src/main/java/com/faf223/expensetrackerfaf/controller/ExpenseController.java b/src/main/java/com/faf223/expensetrackerfaf/controller/ExpenseController.java index 0c6b710..501035f 100644 --- a/src/main/java/com/faf223/expensetrackerfaf/controller/ExpenseController.java +++ b/src/main/java/com/faf223/expensetrackerfaf/controller/ExpenseController.java @@ -98,4 +98,9 @@ public class ExpenseController { if (!categories.isEmpty()) return ResponseEntity.ok(categories); else return ResponseEntity.notFound().build(); } + + @DeleteMapping("/delete/{id}") + public void deleteCategory(@PathVariable long id) { + expenseService.deleteExpenseById(id); + } } \ No newline at end of file diff --git a/src/main/java/com/faf223/expensetrackerfaf/controller/IncomeController.java b/src/main/java/com/faf223/expensetrackerfaf/controller/IncomeController.java index e43b5ba..164ec9e 100644 --- a/src/main/java/com/faf223/expensetrackerfaf/controller/IncomeController.java +++ b/src/main/java/com/faf223/expensetrackerfaf/controller/IncomeController.java @@ -96,4 +96,9 @@ public class IncomeController { if (!categories.isEmpty()) return ResponseEntity.ok(categories); else return ResponseEntity.notFound().build(); } + + @DeleteMapping("/delete/{id}") + public void deleteIncome(@PathVariable long id) { + incomeService.deleteIncomeById(id); + } } \ No newline at end of file diff --git a/src/main/java/com/faf223/expensetrackerfaf/service/ExpenseService.java b/src/main/java/com/faf223/expensetrackerfaf/service/ExpenseService.java index 4209fd6..7976a33 100644 --- a/src/main/java/com/faf223/expensetrackerfaf/service/ExpenseService.java +++ b/src/main/java/com/faf223/expensetrackerfaf/service/ExpenseService.java @@ -40,4 +40,8 @@ public class ExpenseService implements ITransactionService { public Expense getTransactionById(long id) { return expenseRepository.findById(id).orElse(null); } + + public void deleteExpenseById(long id) { + expenseRepository.deleteById(id); + } } \ No newline at end of file diff --git a/src/main/java/com/faf223/expensetrackerfaf/service/IncomeService.java b/src/main/java/com/faf223/expensetrackerfaf/service/IncomeService.java index 8287d01..dcfb2e9 100644 --- a/src/main/java/com/faf223/expensetrackerfaf/service/IncomeService.java +++ b/src/main/java/com/faf223/expensetrackerfaf/service/IncomeService.java @@ -40,4 +40,8 @@ public class IncomeService implements ITransactionService { public Income getTransactionById(long id) { return incomeRepository.findById(id).orElse(null); } + + public void deleteIncomeById(long id) { + incomeRepository.deleteById(id); + } }