nl.nn.testtool.echo2.FileDownloadProvider Maven / Gradle / Ivy

Go to download
/*
 * Created on 14-Feb-08
 *
 * To change the template for this generated file go to
 * Window>Preferences>Java>Code Generation>Code and Comments
 */
package nl.nn.testtool.echo2;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.ObjectOutputStream;
import java.io.OutputStream;
import java.util.zip.GZIPOutputStream;

import nextapp.echo2.app.filetransfer.DownloadProvider;
import nl.nn.testtool.util.LogUtil;

import org.apache.log4j.Logger;

/**
 * @author m00f069
 *
 * To change the template for this generated type comment go to
 * Window>Preferences>Java>Code Generation>Code and Comments
 */
public class FileDownloadProvider implements DownloadProvider {
	private Logger log = LogUtil.getLogger(this);
	private File file;
	private String fileName;

	public FileDownloadProvider(File file, String fileName) {
		this.file = file;
		this.fileName = fileName;
	}

	public String getContentType() {
		return "application/octet-stream";
	}

	public String getFileName() {
		return fileName;
	}

	public int getSize() {
		return (int)file.length();
	}

	public void writeFile(OutputStream outputStream) throws IOException {
		log.debug("Start download for file: " + fileName);
		FileInputStream fileInputStream = null;
		try {
			fileInputStream = new FileInputStream(file);
			byte[] buffer = new byte[1024];
			int i = fileInputStream.read(buffer);
			while (i != -1) {
				outputStream.write(buffer, 0, i);
				i = fileInputStream.read(buffer);
			}
		} catch(IOException e) {
			// Log exception.
			log.error("IOException during download", e);
			// Throw exception as it is allowed by the method signature.
			throw e;
		} finally {
			try {
				outputStream.close();
				fileInputStream.close();
				file.delete();
			} catch(IOException e) {
				// Log exception.
				log.error("IOException closing streams", e);
				// Throw exception as it is allowed by the method signature.
				throw e;
			}
		}
	}

}