java 限制上传的文件类型

原创

mob64ca12d9b014 2024-08-06 11:02:33 ©著作权

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

Java 限制上传文件类型的实现

对于刚入行的小白来说，学习如何在Java中限制文件上传类型是一个重要的技能。在这个过程中，我们将逐步探索文件上传的工作流程，并确保上传的文件符合特定类型的限制。本文将详细介绍实现这一目标的步骤、对应的代码示例以及重要的注释，以便于理解。

流程步骤

在开始编码之前，我们首先需要理解整个流程。以下是我们实现文件类型限制的步骤：

步骤	描述
1	创建前端上传文件的HTML表单
2	编写Servlet处理文件上传请求
3	检查文件类型并添加限制
4	返回响应给用户

每一步的详细内容

步骤 1：创建前端上传文件的HTML表单

首先，我们需要构建一个简单的HTML表单，允许用户上传文件。

<!DOCTYPE html>
<html>
<head>
    <title>文件上传</title>
</head>
<body>
    上传文件
    <form action="upload" method="post" enctype="multipart/form-data">
        <label for="file">选择文件：</label>
        <input type="file" id="file" name="file" />
        <input type="submit" value="上传" />
    </form>
</body>
</html>

在上面的代码中，我们创建了一个HTML表单，表单的action属性指向了处理文件上传请求的Servlet。

步骤 2：编写Servlet处理文件上传请求

接下来，我们需要创建一个Servlet来处理上传的文件。

import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.nio.file.Paths;
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
public class FileUploadServlet extends HttpServlet {
    protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        Part filePart = request.getPart("file"); // 获取上传的文件
        String fileName = Paths.get(filePart.getSubmittedFileName()).getFileName().toString(); // 获取文件名

        // 处理文件上传
        // ...
    }
}

在此代码中，我们创建了一个名为FileUploadServlet的Servlet，并使用@MultipartConfig注解来指示Servlet可以处理文件上传。
我们使用request.getPart("file")获取上传的文件，并通过getSubmittedFileName()获取文件名。

步骤 3：检查文件类型并添加限制

现在我们需要检查上传文件的类型，并确保其符合预定义的限制。

String[] allowedTypes = {"image/jpeg", "image/png", "application/pdf"};

String fileType = filePart.getContentType(); // 获取文件的MIME类型
boolean isValid = false;

// 检查文件类型是否在允许的列表中
for (String type : allowedTypes) {
    if (fileType.equals(type)) {
        isValid = true;
        break;
    }
}

if (!isValid) {
    response.getWriter().print("上传失败：文件类型不支持！");
    return;
}

// 如果文件类型合法，继续保存文件

在这里，我们首先定义了一个允许的文件类型数组allowedTypes。
通过filePart.getContentType()获取上传文件的MIME类型，并检查它是否在允许的类型中。
如果文件类型不合法，我们便返回一个错误消息并结束处理。

步骤 4：返回响应给用户

最后，处理完上传之后，我们需要在服务器上保存文件并给用户一个响应。

String savePath = "upload/" + fileName; // 保存文件的路径
try (FileOutputStream fos = new FileOutputStream(new File(savePath));
     InputStream is = filePart.getInputStream()) {
    
    byte[] buffer = new byte[1024];
    int bytesRead;
    // 将文件内容写入到服务器本地
    while ((bytesRead = is.read(buffer)) != -1) {
        fos.write(buffer, 0, bytesRead);
    }
    response.getWriter().print("文件上传成功！");

} catch (IOException e) {
    response.getWriter().print("上传失败：出现错误！");
}