001: /*
002: * Copyright (c) 1998-2008 Caucho Technology -- all rights reserved
003: *
004: * This file is part of Resin(R) Open Source
005: *
006: * Each copy or derived work must preserve the copyright notice and this
007: * notice unmodified.
008: *
009: * Resin Open Source is free software; you can redistribute it and/or modify
010: * it under the terms of the GNU General Public License as published by
011: * the Free Software Foundation; either version 2 of the License, or
012: * (at your option) any later version.
013: *
014: * Resin Open Source is distributed in the hope that it will be useful,
015: * but WITHOUT ANY WARRANTY; without even the implied warranty of
016: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE, or any warranty
017: * of NON-INFRINGEMENT. See the GNU General Public License for more
018: * details.
019: *
020: * You should have received a copy of the GNU General Public License
021: * along with Resin Open Source; if not, write to the
022: * Free SoftwareFoundation, Inc.
023: * 59 Temple Place, Suite 330
024: * Boston, MA 02111-1307 USA
025: *
026: * @author Scott Ferguson
027: */
028:
029: package com.caucho.vfs;
030:
031: import java.io.IOException;
032: import java.io.InputStream;
033: import java.io.OutputStream;
034:
035: /**
036: * Stream encapsulating InputStream/OutputStream.
037: */
038: public class VfsStream extends StreamImpl {
039: private static byte[] unixNewline = new byte[] { (byte) '\n' };
040:
041: private InputStream is;
042: private OutputStream _os;
043: private boolean flushOnNewline;
044: private boolean closeChildOnClose = true;
045: private byte[] newline = unixNewline;
046:
047: private long position;
048:
049: /**
050: * Create an empty VfsStream.
051: */
052: public VfsStream() {
053: }
054:
055: /**
056: * Create a new VfsStream based on the java.io.* stream.
057: */
058: public VfsStream(InputStream is, OutputStream os) {
059: init(is, os);
060: }
061:
062: public VfsStream(InputStream is, OutputStream os, Path path) {
063: init(is, os);
064: setPath(path);
065: }
066:
067: /**
068: * Initializes a VfsStream with an input/output stream pair. Before a
069: * read, the output will be flushed to avoid deadlocks.
070: *
071: * @param is the underlying InputStream.
072: * @param os the underlying OutputStream.
073: */
074: public void init(InputStream is, OutputStream os) {
075: this .is = is;
076: _os = os;
077: setPath(null);
078: flushOnNewline = false;
079: closeChildOnClose = true;
080: position = 0;
081: }
082:
083: public void setNewline(byte[] newline) {
084: this .newline = newline;
085: }
086:
087: public byte[] getNewline() {
088: return newline;
089: }
090:
091: public static ReadWritePair openReadWrite(InputStream is,
092: OutputStream os) {
093: VfsStream s = new VfsStream(is, os);
094: WriteStream writeStream = new WriteStream(s);
095: ReadStream readStream = new ReadStream(s, writeStream);
096: return new ReadWritePair(readStream, writeStream);
097: }
098:
099: /**
100: * Opens a read stream based on a java.io.InputStream.
101: *
102: * @param is the underlying InputStream.
103: *
104: * @return the new ReadStream
105: */
106: public static ReadStream openRead(InputStream is) {
107: VfsStream s = new VfsStream(is, null);
108: return new ReadStream(s);
109: }
110:
111: public static ReadStream openRead(InputStream is, WriteStream ws) {
112: VfsStream s = new VfsStream(is, null);
113: return new ReadStream(s, ws);
114: }
115:
116: public static WriteStream openWrite(OutputStream os) {
117: VfsStream s = new VfsStream(null, os);
118: return new WriteStream(s);
119: }
120:
121: public boolean canRead() {
122: return is != null;
123: }
124:
125: public int read(byte[] buf, int offset, int length)
126: throws IOException {
127: if (is == null)
128: return -1;
129:
130: int len = is.read(buf, offset, length);
131:
132: if (len > 0)
133: position += len;
134:
135: return len;
136: }
137:
138: public boolean hasSkip() {
139: return true;
140: }
141:
142: public long skip(long n) throws IOException {
143: return is.skip(n);
144: }
145:
146: public int getAvailable() throws IOException {
147: if (is == null)
148: return -1;
149: else
150: return is.available();
151: }
152:
153: public long getReadPosition() {
154: return position;
155: }
156:
157: public boolean canWrite() {
158: return _os != null;
159: }
160:
161: public boolean getFlushOnNewline() {
162: return flushOnNewline;
163: }
164:
165: public void setFlushOnNewline(boolean value) {
166: flushOnNewline = value;
167: }
168:
169: /**
170: * Writes a buffer to the underlying stream.
171: *
172: * @param buffer the byte array to write.
173: * @param offset the offset into the byte array.
174: * @param length the number of bytes to write.
175: * @param isEnd true when the write is flushing a close.
176: */
177: public void write(byte[] buf, int offset, int length, boolean isEnd)
178: throws IOException {
179: OutputStream os = _os;
180:
181: if (os != null) {
182: os.write(buf, offset, length);
183: }
184: }
185:
186: public void flushToDisk() throws IOException {
187: flush();
188: }
189:
190: public void flush() throws IOException {
191: if (_os != null) {
192: _os.flush();
193: }
194: }
195:
196: public void setCloseChildOnClose(boolean close) {
197: closeChildOnClose = close;
198: }
199:
200: public void close() throws IOException {
201: try {
202: if (_os != null && closeChildOnClose) {
203: _os.close();
204: _os = null;
205: }
206: } finally {
207: if (is != null && closeChildOnClose) {
208: is.close();
209: is = null;
210: }
211: }
212: }
213: }
|