Add get mapping for check balance

bifrurcated · Jan 13, 2024 · 0d1b10d · 0d1b10d
1 parent b9f08e0
commit 0d1b10d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/src/main/java/com/bifurcated/wallet/controller/WalletController.java b/src/main/java/com/bifurcated/wallet/controller/WalletController.java
@@ -12,7 +12,7 @@
 import java.util.UUID;
 
 @RestController
-@RequestMapping("api/v1/wallet")
+@RequestMapping("/api/v1")
 public class WalletController {
 
     private final WalletService walletService;
@@ -40,4 +40,11 @@ public WalletResponse wallet(@RequestBody WalletRequest request) {
 
         return new WalletResponse(wallet.getId(), wallet.getAmount());
     }
+
+    public record BalanceResponse(Float amount){}
+    @GetMapping("/wallets/{WALLET_UUID}")
+    public BalanceResponse balance(@PathVariable(value = "WALLET_UUID") UUID id) {
+        Float amount = walletService.amount(id);
+        return new BalanceResponse(amount);
+    }
 }
diff --git a/src/main/java/com/bifurcated/wallet/service/WalletService.java b/src/main/java/com/bifurcated/wallet/service/WalletService.java
@@ -8,7 +8,6 @@
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
-import java.util.Optional;
 import java.util.UUID;
 
 
@@ -39,7 +38,7 @@ public Wallet reduceAmount(UUID id, Float amount) {
         return walletRepo.save(wallet);
     }
 
-    public void amount() {
-
+    public Float amount(UUID id) {
+        return walletRepo.findById(id).orElseThrow(WalletNotFoundError::new).getAmount();
     }
 }