product-api: dto exercise sol

Author: mama-samba-braima

src/main/java/com/amigoscode/product/ProductController.java

@@ -16,12 +16,12 @@ public ProductController(ProductService productService) {
     }
 
     @GetMapping
-    public List<Product> getAllProducts() {
+    public List<ProductResponse> getAllProducts() {
         return productService.getAllProducts();
     }
 
     @GetMapping("{id}")
-    public Product getProductById(@PathVariable("id") UUID id) {
+    public ProductResponse getProductById(@PathVariable("id") UUID id) {
         return productService.getProductById(id);
     }
 

src/main/java/com/amigoscode/product/ProductResponse.java

@@ -0,0 +1,18 @@
+package com.amigoscode.product;
+
+import java.math.BigDecimal;
+import java.time.Instant;
+import java.util.UUID;
+
+public record ProductResponse(
+        UUID id,
+        String name,
+        String description,
+        BigDecimal price,
+        String imageUrl,
+        Integer stockLevel,
+        Instant createdAt,
+        Instant updatedAt,
+        Instant deletedAt
+) {
+}

src/main/java/com/amigoscode/product/ProductService.java

@@ -5,6 +5,8 @@
 
 import java.util.List;
 import java.util.UUID;
+import java.util.function.Function;
+import java.util.stream.Collectors;
 
 @Service
 public class ProductService {
@@ -14,12 +16,15 @@ public ProductService(ProductRepository productRepository) {
         this.productRepository = productRepository;
     }
 
-    public List<Product> getAllProducts() {
-        return productRepository.findAll();
+    public List<ProductResponse> getAllProducts() {
+        return productRepository.findAll().stream()
+                .map(mapToResponse())
+                .collect(Collectors.toList());
     }
 
-    public Product getProductById(UUID id) {
+    public ProductResponse getProductById(UUID id) {
         return productRepository.findById(id)
+                .map(mapToResponse())
                 .orElseThrow(() -> new ResourceNotFound(
                         "product with id [" + id + "] not found"
                 ));
@@ -48,4 +53,19 @@ public UUID saveNewProduct(NewProductRequest product) {
         productRepository.save(newProduct);
         return id;
     }
+
+    private Function<Product, ProductResponse> mapToResponse() {
+        return p -> new ProductResponse(
+                p.getId(),
+                p.getName(),
+                p.getDescription(),
+                p.getPrice(),
+                p.getImageUrl(),
+                p.getStockLevel(),
+                p.getCreatedAt(),
+                p.getUpdatedAt(),
+                p.getDeletedAt()
+        );
+    }
+
 }