com.firefly.utils.io.FilePipedStream Maven / Gradle / Ivy

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

There is a newer version: 5.0.2

package com.firefly.utils.io;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.UUID;

public class FilePipedStream implements PipedStream {
	
	private OutputStream out;
	private InputStream in;
	private File temp;
	
	public FilePipedStream(String tempdir) {
		temp = new File(tempdir, UUID.randomUUID().toString());
	}
	
	public FilePipedStream(File file) {
		temp = file;
	}

	@Override
	public void close() throws IOException {
		if(temp == null)
			return;
		
		try {
			temp.delete();
		} finally {
			if(in != null)
				in.close();
			
			if(out != null)
				out.close();
		}

		in = null;
		out = null;
		temp = null;
	}

	@Override
	public InputStream getInputStream() throws IOException{
		if(in == null) {
			in = new BufferedInputStream(new FileInputStream(temp));
		}
		return in;
	}

	@Override
	public OutputStream getOutputStream() throws IOException {
		if(out == null) {
			out = new BufferedOutputStream(new FileOutputStream(temp));
		}
		return out;
	}
}