internal.io.text.FunctionalTextFormatter Maven / Gradle / Ivy

Go to download
package internal.io.text;

import lombok.NonNull;
import nbbrd.io.function.IOBiConsumer;
import nbbrd.io.text.TextFormatter;

import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.nio.charset.Charset;

import static nbbrd.io.Resource.uncloseableOutputStream;
import static nbbrd.io.text.TextResource.uncloseableWriter;

@lombok.RequiredArgsConstructor
public final class FunctionalTextFormatter implements TextFormatter {

    @NonNull
    private final IOBiConsumer function;

    @Override
    public void formatWriter(@NonNull T value, @NonNull Writer resource) throws IOException {
        doWrite(value, uncloseableWriter(resource));
    }

    @Override
    public void formatStream(@NonNull T value, @NonNull OutputStream resource, @NonNull Charset encoding) throws IOException {
        try (OutputStreamWriter writer = new OutputStreamWriter(uncloseableOutputStream(resource), encoding)) {
            doWrite(value, writer);
        }
    }

    private void doWrite(T value, Writer resource) throws IOException {
        function.acceptWithIO(value, resource);
    }
}