复制InputStream，如果大小超出限制，则中止操作

Question

我试图将InputStream复制到文件，如果InputStream的大小大于1MB，则中止复制。在Java7中，我编写了如下代码：

public void copy(InputStream input, Path target) {
    OutputStream out = Files.newOutputStream(target,
            StandardOpenOption.CREATE_NEW, StandardOpenOption.WRITE);
    boolean isExceed = false;
    try {
        long nread = 0L;
        byte[] buf = new byte[BUFFER_SIZE];
        int n;
        while ((n = input.read(buf)) > 0) {
            out.write(buf, 0, n);
            nread += n;
            if (nread > 1024 * 1024) {// Exceed 1 MB
                isExceed = true;
                break;
            }
        }
    } catch (IOException ex) {
        throw ex;
    } finally {
        out.close();
        if (isExceed) {// Abort the copy
            Files.deleteIfExists(target);
            throw new IllegalArgumentException();
        }
    }}

• 第一个问题：是否有更好的解决方案？
• 第二个问题：我的另一个解决方案-在复制操作之前，我先计算此InputStream的大小。所以我复制InputStream ByteArrayOutputStream然后得到size()。但是问题是InputStream可能不是markSupported()，因此InputStream无法在复制文件操作中重用。

Answer 1

第一个问题：是否有更好的解决方案？

并不是的。当然，不会明显更好。

第二个问题：我的另一个解决方案-
在复制操作之前，我先计算InputStream的大小。所以我将InputStream复制到ByteArrayOutputStream然后获取size（）。但是问题是InputStream可能没有markSupported（），因此InputStream无法在复制文件操作中重用。

撇开上面的说法不是问题…

如果您已将字节复制到ByteArrayOutputStream，则可以ByteArrayInputStream从返回的字节数组中创建一个baos.toByteArray()。因此，您无需标记/重置原始流。

但是，这是实现此方法的非常难看的方法。尤其重要，因为无论如何您都在读取和缓冲 整个 输入流。