com.king.platform.net.http.ResponseBodyStream Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of king-http-client Show documentation

A asyncronous http client built ontop of netty.

There is a newer version: 3.0.26

// Copyright (C) king.com Ltd 2015
// https://github.com/king/king-http-client
// Author: Magnus Gustafsson
// License: Apache 2.0, https://raw.github.com/king/king-http-client/LICENSE-APACHE

package com.king.platform.net.http;


import java.nio.ByteBuffer;
import java.nio.channels.WritableByteChannel;

public class ResponseBodyStream implements ResponseBodyConsumer {
	private final WritableByteChannel channel;
	private final ProgressInvoker progressInvoker;

	public ResponseBodyStream(WritableByteChannel channel, ProgressCallback progressCallback) {
		this.channel = channel;
		this.progressInvoker = new ProgressInvoker(progressCallback);
	}

	@Override
	public void onBodyStart(String contentType, String charset, long contentLength) throws Exception {
		progressInvoker.setContentLength(contentLength);
	}

	@Override
	public void onReceivedContentPart(ByteBuffer buffer) throws Exception {

		int remaining = buffer.remaining();
		channel.write(buffer);

		progressInvoker.updateReceivedContent(remaining);
	}

	@Override
	public void onCompletedBody() throws Exception {

		channel.close();
	}

	@Override
	public WritableByteChannel getBody() {
		return channel;
	}


	private static class ProgressInvoker {
		private ProgressCallback progressCallback;
		private long contentLength;
		private long readLength;

		public ProgressInvoker(ProgressCallback progressCallback) {
			this.progressCallback = progressCallback;
		}

		public long getContentLength() {
			return contentLength;
		}

		public void setContentLength(long contentLength) {
			this.contentLength = contentLength;
		}

		public long getReadLength() {
			return readLength;
		}


		public void updateReceivedContent(int remaining) {
			if (progressCallback != null) {
				int prev = Math.round (((float) readLength / (float) contentLength) * 100f);

				readLength += remaining;

				int current = Math.round(((float) readLength / (float) contentLength) * 100f);

				if (current != prev) {
					progressCallback.onProgress(current, readLength, contentLength);
				}

			}
		}

	}

	public interface ProgressCallback {
		void onProgress(int percentageCompleted, long readLength, long contentLength);
	}

}