我用下面的代码片段下载PDF文件(我把它从here,学分Josh M)如何区分pdf和非pdf文件?
public final class FileDownloader {
private FileDownloader(){}
public static void main(String args[]) throws IOException{
download("http://pdfobject.com/pdf/sample.pdf", new File("sample.pdf"));
}
public static void download(final String url, final File destination) throws IOException {
final URLConnection connection = new URL(url).openConnection();
connection.setConnectTimeout(60000);
connection.setReadTimeout(60000);
connection.addRequestProperty("User-Agent", "Mozilla/5.0");
final FileOutputStream output = new FileOutputStream(destination, false);
final byte[] buffer = new byte[2048];
int read;
final InputStream input = connection.getInputStream();
while((read = input.read(buffer)) > -1)
output.write(buffer, 0, read);
output.flush();
output.close();
input.close();
}
}
它可以完美兼容PDF文件。然而,正如我遇到一个“坏档案”......我不知道该文件的扩展名是什么,但似乎我陷入了无限循环while((read = input.read(buffer)) > -1)
。我该如何改进这个片段来丢弃任何不适当的文件(非pdf)?
*它适用于pdf文件。但是,正如我遇到一个“坏档案”* - 您是否检查过这是否真的是PDF还是PDF的问题?你在这种情况下检查过目标文件的内容吗? – mkl