edu.isi.nlp.io.ByteArraySink Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of common-core-open Show documentation

The newest version!

package edu.isi.nlp.io;

import com.google.common.io.ByteSink;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;

/**
 * A {@link com.google.common.io.ByteSink} which writes to a byte array. The contents of this array
 * can be recovered by calling toByteArray().
 *
 * Before anything is written to this {@code ByteSink}, the wrapped array is empty.
 */
public class ByteArraySink extends ByteSink {

  private final ByteArrayOutputStream stream = new ByteArrayOutputStream();

  private ByteArraySink() {}

  public static ByteArraySink create() {
    return new ByteArraySink();
  }

  @Override
  public OutputStream openStream() throws IOException {
    stream.reset();
    return stream;
  }

  public byte[] toByteArray() {
    return stream.toByteArray();
  }
}