一个文件下载的Servlet

最新推荐文章于 2022-02-04 16:05:38 发布

kingsui69

最新推荐文章于 2022-02-04 16:05:38 发布

阅读量147

点赞数

CC 4.0 BY-SA版权

分类专栏： Java 文章标签： Servlet Web XML

本文链接：https://blog.youkuaiyun.com/kingsui69/article/details/83323941

Java 专栏收录该内容

13 篇文章

订阅专栏

本文介绍了一个安全的文件下载Servlet实现，该实现避免了直接暴露文件目录给用户的风险，并提供了灵活的文件来源配置，如文件系统、数据库等。通过设置contentType、编码及文件根路径等参数，实现了对下载文件类型的控制。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

把文件目录直接暴露给用户是很不安全的。所以要用Servlet来做，而且这样做，文件的存储方式就更丰富了，可以是从文件系统上取来的，也可以是数据库中经过计算生成的，或者从其它什么稀奇古怪的地方取来的。

public class DownloadServlet extends HttpServlet {
    private String contentType = "application/x-msdownload";
    private String enc = "utf-8";
    private String fileRoot = "";

    /**
     * 初始化contentType，enc，fileRoot
     */
    public void init(ServletConfig config) throws ServletException {
        String tempStr = config.getInitParameter("contentType");
        if (tempStr != null && !tempStr.equals("")) {
            contentType = tempStr;
        }
        tempStr = config.getInitParameter("enc");
        if (tempStr != null && !tempStr.equals("")) {
            enc = tempStr;
        }
        tempStr = config.getInitParameter("fileRoot");
        if (tempStr != null && !tempStr.equals("")) {
            fileRoot = tempStr;
        }
    }

    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        String filepath = request.getParameter("filepath");
        String fullFilePath = fileRoot + filepath;
        /*读取文件*/
        File file = new File(fullFilePath);
        /*如果文件存在*/
        if (file.exists()) {
            String filename = URLEncoder.encode(file.getName(), enc);
            response.reset();
            response.setContentType(contentType);
            response.addHeader("Content-Disposition", "attachment; filename=\"" + filename + "\"");
            int fileLength = (int) file.length();
            response.setContentLength(fileLength);
            /*如果文件长度大于0*/
            if (fileLength != 0) {
                /*创建输入流*/
                InputStream inStream = new FileInputStream(file);
                byte[] buf = new byte[4096];
                /*创建输出流*/
                ServletOutputStream servletOS = response.getOutputStream();
                int readLength;
                while (((readLength = inStream.read(buf)) != -1)) {
                    servletOS.write(buf, 0, readLength);
                }
                inStream.close();
                servletOS.flush();
                servletOS.close();
            }
        }
    }

web.xml

    <servlet>
        <servlet-name>downloadservlet-name>
        <servlet-class>org.mstar.servlet.DownloadServletservlet-class>
        <init-param>
            <param-name>fileRootparam-name>
            <param-value>d:/tempparam-value>
        init-param>
        <init-param>
            <param-name>contentTypeparam-name>
            <param-value>application/x-msdownloadparam-value>
        init-param>
        <init-param>
            <param-name>encparam-name>
            <param-value>utf-8param-value>
        init-param>
    servlet>
    <servlet-mapping>
        <servlet-name>downloadservlet-name>
        <url-pattern>/downurl-pattern>
    servlet-mapping>