diff --git a/paimon-benchmark/paimon-cluster-benchmark/src/main/java/org/apache/paimon/benchmark/Sink.java b/paimon-benchmark/paimon-cluster-benchmark/src/main/java/org/apache/paimon/benchmark/Sink.java index e853d82ee1fc..2054b0c28512 100644 --- a/paimon-benchmark/paimon-cluster-benchmark/src/main/java/org/apache/paimon/benchmark/Sink.java +++ b/paimon-benchmark/paimon-cluster-benchmark/src/main/java/org/apache/paimon/benchmark/Sink.java @@ -30,6 +30,7 @@ import java.util.Iterator; import java.util.List; import java.util.Map; +import java.util.stream.Stream; /** Benchmark sink. */ public class Sink { @@ -66,18 +67,20 @@ public static List load(Path location) throws IOException { Path sinkLocation = location.resolve("sinks"); List result = new ArrayList<>(); - Iterator it = Files.list(sinkLocation).iterator(); - while (it.hasNext()) { - Path p = it.next(); - Map yaml = - BenchmarkUtils.YAML_MAPPER.readValue( - FileUtils.readFileUtf8(p.toFile()), Map.class); - result.add( - new Sink( - FilenameUtils.removeExtension(p.getFileName().toString()), - (String) yaml.get("before"), - (String) yaml.get("sink-name"), - (String) yaml.get("sink-properties"))); + try (Stream paths = Files.list(sinkLocation)) { + Iterator it = paths.iterator(); + while (it.hasNext()) { + Path p = it.next(); + Map yaml = + BenchmarkUtils.YAML_MAPPER.readValue( + FileUtils.readFileUtf8(p.toFile()), Map.class); + result.add( + new Sink( + FilenameUtils.removeExtension(p.getFileName().toString()), + (String) yaml.get("before"), + (String) yaml.get("sink-name"), + (String) yaml.get("sink-properties"))); + } } return result; }