00001 /* 00002 * $Header: /home/cvs/jakarta-commons/fileupload/src/java/org/apache/commons/fileupload/DeferredFileOutputStream.java,v 1.2 2003/05/31 22:31:08 martinc Exp $ 00003 * $Revision: 1.2 $ 00004 * $Date: 2003/05/31 22:31:08 $ 00005 * 00006 * ==================================================================== 00007 * 00008 * The Apache Software License, Version 1.1 00009 * 00010 * Copyright (c) 2001-2003 The Apache Software Foundation. All rights 00011 * reserved. 00012 * 00013 * Redistribution and use in source and binary forms, with or without 00014 * modification, are permitted provided that the following conditions 00015 * are met: 00016 * 00017 * 1. Redistributions of source code must retain the above copyright 00018 * notice, this list of conditions and the following disclaimer. 00019 * 00020 * 2. Redistributions in binary form must reproduce the above copyright 00021 * notice, this list of conditions and the following disclaimer in 00022 * the documentation and/or other materials provided with the 00023 * distribution. 00024 * 00025 * 3. The end-user documentation included with the redistribution, if 00026 * any, must include the following acknowlegement: 00027 * "This product includes software developed by the 00028 * Apache Software Foundation (http://www.apache.org/)." 00029 * Alternately, this acknowlegement may appear in the software itself, 00030 * if and wherever such third-party acknowlegements normally appear. 00031 * 00032 * 4. The names "The Jakarta Project", "Commons", and "Apache Software 00033 * Foundation" must not be used to endorse or promote products derived 00034 * from this software without prior written permission. For written 00035 * permission, please contact apache@apache.org. 00036 * 00037 * 5. Products derived from this software may not be called "Apache" 00038 * nor may "Apache" appear in their names without prior written 00039 * permission of the Apache Group. 00040 * 00041 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED 00042 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 00043 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 00044 * DISCLAIMED. IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR 00045 * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 00046 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 00047 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF 00048 * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 00049 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, 00050 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT 00051 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 00052 * SUCH DAMAGE. 00053 * ==================================================================== 00054 * 00055 * This software consists of voluntary contributions made by many 00056 * individuals on behalf of the Apache Software Foundation. For more 00057 * information on the Apache Software Foundation, please see 00058 * <http://www.apache.org/>. 00059 * 00060 */ 00061 00062 00063 package org.apache.commons.fileupload; 00064 00065 import java.io.ByteArrayOutputStream; 00066 import java.io.File; 00067 import java.io.FileOutputStream; 00068 import java.io.IOException; 00069 import java.io.OutputStream; 00070 00081 public class DeferredFileOutputStream 00082 extends ThresholdingOutputStream 00083 { 00084 00085 // ----------------------------------------------------------- Data members 00086 00087 00092 private ByteArrayOutputStream memoryOutputStream; 00093 00094 00099 private FileOutputStream diskOutputStream; 00100 00101 00107 private OutputStream currentOutputStream; 00108 00109 00113 private File outputFile; 00114 00115 00116 // ----------------------------------------------------------- Constructors 00117 00118 00126 public DeferredFileOutputStream(int threshold, File outputFile) 00127 { 00128 super(threshold); 00129 this.outputFile = outputFile; 00130 00131 memoryOutputStream = new ByteArrayOutputStream(threshold); 00132 currentOutputStream = memoryOutputStream; 00133 } 00134 00135 00136 // --------------------------------------- ThresholdingOutputStream methods 00137 00138 00147 protected OutputStream getStream() throws IOException 00148 { 00149 return currentOutputStream; 00150 } 00151 00152 00161 protected void thresholdReached() throws IOException 00162 { 00163 byte[] data = memoryOutputStream.toByteArray(); 00164 FileOutputStream fos = new FileOutputStream(outputFile); 00165 fos.write(data); 00166 diskOutputStream = fos; 00167 currentOutputStream = fos; 00168 memoryOutputStream = null; 00169 } 00170 00171 00172 // --------------------------------------------------------- Public methods 00173 00174 00182 public boolean isInMemory() 00183 { 00184 return (!isThresholdExceeded()); 00185 } 00186 00187 00196 public byte[] getData() 00197 { 00198 if (memoryOutputStream != null) 00199 { 00200 return memoryOutputStream.toByteArray(); 00201 } 00202 return null; 00203 } 00204 00205 00214 public File getFile() 00215 { 00216 return outputFile; 00217 } 00218 }