net.grinder.testutility.StreamCopier Maven / Gradle / Ivy

Go to download
// Copyright (C) 2000 Phil Dawes
// Copyright (C) 2000 - 2011 Philip Aston
// All rights reserved.
//
// This file is part of The Grinder software distribution. Refer to
// the file LICENSE which is part of The Grinder distribution for
// licensing details. The Grinder distribution is available on the
// Internet at http://grinder.sourceforge.net/
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
// FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
// COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
// INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
// (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
// SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
// HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
// STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
// OF THE POSSIBILITY OF SUCH DAMAGE.

package net.grinder.testutility;

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


/**
 * Class that copies from {@code InputStream}s to
 * {@code OutputStream}s.
 *
 * @author Philip Aston
 */
public class StreamCopier {

  private final byte[] m_buffer;

  /**
   * Constructor.
   *
   * @param bufferSize The buffer size.
   */
  public StreamCopier(int bufferSize) {
    m_buffer = new byte[bufferSize];
  }

  public StreamCopier() {
    this(4096);
  }

  /**
   * Copies from the input stream to the output stream until the input
   * stream is empty or one of the streams reports an error.
   *
   * Not thread safe - use multiple {@code StreamCopier}
   * instances instead.
   *
   * @param in Input stream.
   * @param out Output stream.
   * @throws IOException If an IO problem occurred during the copy.
   */
  public void copy(InputStream in, OutputStream out) throws IOException {

    try {
      while (true) {
        final int bytesRead = in.read(m_buffer, 0, m_buffer.length);

        if (bytesRead == -1) {
          break;
        }

        out.write(m_buffer, 0, bytesRead);
        out.flush();
      }
    }
    finally {
      out.close();
      in.close();
    }
  }
}