Updated exception handling

src/main/java/org/hsbc/controller/PmsController.java

@@ -1,6 +1,7 @@
 package org.hsbc.controller;
 
 import org.hsbc.entity.PmsEntity;
+import org.hsbc.exception.InvalidPmsIdException;
 import org.hsbc.service.PmsService;
 import org.hsbc.service.PmsServiceimp;
 import org.slf4j.Logger;
@@ -26,7 +27,7 @@ public List<PmsEntity> getAllAssets() {
     return service.getAllAssets();
 }
     @GetMapping("/{id}")
-    public PmsEntity getAssetById(@PathVariable Long id) {
+    public PmsEntity getAssetById(@PathVariable Long id) throws InvalidPmsIdException {
         return service.getAssetById(id);
     }
 
@@ -39,25 +40,25 @@ public PmsEntity addAsset(@RequestBody PmsEntity asset) {
         }
 
         @DeleteMapping("/remove/{id}")
-        public String removeAsset(@PathVariable Long id) {
+        public String removeAsset(@PathVariable Long id) throws InvalidPmsIdException {
             service.removeAsset(id);
             return "Asset removed successfully";
         }
 
         @PutMapping("/update-quantity/{id}")
         public PmsEntity updateQuantity(
                 @PathVariable Long id,
-                @RequestParam int quantity) {
+                @RequestParam int quantity) throws InvalidPmsIdException {
             return service.updateQuantity(id, quantity);
         }
 
         @GetMapping("/pl/{id}")
-        public double getPL(@PathVariable Long id) {
+        public double getPL(@PathVariable Long id) throws InvalidPmsIdException {
             return service.calculatePL(id);
         }
 
         @GetMapping("/pl-percentage/{id}")
-        public double getPLPercentage(@PathVariable Long id) {
+        public double getPLPercentage(@PathVariable Long id) throws InvalidPmsIdException {
             return service.calculatePLPercentage(id);
         }
 

src/main/java/org/hsbc/exception/InvalidPmsIdException.java

@@ -0,0 +1,25 @@
+package org.hsbc.exception;
+
+public class InvalidPmsIdException extends Exception {
+
+    public InvalidPmsIdException() {
+    }
+
+    public InvalidPmsIdException(String message) {
+        super(message);
+    }
+
+    public InvalidPmsIdException(Throwable cause) {
+        super(cause);
+    }
+
+    public InvalidPmsIdException(String message, Throwable cause) {
+        super(message, cause);
+    }
+
+    public InvalidPmsIdException(String message, Throwable cause, boolean enableSuppression,
+                                 boolean writableStackTrace) {
+        super(message, cause, enableSuppression, writableStackTrace);
+    }
+
+}

src/main/java/org/hsbc/exception/InvalidTransactionIdException.java

@@ -0,0 +1,25 @@
+package org.hsbc.exception;
+
+public class InvalidTransactionIdException extends Exception {
+
+    public InvalidTransactionIdException() {
+    }
+
+    public InvalidTransactionIdException(String message) {
+        super(message);
+    }
+
+    public InvalidTransactionIdException(Throwable cause) {
+        super(cause);
+    }
+
+    public InvalidTransactionIdException(String message, Throwable cause) {
+        super(message, cause);
+    }
+
+    public InvalidTransactionIdException(String message, Throwable cause, boolean enableSuppression,
+                                         boolean writableStackTrace) {
+        super(message, cause, enableSuppression, writableStackTrace);
+    }
+
+}

src/main/java/org/hsbc/service/PmsService.java

@@ -1,23 +1,25 @@
 package org.hsbc.service;
 
-import org.hsbc.entity.PmsEntity;
-
 import java.util.List;
+import org.hsbc.entity.PmsEntity;
+import org.hsbc.exception.InvalidPmsIdException;
 
 public interface PmsService {
+
     PmsEntity addAsset(PmsEntity asset);
 
-    void removeAsset(Long id);
+    void removeAsset(Long id) throws InvalidPmsIdException;
 
-    PmsEntity updateQuantity(Long id, int newQuantity);
+    PmsEntity updateQuantity(Long id, int newQuantity) throws InvalidPmsIdException;
 
-    double calculatePL(Long id);
+    double calculatePL(Long id) throws InvalidPmsIdException;
 
-    double calculatePLPercentage(Long id);
+    double calculatePLPercentage(Long id) throws InvalidPmsIdException;
 
     double getTotalPortfolioValue();
+
     List<PmsEntity> getAllAssets();
-    PmsEntity getAssetById(Long id);
 
+    PmsEntity getAssetById(Long id) throws InvalidPmsIdException;
 
-}
+}