EasyExcel工具类

校验上传的文件是否是Excel文件及是否是xls或xlsx格式工具类

import org.apache.poi.poifs.filesystem.FileMagic;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.multipart.MultipartFile;

import java.io.InputStream;
import java.util.Objects;

public class EasyExcelUtil {
    
    
    private static final Logger logger = LoggerFactory.getLogger(EasyExcelUtil.class);

    /**
     * 校验上传的文件是否是Excel文件及是否是xls或xlsx格式
     *
     * @param file
     * @return
     * @throws Exception
     */
    private static Boolean checkExcelFile(MultipartFile file) throws Exception {
    
    

        Boolean result = false;

        try {
    
    
            //校验文件名称
            String fileName = file.getOriginalFilename();
            if (fileName == null) {
    
    
                logger.error("fileName is null");
                throw new Exception("未获取到文件名");
            }
            //校验文件后缀名
            String fileSuffixName = fileName.substring(fileName.lastIndexOf(".") + 1);
            if (fileSuffixName == null) {
    
    
                logger.error("Illegal file");
                throw new Exception("非法文件");
            }
            //校验文件后缀名是否为xls或者xlsx格式
            if (!"xls".equals(fileSuffixName) && !"xlsx".equals(fileSuffixName)) {
    
    
                logger.error("file format error,please upload an Excel file in xls or xlsx format");
                throw new Exception("文件格式错误，请上传xls或xlsx格式的Excel文件");
            }
            //根据Excel魔数判断该文件是否为Excel文件
            InputStream inputStream = file.getInputStream();
            InputStream fileMagics = FileMagic.prepareToCheckMagic(inputStream);
            FileMagic fileMagic = FileMagic.valueOf(fileMagics);

            //FileMagic.OLE2表示xls格式 FileMagic.OOXML表示xlsx格式
            if (Objects.equals(fileMagic, FileMagic.OLE2) || Objects.equals(fileMagic, FileMagic.OOXML)) {
    
    
                result = true;
            } else {
    
    
                logger.error("file format error,please upload an Excel file in xls or xlsx format");
                throw new Exception("文件格式错误，请上传xls或xlsx格式的Excel文件");
            }
        } catch (Exception e) {
    
    
            logger.error("请上传xls或xlsx格式的Excel文件", e);
            throw e;
        }
        return result;
    }
}