java 限制上传文件大小

原创

mob649e8158a948 2023-10-23 15:20:29 ©著作权

文章标签 java 文件上传文件大小 文章分类 Java 后端开发

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

Java限制上传文件大小

在Web开发中，文件上传是一项常见的功能。然而，为了保护服务器和网络资源，我们通常需要对上传的文件进行大小限制。本文将介绍如何使用Java限制上传文件大小，并提供代码示例。

文件上传的基本原理

在Web开发中，文件上传是通过HTTP协议实现的。当用户选择文件并提交表单时，浏览器会将文件数据编码成多部分表单数据（multipart/form-data）并发送到服务器。

服务器端的处理程序接收到这些数据后，会解析多部分表单数据，并提取出文件的相关信息和内容。在这个过程中，我们可以对文件的大小进行限制。

Java实现文件上传限制

Java在Servlet规范中提供了@MultipartConfig注解，用于配置文件上传的相关参数。其中，maxFileSize和maxRequestSize属性用于限制文件的大小。

maxFileSize属性用于限制单个文件的大小，单位为字节。默认值为-1，表示不限制文件的大小。

maxRequestSize属性用于限制整个请求的大小，单位为字节。默认值为-1，表示不限制请求的大小。

下面是一个使用@MultipartConfig注解限制文件上传大小的示例代码：

import java.io.IOException;
import java.io.PrintWriter;

import javax.servlet.ServletException;
import javax.servlet.annotation.MultipartConfig;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.Part;

@WebServlet("/upload")
@MultipartConfig(maxFileSize = 1024 * 1024, maxRequestSize = 1024 * 1024 * 5)
public class UploadServlet extends HttpServlet {
    private static final long serialVersionUID = 1L;

    protected void doPost(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        response.setContentType("text/html;charset=UTF-8");

        PrintWriter out = response.getWriter();
        for (Part part : request.getParts()) {
            String fileName = getFileName(part);
            if (fileName != null && !fileName.isEmpty()) {
                part.write(fileName);
                out.println(fileName + " 上传成功！");
            }
        }
    }

    private String getFileName(final Part part) {
        final String partHeader = part.getHeader("content-disposition");
        for (String content : partHeader.split(";")) {
            if (content.trim().startsWith("filename")) {
                return content.substring(content.indexOf('=') + 1).trim().replace("\"", "");
            }
        }
        return null;
    }
}

在上述示例中，@MultipartConfig注解用于配置文件上传的参数。maxFileSize属性设置为1MB，maxRequestSize属性设置为5MB，表示限制单个文件的大小不超过1MB，整个请求的大小不超过5MB。

文件上传限制的前端验证

在服务器端对文件大小进行限制是非常重要的，但为了提升用户体验，我们也可以在前端进行一些文件大小的验证。

以下是一个使用HTML和JavaScript实现文件上传限制的示例代码：

<!DOCTYPE html>
<html>
<head>
    <meta charset="UTF-8">
    <title>文件上传</title>
    <script>
        function validateFileSize(fileInput) {
            var maxSize = 1024 * 1024; // 1MB
            if (fileInput.files[0].size > maxSize) {
                alert("文件大小不能超过1MB！");
                fileInput.value = "";
            }
        }
    </script>
</head>
<body>
    <form action="/upload" method="post" enctype="multipart/form-data">
        <input type="file" name="file" onchange="validateFileSize(this)">
        <input type="submit" value="上传">
    </form>
</body>
</html>

在上述示例中，validateFileSize函数用于验证文件大小是否超过1MB。如果文件大小超过1MB，会弹出一个提示框并清空文件输入框的值。