如何区分pdf和非pdf文件？

Question

我使用下面的 sn-p 下载了 pdf 文件（我从 here 拿来的，感谢 Josh M）

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)的无限循环。如何改进这个 sn-p 以丢弃任何类型的不适当文件（非 pdf）？

Answer 1

有一个类似问题的问题：Infinite Loop in Input Stream。

查看可能的解决方案：Abort loop after fixed time。

您可以尝试为连接设置超时：Java URLConnection Timeout。