001 /* 002 * Copyright (C) 2003-2009 eXo Platform SAS. 003 * 004 * This is free software; you can redistribute it and/or modify it 005 * under the terms of the GNU Lesser General Public License as 006 * published by the Free Software Foundation; either version 2.1 of 007 * the License, or (at your option) any later version. 008 * 009 * This software is distributed in the hope that it will be useful, 010 * but WITHOUT ANY WARRANTY; without even the implied warranty of 011 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 012 * Lesser General Public License for more details. 013 * 014 * You should have received a copy of the GNU Lesser General Public 015 * License along with this software; if not, write to the Free 016 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 017 * 02110-1301 USA, or see the FSF site: http://www.fsf.org. 018 */ 019 package org.crsh.util; 020 021 import java.io.ByteArrayOutputStream; 022 import java.io.IOException; 023 import java.io.InputStream; 024 025 /** 026 * Extends the {@link java.io.ByteArrayOutputStream} for providing an input stream over the 027 * buffer. 028 * 029 * @author <a href="mailto:julien.viet@exoplatform.com">Julien Viet</a> 030 * @version $Revision$ 031 */ 032 public class BytesOutputStream extends ByteArrayOutputStream { 033 034 /** . */ 035 private boolean closed; 036 037 public BytesOutputStream() { 038 } 039 040 public BytesOutputStream(int size) { 041 super(size); 042 } 043 044 public InputStream getInputStream() { 045 if (!closed) { 046 throw new IllegalStateException("Not yet closed"); 047 } 048 return new InputStream() { 049 050 /** . */ 051 int read = 0; 052 053 @Override 054 public int read() throws IOException { 055 if (read < count) { 056 return buf[read++]; 057 } else { 058 return -1; 059 } 060 } 061 }; 062 } 063 064 @Override 065 public void close() throws IOException { 066 super.close(); 067 068 // 069 closed = true; 070 } 071 }