Java附件下载

原创

mob64ca12ef217e 2023-08-13 05:11:01 ©著作权

文章标签 HTTP 字节流 Java 文章分类 Java 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12ef217e的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java附件下载

介绍

在开发Web应用程序时，经常会遇到需要提供文件下载的场景，例如下载用户上传的附件、导出数据为Excel文件等。Java提供了多种方法来实现附件下载，本文将介绍两种常用的实现方式。

1. 使用Spring框架实现附件下载

Spring框架提供了一个Resource接口，可以方便地处理文件资源。我们可以使用Resource接口来返回文件的字节流，从而实现附件下载。

下面是一个示例代码，展示了如何使用Spring框架实现附件下载：

@RestController
public class FileDownloadController {
    
    @GetMapping("/download")
    public ResponseEntity<Resource> downloadFile() {
        // 从文件系统或数据库中获取文件
        File file = new File("/path/to/file.pdf");
        
        // 创建一个Resource对象，封装文件字节流
        Resource resource = new FileSystemResource(file);
        
        // 设置HTTP头信息
        HttpHeaders headers = new HttpHeaders();
        headers.add(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=" + file.getName());
        
        // 返回ResponseEntity对象，包含文件的字节流和HTTP头信息
        return ResponseEntity.ok()
                .headers(headers)
                .contentLength(file.length())
                .contentType(MediaType.APPLICATION_OCTET_STREAM)
                .body(resource);
    }
}

在上面的代码中，通过@GetMapping注解将/download路径映射到downloadFile()方法。该方法返回一个ResponseEntity<Resource>对象，其中Resource对象封装了文件的字节流。通过设置HTTP头信息，指定了文件的下载方式以及文件名。

2. 使用Servlet实现附件下载

除了使用Spring框架，我们还可以使用Servlet来实现附件下载。Java Servlet规范中提供了一个HttpServletResponse类，可以用于向客户端发送HTTP响应。

下面是一个示例代码，展示了如何使用Servlet实现附件下载：

@WebServlet("/download")
public class FileDownloadServlet extends HttpServlet {
    
    @Override
    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        // 从文件系统或数据库中获取文件
        File file = new File("/path/to/file.pdf");
        
        // 设置HTTP头信息
        response.setHeader("Content-Disposition", "attachment; filename=" + file.getName());
        response.setHeader("Content-Length", String.valueOf(file.length()));
        
        // 将文件的字节流写入响应输出流
        try (InputStream is = new FileInputStream(file);
             OutputStream os = response.getOutputStream()) {
            byte[] buffer = new byte[4096];
            int bytesRead;
            while ((bytesRead = is.read(buffer)) != -1) {
                os.write(buffer, 0, bytesRead);
            }
        }
    }
}

在上面的代码中，通过@WebServlet注解将/download路径映射到doGet()方法。在该方法中，首先设置了HTTP头信息，然后将文件的字节流写入响应输出流，实现文件的下载。