Servlet中的文件上传与下载功能实现

引言

在Web开发中，文件上传和下载是非常常见的功能需求。在Java的Web开发中，Servlet是实现这些功能的一种较为常用的方式。本文将介绍如何使用Servlet实现文件的上传和下载功能。

文件上传实现步骤

1. 创建一个HTML表单，通过该表单实现文件的选择和提交。在这个表单中，必须指定form的enctype属性为multipart/form-data，这样才能支持文件上传。
2. 创建一个Servlet来处理文件上传的请求。在这个Servlet中，重写doPost方法来处理表单的提交操作。获取到上传的文件并存储到指定的位置。
3. 在Servlet中实现文件存储的逻辑。可以使用Java IO或者Apache Commons FileUpload等工具来实现文件的存储。

示例代码

@WebServlet("/upload")
@MultipartConfig
public class FileUploadServlet extends HttpServlet {

    protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        Part filePart = request.getPart("file");
        String fileName = getFileName(filePart);
        String uploadPath = getServletContext().getRealPath("") + File.separator + "uploads" + File.separator + fileName;

        try (InputStream inputStream = filePart.getInputStream();
             OutputStream outputStream = new FileOutputStream(uploadPath)) {
            byte[] buffer = new byte[1024];
            int bytesRead;

            while ((bytesRead = inputStream.read(buffer)) != -1) {
                outputStream.write(buffer, 0, bytesRead);
            }
        }

        response.getWriter().println("File uploaded successfully!");
    }

    private String getFileName(Part part) {
        String contentDisposition = part.getHeader("content-disposition");
        String[] elements = contentDisposition.split(";");

        for (String element : elements) {
            if (element.trim().startsWith("filename")) {
                return element.substring(element.indexOf('=') + 1).trim().replace("\"", "");
            }
        }
        return "";
    }
}

文件下载实现步骤

1. 创建一个Servlet，用于处理文件下载请求。在这个Servlet中，重写doGet方法来处理下载文件的请求。
2. 根据请求中的参数，确定要下载的文件路径和文件名。
3. 创建一个输入流，读取文件并将其写入到response对象的输出流中，将文件发送给客户端。

示例代码

@WebServlet("/download")
public class FileDownloadServlet extends HttpServlet {

    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        String fileName = request.getParameter("fileName");
        String filePath = getServletContext().getRealPath("") + File.separator + "uploads" + File.separator + fileName;

        File file = new File(filePath);
        FileInputStream fis = new FileInputStream(file);

        response.setContentType("application/octet-stream");
        response.setHeader("Content-Disposition", "attachment; filename=\"" + fileName + "\"");

        try (OutputStream outputStream = response.getOutputStream()) {
            byte[] buffer = new byte[1024];
            int bytesRead;

            while ((bytesRead = fis.read(buffer)) != -1) {
                outputStream.write(buffer, 0, bytesRead);
            }
        }

        fis.close();
    }
}

总结

使用Servlet来实现文件上传和下载功能并不复杂。上传文件时需要注意设置表单的enctype属性为multipart/form-data，下载文件时需要将要下载的文件写入response对象的输出流中。通过上述步骤和示例代码，可以很容易地实现文件上传和下载功能。

以上就是Servlet中文件上传和下载的实现方法，希望对你有所帮助！

注意：本文归作者所有，未经作者允许，不得转载