com.dg.libs.rest.entities.CountingInputStreamEntity Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of android-rest Show documentation

This is a library for android for Making RestFul Requests

There is a newer version: 2.9.0

package com.dg.libs.rest.entities;

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

import org.apache.http.entity.InputStreamEntity;

public class CountingInputStreamEntity extends InputStreamEntity {

  public interface UploadListener {

    public void onChange(long current);
  }

  private UploadListener listener;

  public CountingInputStreamEntity(final InputStream instream, final long length) {
    super(instream, length);
  }

  public void setUploadListener(final UploadListener listener) {
    this.listener = listener;
  }

  @Override
  public void writeTo(final OutputStream outstream) throws IOException {
    super.writeTo(new CountingOutputStream(outstream));
  }

  class CountingOutputStream extends OutputStream {

    private long counter = 0L;
    private final OutputStream outputStream;

    public CountingOutputStream(final OutputStream outputStream) {
      this.outputStream = outputStream;
    }

    @Override
    public void write(final byte[] buffer, final int offset, final int count)
        throws IOException {
      this.outputStream.write(buffer, offset, count);
      this.counter += count;
      if (listener != null) {
        listener.onChange(counter);
      }
    }

    @Override
    public void write(final int oneByte) throws IOException {
      this.outputStream.write(oneByte);
      counter++;
      if (listener != null) {
        // int percent = (int) ((counter * 100) / length);
        listener.onChange(counter);
      }
    }
  }

}