Product image and image gallery are implemented.

Author: manirDev

src/main/java/com/manir/springbootecommercerestapi/controller/ProductController.java

@@ -4,9 +4,12 @@
 import com.manir.springbootecommercerestapi.service.ProductService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.http.HttpStatus;
+import org.springframework.http.MediaType;
 import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
 
+import java.io.IOException;
 import java.util.List;
 
 @RestController
@@ -17,9 +20,10 @@ public class ProductController {
     private ProductService productService;
 
     //product create api
-    @PostMapping("/createProduct")
-    public ResponseEntity<ProductDto> createProduct(@RequestBody ProductDto productDto){
-        ProductDto responseProduct = productService.createProduct(productDto);
+    @RequestMapping(value = "/createProduct", method = RequestMethod.POST, consumes = {MediaType.APPLICATION_JSON_VALUE, MediaType.MULTIPART_FORM_DATA_VALUE})
+    public ResponseEntity<ProductDto> createProduct(@RequestPart("productDto") ProductDto productDto,
+                                                    @RequestPart("file") MultipartFile file){
+        ProductDto responseProduct = productService.createProduct(productDto, file);
         return   new ResponseEntity<>(responseProduct, HttpStatus.CREATED);
     }
 

src/main/java/com/manir/springbootecommercerestapi/controller/ProductImageGalleryController.java

@@ -0,0 +1,26 @@
+package com.manir.springbootecommercerestapi.controller;
+
+import com.manir.springbootecommercerestapi.resource.ImageData;
+import com.manir.springbootecommercerestapi.service.ProductImageGalleryService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
+
+@RestController
+@RequestMapping(value = "api/v1/products")
+public class ProductImageGalleryController {
+
+    @Autowired
+    private ProductImageGalleryService productImageGalleryService;
+
+
+    //add image to product api
+    @PostMapping("/{productId}/addImage")
+    public ResponseEntity<String> addImageToProduct(@PathVariable Long productId,
+                                                    @RequestParam("file")MultipartFile file){
+        productImageGalleryService.addImageToProduct(productId, file);
+        return new ResponseEntity<>("Image  successfully added to product with id : " + productId, HttpStatus.CREATED);
+    }
+}

src/main/java/com/manir/springbootecommercerestapi/dto/ImageDataDto.java

@@ -0,0 +1,13 @@
+package com.manir.springbootecommercerestapi.dto;
+
+import lombok.Data;
+
+@Data
+public class ImageDataDto {
+
+    private Long id;
+    private String name;
+    private String type;
+    private byte[] imageData;
+
+}

src/main/java/com/manir/springbootecommercerestapi/dto/ProductDto.java

@@ -1,6 +1,10 @@
 package com.manir.springbootecommercerestapi.dto;
+import com.manir.springbootecommercerestapi.resource.ImageData;
 import lombok.Data;
 
+import java.util.List;
+import java.util.Set;
+
 @Data
 public class ProductDto {
     private Long id;
@@ -12,5 +16,9 @@ public class ProductDto {
     private Integer quantity;
     private String status;
 
+    private String image;
+
     private CategoryDto category;
+
+    private Set<ImageDataDto> images;
 }

src/main/java/com/manir/springbootecommercerestapi/repository/ProductImageGalleryRepository.java

@@ -0,0 +1,7 @@
+package com.manir.springbootecommercerestapi.repository;
+
+import com.manir.springbootecommercerestapi.resource.ImageData;
+import org.springframework.data.jpa.repository.JpaRepository;
+
+public interface ProductImageGalleryRepository extends JpaRepository<ImageData, Long> {
+}

src/main/java/com/manir/springbootecommercerestapi/resource/ImageData.java

@@ -0,0 +1,31 @@
+package com.manir.springbootecommercerestapi.resource;
+
+import lombok.*;
+
+import javax.persistence.*;
+
+@Getter
+@Setter
+@AllArgsConstructor
+@NoArgsConstructor
+@Entity
+@Table(name = "product_image_gallery")
+public class ImageData {
+    @Id
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    private Long id;
+    @Column(name = "name")
+    private String name;
+    @Column(name = "type")
+    private String type;
+
+    @Column(name = "imageData", length = 1000)
+    @Lob
+    private byte[] imageData;
+
+    //relation with product entity
+    @ManyToOne(fetch = FetchType.EAGER)
+    @JoinColumn(name = "product_id")
+    private Product product;
+
+}

src/main/java/com/manir/springbootecommercerestapi/resource/Product.java

@@ -3,6 +3,8 @@
 import lombok.*;
 
 import javax.persistence.*;
+import java.util.List;
+import java.util.Set;
 
 @AllArgsConstructor
 @NoArgsConstructor
@@ -29,8 +31,18 @@ public class Product {
     @Column(name = "status")
     private String status;
 
+    //product image
+    @Lob
+    @Column(name = "image")
+    private String image;
+
+
     //relation between category and product
     @ManyToOne(fetch = FetchType.LAZY)
     @JoinColumn(name = "category_id")
-    Category category;
+    private Category category;
+
+    //relation image gallery
+    @OneToMany(mappedBy = "product", cascade = CascadeType.ALL, orphanRemoval = true)
+    private Set<ImageData> images;
 }

src/main/java/com/manir/springbootecommercerestapi/service/Impl/ProductImageGalleryServiceImpl.java

@@ -0,0 +1,54 @@
+package com.manir.springbootecommercerestapi.service.Impl;
+
+import com.manir.springbootecommercerestapi.exception.ResourceNotFoundException;
+import com.manir.springbootecommercerestapi.repository.ProductImageGalleryRepository;
+import com.manir.springbootecommercerestapi.repository.ProductRepository;
+import com.manir.springbootecommercerestapi.resource.ImageData;
+import com.manir.springbootecommercerestapi.resource.Product;
+import com.manir.springbootecommercerestapi.service.ProductImageGalleryService;
+import com.manir.springbootecommercerestapi.utils.ImageUtils;
+import org.springframework.stereotype.Service;
+import org.springframework.web.multipart.MultipartFile;
+
+import javax.annotation.Resource;
+import java.io.IOException;
+
+@Service
+public class ProductImageGalleryServiceImpl implements ProductImageGalleryService {
+
+    @Resource
+    private ProductImageGalleryRepository productImageGalleryRepository;
+    @Resource
+    private ProductRepository productRepository;
+
+    @Override
+    public ImageData addImageToProduct(Long productId, MultipartFile file) {
+
+        Product product = productRepository.findById(productId)
+                                            .orElseThrow(
+                                                    () -> new ResourceNotFoundException("Product", productId)
+                                            );
+
+        ImageData addedImage = null;
+        try {
+            addedImage = productImageGalleryRepository.save(uploadImage(file,product));
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+
+        return addedImage;
+    }
+
+
+    private ImageData uploadImage(MultipartFile file, Product product ) throws IOException {
+        ImageData imageData = new ImageData();
+        imageData.setName(file.getOriginalFilename());
+        imageData.setType(file.getContentType());
+        imageData.setImageData(ImageUtils.compressImage(file.getBytes()));
+
+        imageData.setProduct(product);
+        return imageData;
+    }
+
+
+}

src/main/java/com/manir/springbootecommercerestapi/service/Impl/ProductServiceImpl.java

@@ -5,12 +5,18 @@
 import com.manir.springbootecommercerestapi.repository.CategoryRepository;
 import com.manir.springbootecommercerestapi.repository.ProductRepository;
 import com.manir.springbootecommercerestapi.resource.Category;
+import com.manir.springbootecommercerestapi.resource.ImageData;
 import com.manir.springbootecommercerestapi.resource.Product;
 import com.manir.springbootecommercerestapi.service.ProductService;
+import com.manir.springbootecommercerestapi.utils.ImageUtils;
 import org.modelmapper.ModelMapper;
 import org.springframework.stereotype.Service;
+import org.springframework.util.StringUtils;
+import org.springframework.web.multipart.MultipartFile;
 
 import javax.annotation.Resource;
+import java.io.IOException;
+import java.util.Base64;
 import java.util.List;
 import java.util.stream.Collectors;
 
@@ -25,8 +31,9 @@ public class ProductServiceImpl implements ProductService {
     private CategoryRepository categoryRepository;
 
     @Override
-    public ProductDto createProduct(ProductDto productDto) {
+    public ProductDto createProduct(ProductDto productDto, MultipartFile file) {
         //map to entity
+        productDto.setImage(uploadProductImage(file));
         Product product = mapToEntity(productDto);
         Product createdProduct = productRepository.save(product);
         //map to dto
@@ -105,6 +112,22 @@ public ProductDto saveProductByCategoryId(Long categoryId, ProductDto productDto
         return responseProduct;
     }
 
+    //upload image
+    private String uploadProductImage(MultipartFile file){
+        ProductDto productDto = new ProductDto();
+        String fileName = StringUtils.cleanPath(file.getOriginalFilename());
+        if(fileName.contains("..")){
+            System.out.println("Not a valid file");
+        }
+        try {
+            productDto.setImage(Base64.getEncoder().encodeToString(file.getBytes()));
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+        return productDto.getImage();
+    }
+
+
     //map to dto
     private ProductDto mapToDto(Product product){
         ProductDto productDto = modelMapper.map(product, ProductDto.class);

src/main/java/com/manir/springbootecommercerestapi/service/ProductImageGalleryService.java

@@ -0,0 +1,10 @@
+package com.manir.springbootecommercerestapi.service;
+
+import com.manir.springbootecommercerestapi.resource.ImageData;
+import org.springframework.web.multipart.MultipartFile;
+
+public interface ProductImageGalleryService {
+
+     ImageData addImageToProduct(Long productId, MultipartFile file);
+
+}

src/main/java/com/manir/springbootecommercerestapi/service/ProductService.java

@@ -1,11 +1,12 @@
 package com.manir.springbootecommercerestapi.service;
 
 import com.manir.springbootecommercerestapi.dto.ProductDto;
+import org.springframework.web.multipart.MultipartFile;
 
 import java.util.List;
 
 public interface ProductService {
-    ProductDto createProduct(ProductDto productDto);
+    ProductDto createProduct(ProductDto productDto, MultipartFile file);
     List<ProductDto> getAllProduct();
     ProductDto getProductById(Long productId);
     ProductDto updateProduct(Long categoryId, ProductDto productDto, Long productId);

src/main/java/com/manir/springbootecommercerestapi/utils/ImageUtils.java

@@ -0,0 +1,47 @@
+package com.manir.springbootecommercerestapi.utils;
+
+import java.io.ByteArrayOutputStream;
+import java.util.zip.Deflater;
+import java.util.zip.Inflater;
+
+public class ImageUtils {
+
+    public static byte[] compressImage(byte[] data) {
+        Deflater deflater = new Deflater();
+        deflater.setLevel(Deflater.BEST_COMPRESSION);
+        deflater.setInput(data);
+        deflater.finish();
+
+        ByteArrayOutputStream outputStream = new ByteArrayOutputStream(data.length);
+        byte[] tmp = new byte[4*1024];
+        while (!deflater.finished()) {
+            int size = deflater.deflate(tmp);
+            outputStream.write(tmp, 0, size);
+        }
+        try {
+            outputStream.close();
+        } catch (Exception ignored) {
+        }
+        return outputStream.toByteArray();
+    }
+
+
+
+    public static byte[] decompressImage(byte[] data) {
+        Inflater inflater = new Inflater();
+        inflater.setInput(data);
+        ByteArrayOutputStream outputStream = new ByteArrayOutputStream(data.length);
+        byte[] tmp = new byte[4*1024];
+        try {
+            while (!inflater.finished()) {
+                int count = inflater.inflate(tmp);
+                outputStream.write(tmp, 0, count);
+            }
+            outputStream.close();
+        } catch (Exception ignored) {
+        }
+        return outputStream.toByteArray();
+    }
+
+
+}