java怎么区分上传的文件是xls还是xlsx

原创

mob64ca12ea8117 2024-08-02 13:26:57 ©著作权

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

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

Java区分上传文件格式的项目方案

在现代的应用程序中，处理文档上传是一个常见的功能。Excel文档由于其广泛应用的程度，尤其是.xls和.xlsx格式的文件，往往需要进行区分。本文将提出一个基于Java的项目方案，涉及文件的上传、格式识别和相应的处理。

项目目标

本项目的主要目标是设计一个Java应用程序，能够通过文件的MIME类型和文件后缀名来区分用户上传的文件是.xls还是.xlsx格式的Excel文件。在此基础上，程序将提供进一步处理Excel文件的能力。

系统架构

为了实现该功能，我们的系统将分为多个模块，包括：

文件上传模块：负责接收用户上传的文件。
文件类型检测模块：根据文件的MIME类型和后缀名判断其格式。
文件处理模块：加载和处理Excel文件的数据。

类图

以下是本项目的类图，采用Mermaid语法描述：

classDiagram
    class FileUploader {
        +uploadFile(file: File) : void
    }
    
    class FileTypeDetector {
        +detectFileType(file: File) : String
    }
    
    class ExcelFileProcessor {
        +processExcel(file: File) : void
    }
    
    FileUploader --> FileTypeDetector
    FileTypeDetector --> ExcelFileProcessor

流程图

整个文件处理的流程如下，将以图示形式展现：

flowchart TD
    A[用户上传文件] --> B{文件类型检测}
    B -->|xls| C[处理xls文件]
    B -->|xlsx| D[处理xlsx文件]
    C --> E[返回处理结果]
    D --> E

代码示例

在Java中，我们可以使用Apache POI库来处理Excel文件。以下是实现文件上传和格式检测的示例代码。

文件上传模块

首先，我们需要创建一个文件上传模块。可以使用Servlet来处理上传文件的请求。

import java.io.File;
import java.io.IOException;
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 FileUploader extends HttpServlet {
    protected void doPost(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        Part filePart = request.getPart("file"); // 获取文件部分
        FileTypeDetector detector = new FileTypeDetector();
        String fileType = detector.detectFileType(filePart);
        
        // 根据文件类型调用对应的处理方法
        if ("xls".equals(fileType)) {
            new ExcelFileProcessor().processExcel(filePart.getInputStream(), "xls");
        } else if ("xlsx".equals(fileType)) {
            new ExcelFileProcessor().processExcel(filePart.getInputStream(), "xlsx");
        } else {
            response.getWriter().write("不支持的文件格式");
        }
    }
}

文件类型检测模块

import java.io.File;
import java.nio.file.Files;

public class FileTypeDetector {
    public String detectFileType(Part file) throws IOException {
        String fileName = file.getSubmittedFileName();
        String mimeType = Files.probeContentType(new File(fileName).toPath());

        // 检查文件后缀 
        if (fileName.endsWith(".xls") || mimeType.equals("application/vnd.ms-excel")) {
            return "xls";
        } else if (fileName.endsWith(".xlsx") || mimeType.equals("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")) {
            return "xlsx";
        }
        return "unsupported";
    }
}

文件处理模块

import java.io.InputStream;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;

public class ExcelFileProcessor {
    public void processExcel(InputStream inputStream, String fileType) {
        try {
            Workbook workbook;
            if ("xls".equals(fileType)) {
                workbook = new HSSFWorkbook(inputStream);
            } else {
                workbook = new XSSFWorkbook(inputStream);
            }
            // 处理Excel内容
            Sheet sheet = workbook.getSheetAt(0);
            for (Row row : sheet) {
                for (Cell cell : row) {
                    System.out.print(cell.toString() + "\t");
                }
                System.out.println();
            }
            workbook.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}