1 files changed, 41 insertions, 0 deletions
diff --git a/bundle-plugin/src/main/java/com/yahoo/container/plugin/util/IO.java b/bundle-plugin/src/main/java/com/yahoo/container/plugin/util/IO.java
new file mode 100644
index 00000000000..a1e313b920b
--- /dev/null
+++ b/bundle-plugin/src/main/java/com/yahoo/container/plugin/util/IO.java
@@ -0,0 +1,41 @@
+// Copyright 2018 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.container.plugin.util;
+
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileOutputStream;
+import java.io.OutputStream;
+
+/**
+ * Utility methods relating to IO
+ *
+ * @author Tony Vaagenes
+ * @author ollivir
+ */
+public class IO {
+    public static <T> T withFileInputStream(File file, ThrowingFunction<FileInputStream, T> f) {
+        try (FileInputStream fis = new FileInputStream(file)) {
+            return f.apply(fis);
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    /**
+     * Creates a new file and all its parent directories, and provides a file output stream to the file.
+     */
+    public static <T> T withFileOutputStream(File file, ThrowingFunction<OutputStream, T> f) {
+        makeDirectoriesRecursive(file.getParentFile());
+        try (FileOutputStream fos = new FileOutputStream(file)) {
+            return f.apply(fos);
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    private static void makeDirectoriesRecursive(File file) {
+        if (!file.mkdirs() && !file.isDirectory()) {
+            throw new RuntimeException("Could not create directory " + file.getPath());
+        }
+    }
+}