Add output registry for custom outputs

core/src/main/java/io/jstach/rainbowgum/LogConfig.java

@@ -16,7 +16,7 @@
  * The configuration of a RainbowGum. In some other logging implementations this is called
  * "context".
  */
-public interface LogConfig {
+public sealed interface LogConfig {
 
 	/**
 	 * String kv properties.
@@ -50,8 +50,8 @@ default <T> PropertyValue<T> get(Property<T> property) {
 	 * Output provider that uses URI to find output.
 	 * @return output provider.
 	 */
-	default LogOutputProvider outputProvider() {
-		return LogOutputProvider.of();
+	default LogOutputRegistry outputRegistry() {
+		return LogOutputRegistry.of();
 	}
 
 	/**
@@ -61,7 +61,7 @@ default LogOutputProvider outputProvider() {
 	 * @throws IOException if output fails fast
 	 */
 	default LogOutput output(URI uri) throws IOException {
-		return outputProvider().output(uri);
+		return outputRegistry().output(uri);
 	}
 
 	/**
@@ -154,7 +154,7 @@ public boolean isEnabled(String loggerName) {
 
 }
 
-class DefaultLogConfig implements LogConfig {
+final class DefaultLogConfig implements LogConfig {
 
 	private final ServiceRegistry registry;
 

core/src/main/java/io/jstach/rainbowgum/LogOutputProvider.java

@@ -1,16 +1,12 @@
 package io.jstach.rainbowgum;
 
-import java.io.FileOutputStream;
 import java.io.IOException;
 import java.net.URI;
-import java.nio.channels.FileChannel;
-import java.nio.file.Paths;
-import java.nio.file.StandardOpenOption;
 
 /**
  * Finds output based on URI.
  */
-public interface LogOutputProvider extends AutoCloseable {
+public interface LogOutputProvider {
 
 	/**
 	 * Loads an output from a URI.
@@ -20,44 +16,4 @@ public interface LogOutputProvider extends AutoCloseable {
 	 */
 	LogOutput output(URI uri) throws IOException;
 
-	default void close() {
-	}
-
-	/**
-	 * Default output provider.
-	 * @return default output provider.
-	 */
-	public static LogOutputProvider of() {
-		return DefaultOutputProvider.INSTANCE;
-	}
-
-}
-
-enum DefaultOutputProvider implements LogOutputProvider {
-
-	INSTANCE;
-
-	@Override
-	public LogOutput output(URI uri) throws IOException {
-		String scheme = uri.getScheme();
-		String path = uri.getPath();
-		if (scheme == null && path != null) {
-			@SuppressWarnings("resource")
-			FileOutputStream fos = new FileOutputStream(path);
-			return LogOutput.of(uri, fos.getChannel());
-		}
-		else if (LogOutput.STDOUT_SCHEME.equals(scheme)) {
-			return LogOutput.ofStandardOut();
-		}
-		else if (LogOutput.STDERR_SCHEME.equals(scheme)) {
-			return LogOutput.ofStandardErr();
-		}
-		else {
-			var p = Paths.get(uri);
-			var channel = FileChannel.open(p, StandardOpenOption.APPEND, StandardOpenOption.CREATE);
-			channel.close();
-			return LogOutput.of(uri, channel);
-		}
-	}
-
 }

core/src/main/java/io/jstach/rainbowgum/LogOutputRegistry.java

@@ -0,0 +1,70 @@
+package io.jstach.rainbowgum;
+
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.net.URI;
+import java.nio.channels.FileChannel;
+import java.nio.file.Paths;
+import java.nio.file.StandardOpenOption;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+
+/**
+ * Register output providers by URI scheme.
+ */
+public interface LogOutputRegistry extends LogOutputProvider {
+
+	/**
+	 * Register a provider by {@link URI#getScheme() scheme}.
+	 * @param scheme URI scheme to match for.
+	 * @param provider provider for scheme.
+	 */
+	public void put(String scheme, LogOutputProvider provider);
+
+	/**
+	 * Default output provider.
+	 * @return default output provider.
+	 */
+	public static LogOutputRegistry of() {
+		return new DefaultOutputRegistry();
+	}
+
+}
+
+class DefaultOutputRegistry implements LogOutputRegistry {
+
+	private final Map<String, LogOutputProvider> providers = new ConcurrentHashMap<>();
+
+	@Override
+	public void put(String scheme, LogOutputProvider provider) {
+		providers.put(scheme, provider);
+	}
+
+	@Override
+	public LogOutput output(URI uri) throws IOException {
+		String scheme = uri.getScheme();
+		String path = uri.getPath();
+		LogOutputProvider customProvider;
+		if (scheme == null && path != null) {
+			@SuppressWarnings("resource")
+			FileOutputStream fos = new FileOutputStream(path);
+			return LogOutput.of(uri, fos.getChannel());
+		}
+		else if (LogOutput.STDOUT_SCHEME.equals(scheme)) {
+			return LogOutput.ofStandardOut();
+		}
+		else if (LogOutput.STDERR_SCHEME.equals(scheme)) {
+			return LogOutput.ofStandardErr();
+		}
+		else if ((customProvider = providers.get(scheme)) != null) {
+			return customProvider.output(uri);
+		}
+		else {
+			var p = Paths.get(uri);
+			var channel = FileChannel.open(p, StandardOpenOption.APPEND, StandardOpenOption.CREATE);
+			channel.close();
+			return LogOutput.of(uri, channel);
+		}
+	}
+
+}

core/src/main/java/io/jstach/rainbowgum/spi/RainbowGumServiceProvider.java

@@ -7,7 +7,9 @@
 
 import org.eclipse.jdt.annotation.Nullable;
 
+import io.jstach.rainbowgum.Defaults;
 import io.jstach.rainbowgum.LogConfig;
+import io.jstach.rainbowgum.LogOutputProvider;
 import io.jstach.rainbowgum.LogProperties;
 import io.jstach.rainbowgum.RainbowGum;
 import io.jstach.rainbowgum.ServiceRegistry;
@@ -47,7 +49,11 @@ public non-sealed interface ConfigProvider extends RainbowGumServiceProvider {
 	}
 
 	/**
-	 * Called after {@link LogConfig} has been loaded.
+	 * Called after {@link LogConfig} has been loaded to do various custom initialization
+	 * like registering {@link LogOutputProvider}s.
+	 *
+	 * @see LogConfig#outputRegistry()
+	 * @see Defaults#formatterForOutputType(io.jstach.rainbowgum.LogOutput.OutputType)
 	 */
 	public non-sealed interface Initializer extends RainbowGumServiceProvider {