✨ enable the change of quantities for all Products

josuablejeru · Aug 17, 2023 · 65bb7fc · 65bb7fc
1 parent 84d9808
commit 65bb7fc
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 2 deletions.
diff --git a/src/main/java/com/josuablejeru/resupplyapi/controllers/ProductsController.java b/src/main/java/com/josuablejeru/resupplyapi/controllers/ProductsController.java
@@ -1,5 +1,6 @@
 package com.josuablejeru.resupplyapi.controllers;
 
+import com.josuablejeru.resupplyapi.controllers.errors.ProductNotFoundException;
 import com.josuablejeru.resupplyapi.controllers.requestbody.ChangeProductQuantity;
 import com.josuablejeru.resupplyapi.controllers.response.UpdateProductResponse;
 import com.josuablejeru.resupplyapi.models.Product;
@@ -13,6 +14,7 @@
 import org.springframework.web.bind.annotation.*;
 
 import java.util.List;
+import java.util.Optional;
 
 @RestController
 @RequestMapping("/v1/products")
@@ -45,12 +47,17 @@ public ResponseEntity<Product> requestProduct(@RequestBody Product product) {
     @PutMapping(value = "{productId}/quantity", produces = MediaType.APPLICATION_JSON_VALUE)
     @ResponseBody
     public UpdateProductResponse changeQuantity(@PathVariable("productId") String productId, @RequestBody ChangeProductQuantity body) {
-        Integer newQuantity = body.getQuantity();
+        Integer requestedQuantity = body.getQuantity();
+
+        Optional<Product> product = productService.getById(productId);
+        product.orElseThrow(() -> new ProductNotFoundException("Product not found"));
+
+        Integer newCalculatedQuantity = productService.changeQuantity(product.get(), requestedQuantity);
 
         UpdateProductResponse response = new UpdateProductResponse();
 
         response.setProductId(productId);
-        response.setQuantity(newQuantity);
+        response.setQuantity(newCalculatedQuantity);
 
         return response;
     }

diff --git a/src/main/java/com/josuablejeru/resupplyapi/services/ProductService.java b/src/main/java/com/josuablejeru/resupplyapi/services/ProductService.java
@@ -6,6 +6,7 @@
 import org.springframework.stereotype.Service;
 
 import java.util.List;
+import java.util.Optional;
 
 @Service
 public class ProductService {
@@ -24,4 +25,17 @@ public List<Product> list() {
     public Product add(Product product) {
         return productRepository.save(product);
     }
+
+    public Optional<Product> getById(String id) {
+        return productRepository.findById(id);
+    }
+
+    public Integer changeQuantity(Product product, Integer quantityTick) {
+        Integer currentQuantity = product.getQuantity();
+        product.setQuantity(currentQuantity + quantityTick);
+
+        productRepository.save(product);
+
+        return product.getQuantity();
+    }
 }