vip.ipav.okhttp.body.ProgressRequestBody Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of okhttp-tools Show documentation

OKHttp3的简单封装，链式调用

There is a newer version: 1.3.1

package vip.ipav.okhttp.body;


import okhttp3.MediaType;
import okhttp3.RequestBody;
import okio.*;
import vip.ipav.okhttp.response.IResponseHandler;

import java.io.IOException;


public class ProgressRequestBody extends RequestBody {

    /**
     * 回调监听
     */
    private IResponseHandler mResponseHandler;
    private RequestBody mRequestBody;
    private CountingSink mCountingSink;

    public ProgressRequestBody(RequestBody requestBody, IResponseHandler responseHandler) {
        this.mResponseHandler = responseHandler;
        this.mRequestBody = requestBody;
    }

    @Override
    public MediaType contentType() {
        return mRequestBody.contentType();
    }

    @Override
    public long contentLength() throws IOException {
        return mRequestBody.contentLength();
    }

    @Override
    public void writeTo(BufferedSink sink) throws IOException {
        mCountingSink = new CountingSink(sink);
        BufferedSink bufferedSink = Okio.buffer(mCountingSink);
        mRequestBody.writeTo(bufferedSink);
        bufferedSink.flush();
    }

    protected final class CountingSink extends ForwardingSink {
        private long bytesWritten = 0;
        private long contentLength = 0L;
        public CountingSink(Sink delegate) {
            super(delegate);
        }
        @Override
        public void write(Buffer source, long byteCount) throws IOException {
            super.write(source, byteCount);
            if (contentLength == 0) {
                //获得contentLength的值，后续不再调用
                contentLength = contentLength();
            }
            bytesWritten += byteCount;

            mResponseHandler.onProgress(bytesWritten, contentLength);
        }
    }
}