org.unlaxer.util.copy.MultipleOutputStream Maven / Gradle / Ivy

Go to download
package org.unlaxer.util.copy;

import java.io.Closeable;
import java.io.IOException;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.List;
import java.util.Optional;
import java.util.function.Consumer;

public class MultipleOutputStream extends OutputStream implements Closeable{

	private Collection outputs;

	public MultipleOutputStream(Collection outputs) {

		this.outputs = outputs;
	}
	
	public MultipleOutputStream(OutputStream... outputs) {

		this.outputs = Arrays.asList(outputs);
	}


	@Override
	public void write(int value) throws IOException {

		Optional process = process(out->{
			try {
				out.write(value);
			} catch (Exception e) {
				throw new RuntimeException(e);
			}
		});
		if(process.isPresent()){throw process.get();}
	}

	@Override
	public void write(byte[] value) throws IOException {
		
		Optional process = process(out->{
			try {
				out.write(value);
			} catch (Exception e) {
				throw new RuntimeException(e);
			}
		});
		if(process.isPresent()){throw process.get();}
	}

	@Override
	public void write(byte[] b, int off, int len) throws IOException {

		Optional process = process(out->{
			try {
				out.write(b, off, len);
			} catch (Exception e) {
				throw new RuntimeException(e);
			}
		});
		if(process.isPresent()){throw process.get();}

	}

	@Override
	public void close() throws IOException {

		Optional process = process(out->{
			try {
				out.close();
			} catch (Exception e) {
				throw new RuntimeException(e);
			}
		});
		if(process.isPresent()){throw process.get();}
	}

	@Override
	public void flush() throws IOException {

		Optional process = process(out->{
			try {
				out.flush();
			} catch (Exception e) {
				throw new RuntimeException(e);
			}
		});
		if(process.isPresent()){throw process.get();}
	}
	
	public Optional process(Consumer outputConsumer) {

		List exceptions = new ArrayList();
		for (OutputStream current : outputs){
			try{
				outputConsumer.accept(current);
			}catch(Exception e){
				exceptions.add(e);
			}
		}
		return exceptions.isEmpty() ?
				Optional.empty():
				Optional.of(new MultipleIOException(exceptions));
	}

}