com.github.akurilov.commons.io.TextStreamOutput Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of java-commons Show documentation

Common functionality Java library

The newest version!

package com.github.akurilov.commons.io;

import java.io.BufferedWriter;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.util.List;

import static com.github.akurilov.commons.lang.Exceptions.throwUnchecked;

/**
 * The input implementation designed to write the text lines using {@link BufferedWriter}
 */
public abstract class TextStreamOutput
implements Output {
	
	private static final ThreadLocal THRLOC_STRB = ThreadLocal.withInitial(
		StringBuilder::new
	);
	private static final String LINE_SEP = System.getProperty("line.separator");
	
	protected final BufferedWriter writer;

	public TextStreamOutput(final OutputStream out) {
		writer = new BufferedWriter(new OutputStreamWriter(out));
	}

	public TextStreamOutput(final OutputStream out, final int buffSize) {
		writer = new BufferedWriter(new OutputStreamWriter(out), buffSize);
	}
	
	@Override
	public boolean put(final String line) {
		try {
			writer.write(line + LINE_SEP);
		} catch(final IOException e) {
			throwUnchecked(e);
		}
		return true;
	}

	/**
	 * Uses the thread local string builder to concatenate the lines into the single text block for better performance
	 * @throws IOException
	 */
	@Override
	public int put(final List lines, final int from, final int to) {
		final var strb = THRLOC_STRB.get();
		strb.setLength(0);
		for(var i = from; i < to; i ++) {
			strb.append(lines.get(i));
			strb.append(LINE_SEP);
		}
		try {
			writer.write(strb.toString());
		} catch(final IOException e) {
			throwUnchecked(e);
		}
		return to - from;
	}

	/**
	 * Uses the thread local string builder to concatenate the lines into the single text block for better performance
	 * @throws IOException
	 */
	@Override
	public int put(final List lines) {
		final var strb = THRLOC_STRB.get();
		strb.setLength(0);
		for(final var line : lines) {
			strb.append(line);
			strb.append(LINE_SEP);
		}
		try {
			writer.write(strb.toString());
		} catch(final IOException e) {
			throwUnchecked(e);
		}
		return lines.size();
	}
	
	@Override
	public void close() {
		try {
			writer.flush();
			writer.close();
		} catch(final IOException e) {
			throwUnchecked(e);
		}
	}
}