de.siegmar.fastcsv.writer.FastBufferedWriter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of fastcsv Show documentation

Lightning-fast, dependency-free CSV library that conforms to RFC standards.

There is a newer version: 3.3.1

package de.siegmar.fastcsv.writer;

import java.io.FilterWriter;
import java.io.IOException;
import java.io.Writer;

/**
 * High-performance buffered writer (without synchronization).
 */
final class FastBufferedWriter extends FilterWriter implements Writable {

    private final char[] buf;
    private final boolean autoFlushBuffer;
    private final boolean autoFlushWriter;
    private int pos;

    FastBufferedWriter(final Writer writer, final int bufferSize,
                       final boolean autoFlushBuffer, final boolean autoFlushWriter) {
        super(writer);
        buf = new char[bufferSize];
        this.autoFlushBuffer = autoFlushBuffer;
        this.autoFlushWriter = autoFlushWriter;
    }

    @Override
    public void write(final int c) throws IOException {
        if (pos == buf.length) {
            flushBuffer();
        }
        buf[pos++] = (char) c;
    }

    @Override
    public void write(final char[] cbuf, final int off, final int len) throws IOException {
        if (pos + len >= buf.length) {
            flushBuffer();
            if (len >= buf.length) {
                out.write(cbuf, off, len);
                return;
            }
        }

        System.arraycopy(cbuf, off, buf, pos, len);
        pos += len;
    }

    @Override
    public void write(final String str, final int off, final int len) throws IOException {
        if (pos + len >= buf.length) {
            flushBuffer();
            if (len >= buf.length) {
                out.write(str, off, len);
                return;
            }
        }

        str.getChars(off, off + len, buf, pos);
        pos += len;
    }

    @Override
    public void endRecord() throws IOException {
        if (autoFlushWriter) {
            flush();
        } else if (autoFlushBuffer) {
            flushBuffer();
        }
    }

    private void flushBuffer() throws IOException {
        out.write(buf, 0, pos);
        pos = 0;
    }

    @Override
    public void flush() throws IOException {
        flushBuffer();
        super.flush();
    }

    @Override
    public void close() throws IOException {
        flushBuffer();
        super.close();
    }

}