com.github.kubatatami.judonetworking.RequestOutputStream Maven / Gradle / Ivy

Go to download
package com.github.kubatatami.judonetworking;

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

/**
 * Created with IntelliJ IDEA.
 * User: jbogacki
 * Date: 09.09.2013
 * Time: 10:10
 * To change this template use File | Settings | File Templates.
 */
public class RequestOutputStream extends OutputStream {
    private OutputStream stream;
    private TimeStat timeStat;
    private long contentSize;

    public RequestOutputStream(OutputStream stream, TimeStat timeStat, long contentSize) {
        this.stream = stream;
        this.timeStat = timeStat;
        this.contentSize = contentSize;
        timeStat.setSendTimeProgressTick(false);
    }

    @Override
    public void write(int oneByte) throws IOException {
        stream.write(oneByte);
        timeStat.progressTick(1 / (float) contentSize);
    }

    @Override
    public void close() throws IOException {
        stream.close();
        timeStat.tickSendTime();
    }

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

    @Override
    public void write(byte[] buffer, int offset, int count) throws IOException {
        stream.write(buffer, offset, count);
        timeStat.progressTick((float) count / (float) contentSize);
    }
}