2013-01-25 72 views
15

有没有办法在Apache POI中确定MS Office Excel文件类型?我需要知道什么格式的Excel文件:在Excel '97(-2007)(.xls)或Excel 2007 OOXML(.xlsx)中。使用Apache POI确定MS Excel文件类型

我想我可以做这样的事情:

int type = PoiTypeHelper.getType(file); 
switch (type) { 
case PoiType.EXCEL_1997_2007: 
    ... 
    break; 
case PoiType.EXCEL_2007: 
    ... 
    break; 
default: 
    ... 
} 

感谢。

+1

为什么你需要知道前面?你不能只使用WorkbookFactory并让它为你创建合适的类型吗? – Gagravarr

+0

这也是一个很好的变体,谢谢。 –

回答

35

推进到一个答案评论...

如果你打算做与文件一些特别的东西,然后rjokelai's answer是做它的方式。但是,如果你打算使用HSSF/XSSF/Common SS用户模型,那么让POI为你做这件事更简单,并且使用WorkbookFactory来检测并打开你的类型。你会做这样的事情:

Workbook wb = WorkbookFactory.create(new File("something.xls")); 

Workbook wb = WorkbookFactory.create(request.getInputStream()); 

然后,如果你需要做一些特别的东西,测试,如果它是一个HSSFWorkbookXSSFWorkbook。打开文件时,use a File rather than an InputStream if possible可以加快速度并节省内存。

如果您不知道您的文件是什么,请使用Apache Tika来执行检测 - 它可以为您检测大量不同的文件格式。

+0

Thanks Gagravarr ! –

21

您可以使用:

// For .xlsx 
POIXMLDocument.hasOOXMLHeader(new BufferedInputStream(new FileInputStream(file))); 

// For .xls 
POIFSFileSystem.hasPOIFSHeader(new BufferedInputStream(new FileInputStream(file))); 

这些基本上是,WorkbookFactory#create(InputStream)用来判断类型

请注意方法,这两种方法仅支持流支持“标记”功能(或PushBackInputStream) ,所以不支持简单的FileInputStream。使用BufferedInputStream作为包装。出于这个原因,在检测之后,您可以简单地重新使用该流,因为它将被重新设置为起点。

+1

谢谢。当您使用基于低级别基于事件的只读访问来处理大型Excel文件时,这是检测文件类型的唯一方法。 – SWilk

+1

添加了仅可使用“标记”支持流的信息。更新的代码示例。 –

+0

方法'hasOOXMLHeader'在类POIXMLDocument中被描述,在类[DocumentFactoryHelper](https://poi.apache.org/apidocs/org/apache/poi/poifs/filesystem/DocumentFactoryHelper.html)中使用它 – KAD

1

基础上的lib实施org.apache.poi.ss.usermodel.WorkbookFactory#create(java.io.InputStream)

我们可以模仿WorkbookFactory的逻辑,去除不相关的位,返回文件类型来代替。

public static TYPE fileType(File file) { 
    try (
      InputStream inp = new FileInputStream(file) 
    ) { 
     if (!(inp).markSupported()) { 
      return getNotMarkSupportFileType(file); 
     } 
     return getType(inp); 
    } catch (IOException e) { 
     LOGGER.error("Analyse FileType Problem.", e); 
     return TYPE.INVALID; 
    } 
} 

private static TYPE getNotMarkSupportFileType(File file) throws IOException { 
    try (
      InputStream inp = new PushbackInputStream(new FileInputStream(file), 8) 
    ) { 
     return getType(inp); 
    } 
} 

private static TYPE getType(InputStream inp) throws IOException { 
    byte[] header8 = IOUtils.peekFirst8Bytes(inp); 
    if (NPOIFSFileSystem.hasPOIFSHeader(header8)) { 
     NPOIFSFileSystem fs = new NPOIFSFileSystem(inp); 
     return fileType(fs); 
    } else if (DocumentFactoryHelper.hasOOXMLHeader(inp)) { 
     return TYPE.XSSF_WORKBOOK; 
    } 
    return TYPE.INVALID; 
} 

private static TYPE fileType(NPOIFSFileSystem fs) { 
    DirectoryNode root = fs.getRoot(); 
    if (root.hasEntry("EncryptedPackage")) { 
     return TYPE.XSSF_WORKBOOK; 
    } 
    return TYPE.HSSF_WORKBOOK; 

} 

public enum TYPE { 
    HSSF_WORKBOOK, XSSF_WORKBOOK, INVALID 
}