com.envisioniot.enos.iot_http_sdk.progress.CountingSink Maven / Gradle / Ivy

Go to download
package com.envisioniot.enos.iot_http_sdk.progress;

import java.io.IOException;

import lombok.Data;
import lombok.EqualsAndHashCode;
import okio.Buffer;
import okio.ForwardingSink;
import okio.Sink;

/**
 * A Sink inherited class that can count data traffic
 * @author shenjieyuan
 */
@Data
@EqualsAndHashCode(callSuper = true)
public class CountingSink extends ForwardingSink
{
    private long bytesWritten = 0L;
    private IProgressListener listener;
    private long contentLength = 0L;

    public CountingSink(Sink delegate)
    {
        super(delegate);
    }

    public CountingSink(Sink delegate, IProgressListener listener, long contentLength)
    {
        super(delegate);
        this.listener = listener;
        this.contentLength = contentLength;
    }

    @Override
    public void write(Buffer source, long byteCount) throws IOException
    {
        super.write(source, byteCount);

        bytesWritten += byteCount;
        if (listener != null)
        {
            listener.onRequestProgress(bytesWritten, contentLength);
        }
    }
}