Minecraft-SSHD/src/main/java/com/ryanmichela/sshd/FlushyOutputStream.java

package com.ryanmichela.sshd;

import org.apache.sshd.common.SshException;

import java.io.IOException;
import java.io.OutputStream;

/**
 * Copyright 2013 Ryan Michela
 */
public class FlushyOutputStream extends OutputStream {

    private OutputStream base;
    private boolean isClosed = false;

    public FlushyOutputStream(OutputStream base) {
        this.base = base;
    }

    @Override
    public void write(int b) throws IOException {
        if (isClosed) return;
        base.write(b);
        base.flush();
    }

    @Override
    public void write(byte[] b) throws IOException {
        if (isClosed) return;
        base.write(b);
        base.flush();
    }

    @Override
    public void write(byte[] b, int off, int len) throws IOException {
        if (isClosed) return;
        try {
            base.write(b, off, len);
            base.flush();
        } catch (SshException e) {
            if (!e.getMessage().contains("channel already closed")) throw e;
        }
    }

    @Override
    public void close() {
        isClosed = true;
    }
}