diff --git a/src/main/java/com/store/controllers/Products.kt b/src/main/java/com/store/controllers/Products.kt index 7f4ecd5..2307280 100644 --- a/src/main/java/com/store/controllers/Products.kt +++ b/src/main/java/com/store/controllers/Products.kt @@ -63,4 +63,13 @@ open class Products { val products = productService.findProducts(name, type, status) return ResponseEntity(products, HttpStatus.OK) } + + @GetMapping("/products/{id}/inventory") + fun getProductInventory(@PathVariable("id") id: Int): Int { + try { + return productService.getProductInventory(id) + } catch (e: NoSuchElementException) { + throw NotFoundException(e.message!!) + } + } } diff --git a/src/main/java/com/store/services/ProductService.kt b/src/main/java/com/store/services/ProductService.kt index bd0b2b8..fe7ab4d 100644 --- a/src/main/java/com/store/services/ProductService.kt +++ b/src/main/java/com/store/services/ProductService.kt @@ -31,4 +31,9 @@ class ProductService { fun findProducts(name:String?, type:String?, status:String?): List { return DB.findProducts(name, type, status) } + + fun getProductInventory(productId: Int): Int { + val product = DB.findProduct(productId) + return product.inventory + } } \ No newline at end of file