java 上传图片限制了1M

原创

mob64ca12d06991 2024-08-01 14:01:26 ©著作权

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

Java 上传图片限制为 1MB 的实现

在现代的 Web 应用程序中，文件上传是非常常见的功能。但是，上传大文件可能会导致服务器性能下降，甚至出现内存溢出的问题。为了避免这些问题，通常会对上传的文件大小进行限制。本文将介绍如何在 Java 中实现对图片上传大小的限制，具体限制为 1MB。

需求分析

首先，我们需要明确需求：用户在上传图片时，文件大小不能超过 1MB。如果用户上传的文件超过该限制，系统会给出友好的提示。

运行环境

Java 8 及以上版本
Servlet API
Maven（可选）

代码示例

首先，我们需要一个 HTML 表单，让用户可以上传图片。以下是一个简单的 HTML 文件：

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <meta name="viewport" content="width=device-width, initial-scale=1.0">
    <title>上传图片</title>
</head>
<body>
    上传图片
    <form action="upload" method="post" enctype="multipart/form-data">
        <input type="file" name="image" accept="image/*" required>
        <button type="submit">上传</button>
    </form>
</body>
</html>

接下来，我们在 Java 后端实现文件上传功能，并加入1MB的限制。代码如下：

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;
import java.io.IOException;

@WebServlet("/upload")
@MultipartConfig(fileSizeThreshold = 1024 * 1024, // 1MB
                 maxFileSize = 1024 * 1024, // 1MB
                 maxRequestSize = 1024 * 1024) // 1MB
public class FileUploadServlet extends HttpServlet {

    protected void doPost(HttpServletRequest request, HttpServletResponse response) 
            throws ServletException, IOException {
        Part filePart = request.getPart("image"); // 获取上传的文件

        if (filePart.getSize() > 0) {
            String fileName = getFileName(filePart);
            // 处理文件的保存逻辑
            // ......
            response.getWriter().write("上传成功: " + fileName);
        } else {
            response.getWriter().write("文件大小不能超过 1MB，请重试.");
        }
    }

    private String getFileName(Part part) {
        String contentDisp = part.getHeader("content-disposition");
        String[] items = contentDisp.split(";");
        for (String s : items) {
            if (s.trim().startsWith("filename")) {
                return s.substring(s.indexOf("=") + 2, s.length() - 1);
            }
        }
        return "";
    }
}

处理流程

我们可以用流程图来表示用户上传图片的流程：

flowchart TD
    A[用户选择文件] --> B{文件大小}
    B -->|小于或等于1MB| C[上传文件]
    B -->|大于1MB| D[给出提示]
    C --> E[成功上传]
    D --> E

代码分析

HTML 部分：表单以 POST 方式提交，并设置 enctype 为 multipart/form-data，这对于文件上传是必须的。
Servlet 部分:
- @MultipartConfig 注解用于定义文件上传的限制。
- 在 doPost 方法中，获取上传的文件并检查其大小。
- 如果文件大小小于或等于 1MB，则处理文件；否则，返回提示信息。