SpringBoot文件上传和下载的实现源码

compile("org.springframework.boot:spring-boot-starter-web") 
compile("org.springframework.boot:spring-boot-starter-thymeleaf") 

package hello; 
import org.springframework.boot.SpringApplication; 
import org.springframework.boot.autoconfigure.SpringBootApplication; 
import org.springframework.boot.context.properties.EnableConfigurationProperties; 
@SpringBootApplication 
public class Application { 
  public static void main(String[] args) { 
    SpringApplication.run(Application.class, args); 
  } 
} 

 
 
  
 
    
 
  
 
  
 
    
 
      
 
        
File to upload:
 
        
 
      
 
    
 
  
 
  
 
    
 
      
 
         
      
 
    
 
  
 
 
 

package hello.storage; 
import org.springframework.core.io.Resource; 
import org.springframework.web.multipart.MultipartFile; 
import java.nio.file.Path; 
import java.util.stream.Stream; 
public interface StorageService { 
  void init(); 
  void store(MultipartFile file); 
  Stream loadAll(); 
  Path load(String filename); 
  Resource loadAsResource(String filename); 
  void deleteAll(); 
} 

package hello.storage; 
import org.springframework.core.io.Resource; 
import org.springframework.core.io.UrlResource; 
import org.springframework.stereotype.Service; 
import org.springframework.util.FileSystemUtils; 
import org.springframework.util.StringUtils; 
import org.springframework.web.multipart.MultipartFile; 
import java.io.IOException; 
import java.net.MalformedURLException; 
import java.nio.file.Files; 
import java.nio.file.Path; 
import java.nio.file.Paths; 
import java.nio.file.StandardCopyOption; 
import java.util.function.Predicate; 
import java.util.stream.Stream; 
@Service 
public class FileSystemStorageService implements StorageService 
{ 
  private final Path rootLocation = Paths.get("upload-dir"); 
  /** 
   * 保存文件 
   * 
   * @param file 文件 
   */ 
  @Override 
  public void store(MultipartFile file) 
  { 
    String filename = StringUtils.cleanPath(file.getOriginalFilename()); 
    try 
    { 
      if (file.isEmpty()) 
      { 
        throw new StorageException("Failed to store empty file " + filename); 
      } 
      if (filename.contains("..")) 
      { 
        // This is a security check 
        throw new StorageException("Cannot store file with relative path outside current directory " + filename); 
      } 
      Files.copy(file.getInputStream(), this.rootLocation.resolve(filename), StandardCopyOption.REPLACE_EXISTING); 
    } 
    catch (IOException e) 
    { 
      throw new StorageException("Failed to store file " + filename, e); 
    } 
  } 
  /** 
   * 列出upload-dir下面所有文件 
   * @return 
   */ 
  @Override 
  public Stream loadAll() 
  { 
    try 
    { 
      return Files.walk(this.rootLocation, 1) //path -> !path.equals(this.rootLocation) 
          .filter(new Predicate() 
          { 
            @Override 
            public boolean test(Path path) 
            { 
              return !path.equals(rootLocation); 
            } 
          }); 
    } 
    catch (IOException e) 
    { 
      throw new StorageException("Failed to read stored files", e); 
    } 
  } 
  @Override 
  public Path load(String filename) 
  { 
    return rootLocation.resolve(filename); 
  } 
  /** 
   * 获取文件资源 
   * @param filename 文件名 
   * @return Resource 
   */ 
  @Override 
  public Resource loadAsResource(String filename) 
  { 
    try 
    { 
      Path file = load(filename); 
      Resource resource = new UrlResource(file.toUri()); 
      if (resource.exists() || resource.isReadable()) 
      { 
        return resource; 
      } 
      else 
      { 
        throw new StorageFileNotFoundException("Could not read file: " + filename); 
      } 
    } 
    catch (MalformedURLException e) 
    { 
      throw new StorageFileNotFoundException("Could not read file: " + filename, e); 
    } 
  } 
  /** 
   * 删除upload-dir目录所有文件 
   */ 
  @Override 
  public void deleteAll() 
  { 
    FileSystemUtils.deleteRecursively(rootLocation.toFile()); 
  } 
  /** 
   * 初始化 
   */ 
  @Override 
  public void init() 
  { 
    try 
    { 
      Files.createDirectories(rootLocation); 
    } 
    catch (IOException e) 
    { 
      throw new StorageException("Could not initialize storage", e); 
    } 
  } 
} 

package hello.storage; 
public class StorageException extends RuntimeException { 
  public StorageException(String message) { 
    super(message); 
  } 
  public StorageException(String message, Throwable cause) { 
    super(message, cause); 
  } 
} 
StorageFileNotFoundException.java
package hello.storage; 
public class StorageFileNotFoundException extends StorageException { 
  public StorageFileNotFoundException(String message) { 
    super(message); 
  } 
  public StorageFileNotFoundException(String message, Throwable cause) { 
    super(message, cause); 
  } 
} 

package hello;  


import java.io.IOException; 
import java.util.stream.Collectors; 
import org.springframework.beans.factory.annotation.Autowired; 
import org.springframework.core.io.Resource; 
import org.springframework.http.HttpHeaders; 
import org.springframework.http.ResponseEntity; 
import org.springframework.stereotype.Controller; 
import org.springframework.ui.Model; 
import org.springframework.web.bind.annotation.ExceptionHandler; 
import org.springframework.web.bind.annotation.GetMapping; 
import org.springframework.web.bind.annotation.PathVariable; 
import org.springframework.web.bind.annotation.PostMapping; 
import org.springframework.web.bind.annotation.RequestParam; 
import org.springframework.web.bind.annotation.ResponseBody; 
import org.springframework.web.multipart.MultipartFile; 
import org.springframework.web.servlet.mvc.method.annotation.MvcUriComponentsBuilder; 
import org.springframework.web.servlet.mvc.support.RedirectAttributes; 
import hello.storage.StorageFileNotFoundException; 
import hello.storage.StorageService; 
@Controller 
public class FileUploadController { 
  private final StorageService storageService; 
  @Autowired 
  public FileUploadController(StorageService storageService) { 
    this.storageService = storageService; 
  } 
  @GetMapping("/") 
  public String listUploadedFiles(Model model) throws IOException { 
    model.addAttribute("files", storageService.loadAll().map( 
        path -> MvcUriComponentsBuilder.fromMethodName(FileUploadController.class, 
            "serveFile", path.getFileName().toString()).build().toString()) 
        .collect(Collectors.toList())); 
    return "uploadForm"; 
  } 
  @GetMapping("/files/{filename:.+}") 
  @ResponseBody 
  public ResponseEntity serveFile(@PathVariable String filename) { 
    Resource file = storageService.loadAsResource(filename); 
    return ResponseEntity.ok().header(HttpHeaders.CONTENT_DISPOSITION, 
        "attachment; filename=\"" + file.getFilename() + "\"").body(file); 
  } 
  @PostMapping("/") 
  public String handleFileUpload(@RequestParam("file") MultipartFile file, 
      RedirectAttributes redirectAttributes) { 
    storageService.store(file); 
    redirectAttributes.addFlashAttribute("message", 
        "You successfully uploaded " + file.getOriginalFilename() + "!"); 
    return "redirect:/"; 
  } 
  @ExceptionHandler(StorageFileNotFoundException.class) 
  public ResponseEntity<?> handleStorageFileNotFound(StorageFileNotFoundException exc) { 
    return ResponseEntity.notFound().build(); 
  } 
}