/* * RapidMiner * * Copyright (C) 2001-2011 by Rapid-I and the contributors * * Complete list of developers available at our web site: * * http://rapid-i.com * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU Affero General Public License for more details. * * You should have received a copy of the GNU Affero General Public License * along with this program. If not, see http://www.gnu.org/licenses/. */ package com.rapidminer.operator.nio.file; import java.io.ByteArrayInputStream; import java.io.File; import java.io.FileOutputStream; import java.io.IOException; import java.nio.Buffer; import com.rapidminer.operator.OperatorException; /** * Simple implementation of a {@link FileObject} backed by a {@link Buffer}. * * @author Nils Woehler * */ public class BufferedFileObject extends FileObject { private static final long serialVersionUID = 1L; private byte[] buffer; private File file = null; public BufferedFileObject(byte[] buffer) { super(); this.buffer = buffer; } @Override public ByteArrayInputStream openStream() throws OperatorException { return new ByteArrayInputStream(buffer); } @Override public File getFile() throws OperatorException { if(file==null){ try { file = File.createTempFile("rm_file_", ".dump"); FileOutputStream fos = new FileOutputStream(file); fos.write(this.buffer); fos.close(); file.deleteOnExit(); } catch (IOException e) { throw new OperatorException("303", e, file, e.getMessage()); } return file; } else { return file; } } @Override public String toString() { return (file != null) ? ("Buffered file stored in temporary file: " + file.getAbsolutePath()) : ("Memory buffered file"); } @Override protected void finalize() throws Throwable { if(file!=null){ file.delete(); } super.finalize(); } }