se.l4.commons.io.BytesBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of commons-main Show documentation
The newest version!
package se.l4.commons.io;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;

public class BytesBuilder
{
	private final ByteArrayOutputStream out;
	
	public BytesBuilder()
	{
		out = new ByteArrayOutputStream(8192);
	}
	
	public BytesBuilder addChunk(byte[] buffer)
	{
		return addChunk(buffer, 0, buffer.length);
	}
	
	public BytesBuilder addChunk(byte[] buffer, int off, int len)
	{
		out.write(buffer, off, len);
		return this;
	}
	
	public Bytes build()
	{
		return Bytes.create(out.toByteArray());
	}
	
	static Bytes createViaLazyDataOutput(IoConsumer creator)
	{
		return createViaLazyDataOutput(creator, 8192);
	}
	
	static Bytes createViaLazyDataOutput(IoConsumer creator, int expectedSize)
	{
		return new DataOutputBytes(creator, expectedSize);
	}
	
	static Bytes createViaDataOutput(IoConsumer creator)
			throws IOException
	{
		return createViaDataOutput(creator, 8192);
	}
	
	static Bytes createViaDataOutput(IoConsumer creator, int expectedSize)
		throws IOException
	{
		ByteArrayOutputStream out = new ByteArrayOutputStream(expectedSize);
		try(ExtendedDataOutput dataOut = new ExtendedDataOutputStream(out))
		{
			creator.accept(dataOut);
		}
		return Bytes.create(out.toByteArray());
	}
	
	private static class DataOutputBytes
		implements Bytes
	{
		private final IoConsumer creator;
		private final int expectedSize;

		public DataOutputBytes(IoConsumer creator, int expectedSize)
		{
			this.creator = creator;
			this.expectedSize = expectedSize;
		}
		
		@Override
		public InputStream asInputStream()
			throws IOException
		{
			return new ByteArrayInputStream(toByteArray());
		}
		
		@Override
		public byte[] toByteArray()
			throws IOException
		{
			ByteArrayOutputStream out = new ByteArrayOutputStream(expectedSize);
			try(ExtendedDataOutput dataOut = new ExtendedDataOutputStream(out))
			{
				creator.accept(dataOut);
			}
			return out.toByteArray();
		}
		
		@Override
		public void asChunks(ByteArrayConsumer consumer)
			throws IOException
		{
			try(ExtendedDataOutput dataOut = new ExtendedDataOutputStream(new ChunkOutputStream(4096, consumer)))
			{
				creator.accept(dataOut);
			}
		}
	}
}