update with database

Author: jeegarlearner

pom.xml

@@ -41,6 +41,22 @@
             <version>4.4.1-1</version>
         </dependency>
 
+        <!-- https://mvnrepository.com/artifact/javax.servlet/jstl -->
+        <dependency>
+            <groupId>javax.servlet</groupId>
+            <artifactId>jstl</artifactId>
+            <version>1.2</version>
+        </dependency>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-data-jpa</artifactId>
+        </dependency>
+        <!-- https://mvnrepository.com/artifact/mysql/mysql-connector-java -->
+        <dependency>
+            <groupId>mysql</groupId>
+            <artifactId>mysql-connector-java</artifactId>
+            <version>8.0.19</version>
+        </dependency>
         <dependency>
             <groupId>org.springframework.boot</groupId>
             <artifactId>spring-boot-starter-test</artifactId>

src/main/java/com/springfileupload/app/controller/FileStoreController.java

@@ -0,0 +1,67 @@
+package com.springfileupload.app.controller;
+
+import com.springfileupload.app.model.FileStore;
+import com.springfileupload.app.service.FileStoreService;
+import org.springframework.core.io.ByteArrayResource;
+import org.springframework.core.io.Resource;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.MediaType;
+import org.springframework.http.ResponseEntity;
+import org.springframework.stereotype.Controller;
+import org.springframework.ui.ModelMap;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestMethod;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.multipart.MultipartFile;
+
+@Controller
+public class FileStoreController {
+
+    private final FileStoreService fileStoreService;
+
+    public FileStoreController(FileStoreService fileStoreService) {
+        this.fileStoreService = fileStoreService;
+    }
+
+    @RequestMapping("/")
+    public String index(ModelMap modelMap) {
+        modelMap.addAttribute("fileList", fileStoreService.getAllFiles());
+        return "index";
+    }
+
+    @RequestMapping(value = "/uploadFile", method = RequestMethod.POST, consumes = MediaType.MULTIPART_FORM_DATA_VALUE)
+    public String upload(@RequestParam("file") MultipartFile file, ModelMap modelMap) {
+
+        if (fileStoreService.saveFileToDB(file) != null && fileStoreService.saveFileToLocal(file)) {
+            modelMap.addAttribute("msg", "file is uploaded Successfully");
+        } else {
+            modelMap.addAttribute("msg", "sorry!file is not uploaded");
+        }
+
+        modelMap.addAttribute("fileList", fileStoreService.getAllFiles());
+        return "index";
+    }
+
+
+    @RequestMapping("/download")
+    public ResponseEntity<Resource> downloadFile(@RequestParam("id") int id) {
+        FileStore fileStore = fileStoreService.getFileStoreById(id);
+        return ResponseEntity.ok()
+                .header(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=\"" + fileStore.getFileName() + "\"")
+                .body(new ByteArrayResource(fileStore.getFile()));
+    }
+
+    @RequestMapping("/delete")
+    public String deleteFile(@RequestParam("id") int id, ModelMap map) {
+        try {
+            fileStoreService.deleteFileFromLocal(fileStoreService.getFileStoreById(id));
+            fileStoreService.deleteFileFromDB(id);
+            map.addAttribute("msg", "file is deleted");
+            map.addAttribute("fileList", fileStoreService.getAllFiles());
+        } catch (Exception exception) {
+            map.addAttribute("$msg", "file is not deleted");
+        }
+        return "index";
+    }
+
+}

src/main/java/com/springfileupload/app/controller/FileUploadController.java

@@ -0,0 +1,84 @@
+package com.springfileupload.app.model;
+
+import javax.persistence.*;
+import java.sql.Timestamp;
+import java.util.Arrays;
+import java.util.Objects;
+
+@Entity
+@Table(name = "file_store", schema = "files")
+public class FileStore {
+    private int id;
+    private String fileName;
+    private byte[] file;
+    private Timestamp updatedDate;
+
+    @Id
+    @Column(name = "id", nullable = false)
+    public int getId() {
+        return id;
+    }
+
+    public void setId(int id) {
+        this.id = id;
+    }
+
+    @Basic
+    @Column(name = "file_name", nullable = false, length = 30)
+    public String getFileName() {
+        return fileName;
+    }
+
+    public void setFileName(String fileName) {
+        this.fileName = fileName;
+    }
+
+
+    @Basic
+    @Column(name = "file", nullable = false)
+    public byte[] getFile() {
+        return file;
+    }
+
+    public void setFile(byte[] file) {
+        this.file = file;
+    }
+
+    @Basic
+    @Column(name = "updated_date", nullable = false)
+    public Timestamp getUpdatedDate() {
+        return updatedDate;
+    }
+
+    public void setUpdatedDate(Timestamp updatedDate) {
+        this.updatedDate = updatedDate;
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+        FileStore fileStore = (FileStore) o;
+        return id == fileStore.id &&
+                Objects.equals(fileName, fileStore.fileName) &&
+                Arrays.equals(file, fileStore.file) &&
+                Objects.equals(updatedDate, fileStore.updatedDate);
+    }
+
+    @Override
+    public int hashCode() {
+        int result = Objects.hash(id, fileName, updatedDate);
+        result = 31 * result + Arrays.hashCode(file);
+        return result;
+    }
+
+    @Override
+    public String toString() {
+        return "FileStore{" +
+                "id=" + id +
+                ", fileName='" + fileName + '\'' +
+                ", file=" + Arrays.toString(file) +
+                ", updatedDate=" + updatedDate +
+                '}';
+    }
+}

