com.files.util.StreamTransfer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of files-sdk Show documentation

The Files.com Java client library provides convenient access to the Files.com API from JVM based applications.

There is a newer version: 1.4.123

package com.files.util;

import com.files.util.BufferPool.Buffer;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.function.IntConsumer;

public class StreamTransfer {
  private static final int INITIAL_BUFFER_SIZE = Math.min(1024 * 8, BufferPool.T_BUFFER_SIZE);

  private StreamTransfer() {
  }

  public static long smartCopy(long knownSize, InputStream in, OutputStream out) throws IOException {
    return smartCopy(knownSize, in, out, null);
  }

  public static long smartCopy(long knownSize, InputStream in, OutputStream out,
      IntConsumer progressConsumer) throws IOException {
    // in the future full duplex and other copies may be possible
    return bufferedCopy(knownSize, in, out, progressConsumer);
  }

  public static long bufferedCopy(long knownSize, InputStream in, OutputStream out,
      IntConsumer progressConsumer) throws IOException {
    long total = 0;
    try (Buffer b = BufferPool.needBuffer(INITIAL_BUFFER_SIZE)) {
      int c;
      byte[] buffer = b.getBuffer();
      long remaining = Long.MAX_VALUE;
      while ((knownSize < 1 || (remaining = knownSize - total) > 0)
          && (c = in.read(buffer, 0, (int) Math.min(remaining, INITIAL_BUFFER_SIZE))) >= 0) {
        if (c > 0) {
          out.write(buffer, 0, c);
          total += c;
          if (progressConsumer != null) {
            progressConsumer.accept(c);
          }
        } else {
          Thread.yield();
        }
      }
    }
    return total;
  }
}