| | |
| | | } |
| | | |
| | | 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; |
| | | |
| | | |
| | | |