src/main/java/com/springfileupload/app/model/FileStore.java

@@ -0,0 +1,7 @@
+package com.springfileupload.app.repo;
+
+import com.springfileupload.app.model.FileStore;
+import org.springframework.data.jpa.repository.JpaRepository;
+
+public interface FileStoreRepository extends JpaRepository<FileStore ,Integer> {
+}

src/main/java/com/springfileupload/app/repo/FileStoreRepository.java

@@ -0,0 +1,20 @@
+package com.springfileupload.app.service;
+
+import com.springfileupload.app.model.FileStore;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.util.List;
+
+public interface FileStoreService {
+    boolean saveFileToLocal(MultipartFile multipartFile);
+
+    FileStore saveFileToDB(MultipartFile multipartFile);
+
+    FileStore getFileStoreById(int fileId);
+
+    List<FileStore> getAllFiles();
+
+    void deleteFileFromDB(int fileId);
+
+    void deleteFileFromLocal(FileStore fileStore);
+}

src/main/java/com/springfileupload/app/service/FileStoreService.java

@@ -0,0 +1,101 @@
+package com.springfileupload.app.service;
+
+import com.springfileupload.app.model.FileStore;
+import com.springfileupload.app.repo.FileStoreRepository;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Service;
+import org.springframework.util.StringUtils;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.util.List;
+import java.util.Objects;
+
+import static java.nio.file.StandardCopyOption.REPLACE_EXISTING;
+
+@Service
+public class FileStoreServiceImpl implements FileStoreService {
+
+    @Value(value = "${com.fileupload.upload.folder.path}")
+    private String folderPath;
+
+    private final FileStoreRepository fileStoreRepository;
+
+    public FileStoreServiceImpl(FileStoreRepository fileStoreRepository) {
+        this.fileStoreRepository = fileStoreRepository;
+    }
+
+    @Override
+    public boolean saveFileToLocal(MultipartFile file) {
+        String fileName = getFileName(file);
+
+        if (isValidFile(fileName, file)) {
+            try {
+                InputStream inputStream = file.getInputStream();
+                Files.copy(inputStream, Paths.get(folderPath + fileName), REPLACE_EXISTING);
+                return true;
+            } catch (IOException e) {
+                e.printStackTrace();
+            }
+        }
+        return false;
+    }
+
+    @Override
+    public FileStore saveFileToDB(MultipartFile file) {
+        FileStore fileStore = new FileStore();
+        String fileName = getFileName(file);
+
+        if (isValidFile(fileName, file)) {
+            try {
+                fileStore.setFileName(fileName);
+                fileStore.setFile(file.getBytes());
+            } catch (IOException e) {
+                e.printStackTrace();
+            }
+        }
+        return fileStoreRepository.save(fileStore);
+    }
+
+    @Override
+    public FileStore getFileStoreById(int id) {
+        return fileStoreRepository.findById(id).get();
+
+    }
+
+    @Override
+    public List<FileStore> getAllFiles() {
+        return fileStoreRepository.findAll();
+    }
+
+    @Override
+    public void deleteFileFromDB(int fileId) {
+        fileStoreRepository.deleteById(fileId);
+    }
+
+    @Override
+    public void deleteFileFromLocal(FileStore fileStore) {
+        String fileName = fileStore.getFileName();
+
+        try {
+            Files.deleteIfExists(Path.of(folderPath+fileName));
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+
+    }
+
+    public String getFileName(MultipartFile file) {
+        return StringUtils.cleanPath(Objects.requireNonNull(file.getOriginalFilename()));
+    }
+
+    public boolean isValidFile(String fileName, MultipartFile file) {
+        return !file.isEmpty() && !fileName.contains("..");
+    }
+
+
+}