整理的一个校验文件是否是 excel 文件的校验类。
思路是先校验文件后缀名,后校验文件流的头信息。
实现语言是 java.
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
/*
* Excel 文件校验类
* @author saintatgod@gmail.com
*/
public class ExcelFileTypeValidator {
public static void main(String[] args) {
String filePath = "example.xlsx";
// 校验文件后缀名是否是 xlsx
String fileName = filePath.substring(filePath.lastIndexOf("/") + 1);
String extension = fileName.substring(fileName.lastIndexOf(".") + 1);
if (!extension.equalsIgnoreCase("xlsx") && !extension.equalsIgnoreCase("xls")) {
System.out.println("文件后缀不是 excel");
}
// 通过文件流校验是否是 excel
byte[] header = readHeader(filePath);
boolean isXls = checkHeader(header, new byte[]{(byte) 0xD0, (byte) 0xCF, (byte) 0x11, (byte) 0xE0, (byte) 0xA1, (byte) 0xB1, (byte) 0x1A, (byte) 0xE1});
boolean isXlsx = checkHeader(header, new byte[]{(byte) 0x50, (byte) 0x4B, (byte) 0x03, (byte) 0x04, (byte) 0x14, (byte) 0x00, (byte) 0x06, (byte) 0x00});
if(isXls || isXlsx) {
// 是 excel 文件
}else{
// 不是 excel 文件
}
}
/**
* 读取文件的 header 信息
* @param filePath
* @return
*/
private static byte[] readHeader(String filePath) {
byte[] header = new byte[8];
try (InputStream inputStream = new FileInputStream(filePath)) {
inputStream.read(header);
} catch (IOException e) {
e.printStackTrace();
}
return header;
}
/*
* 比较
*/
private static boolean checkHeader(byte[] header, byte[] expectedHeader) {
for (int i = 0; i < header.length; i++) {
if (header[i] != expectedHeader[i]) {
return false;
}
}
return true;
}
}