在 Java8 流之后清理

Question

当我从无限制的 Stream 开始并使用 limit() 方法对其进行限制时，一旦达到限制，我该如何清理 Stream 使用的资源？例如，如果我尝试执行 Files.lines 方法所做的事情，但使用浮点数而不是字符串，我会编写一个看起来像这样的函数：

public static Stream<Float> floats(File f) throws FileNotFoundException {
        FileInputStream fis = new FileInputStream(f);
        return Stream.generate(() -> {

            byte[] buff = new byte[4];
            try { fis.read(buff); }
            catch (Exception e) { e.printStackTrace(); }
            return ByteBuffer.wrap(buff).getFloat(0);

        }).limit(f.length()/4);
    }

这应该会流式传输一个充满浮点数的大型二进制文件，直到我到达文件末尾。我将文件的长度（以字节为单位）除以构成浮点数的四个字节。但是，一旦达到此限制，我想找到某种方法来执行 fis.close() 。 Streams API 有什么方法可以让您这样做吗？

Answer 1

我想出了另一种可能性。这有点像黑客，但它似乎有效。我只是通过一个始终返回 true 的过滤器运行 Stream，但在此过程中，它会检查 EOF 并在到达时关闭输入流。我希望 API 中有一种更简洁的方法来执行此操作。

    Stream<Float> answer;
    FileInputStream fis = new FileInputStream(f);

    answer = Stream
            .generate(() -> {
                byte[] buff = new byte[4];
                try { fis.read(buff); }
                catch (Exception e) { e.printStackTrace(); }
                return ByteBuffer.wrap(buff).getFloat(0);})
            .filter(x -> { 
                try { if (fis.available() == 0) fis.close(); } 
                catch (Exception e) { e.printStackTrace(); }
                return true;})
            .limit(f.length()/4);

    return answer;

Answer 2

我会使用完全不同的方法：

public static Stream<Float> floats(File f) throws FileNotFoundException {
    try(FileChannel fch = FileChannel.open(f.toPath(), StandardOpenOption.READ)) {
        ByteBuffer  bb = fch.map(FileChannel.MapMode.READ_ONLY, 0, fch.size());
        FloatBuffer fb = bb.asFloatBuffer();
        return IntStream.range(0, fb.remaining()).mapToObj(fb::get);
    }
    catch(IOException ex) { // or consider to declare "throws IOException" instead
        FileNotFoundException fnfe = new FileNotFoundException(ex.getMessage());
        fnfe.initCause(ex);
        throw fnfe;
    }
}

这样，FileChannel 在方法返回时已经关闭。它也可能更有效。请注意，对于很多操作，将数据处理为DoubleStream 可能会更有效，即使源值为floats 并且仅将最终结果的类型缩小回float，如有必要：

public static DoubleStream floatsAsDouble(File f) throws FileNotFoundException {
    try(FileChannel fch = FileChannel.open(f.toPath(), StandardOpenOption.READ)) {
        ByteBuffer  bb = fch.map(FileChannel.MapMode.READ_ONLY, 0, fch.size());
        FloatBuffer fb = bb.asFloatBuffer();
        return IntStream.range(0, fb.remaining()).mapToDouble(fb::get);
    }
    catch(IOException ex) { // or consider to declare throwsIOException instead
        FileNotFoundException fnfe = new FileNotFoundException(ex.getMessage());
        fnfe.initCause(ex);
        throw fnfe;
    }
}
…
    float sum=(float)floatsAsDouble(new File("path")).sum();

另外，您可以考虑直接使用Paths 而不是File。