在Stream API中使用自动关闭接口[重复]

Question

Set<Long> ids = new HashSet<>();
for (String fileName : fileSystem.list("my-directory")) {
    InputStream stream = fileSystem.openInputStream(fileName);
    BufferedReader br = new BufferedReader(new InputStreamReader(stream));
    String line;
    while ((line = br.readLine()) != null) {
        ids.add(Long.valueOf(line.trim()));
    }
    br.close();
}

Set<Long> ids = fileSystem.list("my-directory").stream()
    .map(fileName -> fileSystem::openInputStream)
    .map(is -> new BufferedReader(new InputStreamReader(is)))
    .flatMap(BufferedReader::lines)
    .map(String::trim)
    .map(Long::valueOf)
    .collect(Collectors.toSet());

Upd：示例中的文件系统是HDFS，files#lines，不能使用类似的方法。

Answer 1

一旦流的所有元素都被消耗完，就可以挂钩到流中来“关闭”资源。因此，可以在读取所有行后通过以下修改关闭读取器：

.flatMap(reader -> reader.lines().onClose(() -> close(reader)))

其中close(autoclocable)处理IOException。

作为概念证明，已经测试了以下代码和输出：

import java.util.stream.Stream;

class Test {
    public static void main(String[] args) {
        Stream.of(1, 2, 3).flatMap(i ->
                Stream.of(i, i * 2).onClose(() ->
                        System.out.println("Closed!")
                )
        ).forEach(System.out::println);
    }
}

1
2
Closed!
2
4
Closed!
3
6
Closed!