| | |
| | | } |
| | | |
| | | private ResponseEntity<org.springframework.core.io.Resource> downloadFile(UserFile userFile) throws IOException { |
| | | InputStreamResource resource =resolveResource(userFile); |
| | | InputStreamResource resource =userFileService.resolveResource(userFile); |
| | | var fileSize = userFile.getSize(); |
| | | // if(resource == null) { |
| | | // resource = new InputStreamResource(getClass().getClassLoader().getResourceAsStream("images/wukong.png")); |
| | |
| | | .body(resource); |
| | | } |
| | | |
| | | public InputStreamResource resolveResource(UserFile userFile) throws IOException { |
| | | var filePath = fileLocation.resolve(userFile.getName() + "." + userFile.getExtension()).normalize(); |
| | | if(!Files.exists(filePath)){ |
| | | return null; |
| | | } |
| | | return new InputStreamResource(Files.newInputStream(filePath)); |
| | | } |
| | | |
| | | private Path fileLocation; |
| | | |
| | | |
| | | |
| | |
| | | private String UPLOAD_DIR; |
| | | |
| | | |
| | | @PostConstruct |
| | | void init(){ |
| | | fileLocation = Path.of(UPLOAD_DIR).toAbsolutePath().normalize(); |
| | | try { |
| | | Files.createDirectories(this.fileLocation); |
| | | } catch (Exception ex) { |
| | | throw new RuntimeException("Could not create the directory where the uploaded files will be stored.", ex); |
| | | } |
| | | } |
| | | // @PostConstruct |
| | | // void init(){ |
| | | // fileLocation = Path.of(UPLOAD_DIR).toAbsolutePath().normalize(); |
| | | // try { |
| | | // Files.createDirectories(this.fileLocation); |
| | | // } catch (Exception ex) { |
| | | // throw new RuntimeException("Could not create the directory where the uploaded files will be stored.", ex); |
| | | // } |
| | | // } |
| | | |
| | | } |