com.couchbase.lite.router.BufferOutputStream Maven / Gradle / Ivy

Go to download
package com.couchbase.lite.router;

import java.io.IOException;
import java.io.OutputStream;
import java.util.concurrent.atomic.AtomicBoolean;

public class BufferOutputStream extends OutputStream {

    ByteBuffer buffer = new ByteBuffer();
    AtomicBoolean closed = new AtomicBoolean(false);

    public void write(int i) throws IOException {
        if(!isClosed()) {
            synchronized (buffer) {
                buffer.push((byte) i);
                buffer.notify();
            }
        } else {
            throw new IOException("Can't write to closed stream.");
        }
    }

    ByteBuffer getBuffer() {
        return buffer;
    }

    public void close() {
        closed.set(true);
        synchronized(buffer) {
            buffer.notify();
        }
    }

    public boolean isClosed() {
        return closed.get();
    }

}