ResponseStatusException(NOT_FOUND)

PC38-RestSpring/src/main/java/sklep/rest/ProductEndpoint.java

@@ -3,10 +3,12 @@ package sklep.rest;
 import java.util.List;
 import java.util.Optional;
 
+import org.springframework.http.HttpStatus;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PathVariable;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.server.ResponseStatusException;
 
 import sklep.model.Product;
 import sklep.repository.ProductRepository;
@@ -26,9 +28,14 @@ public class ProductEndpoint {
 	}
 	
 	@GetMapping("/{id}")
-	public Optional<Product> getOneProduct(@PathVariable("id") Integer id) {
-		return productRepository.findById(id);
+	public Product getOneProduct(@PathVariable("id") Integer id) {
+		Optional<Product> product = productRepository.findById(id);
+		if(product.isPresent()) {
+			return product.get();
+		} else {
+			throw new ResponseStatusException(HttpStatus.NOT_FOUND);
 		}
 		
+	}
 
 }