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: *
023: * Free Software Foundation, Inc.
024: * 59 Temple Place, Suite 330
025: * Boston, MA 02111-1307 USA
026: *
027: * @author Scott Ferguson
028: */
029:
030: package com.caucho.server.hmux;
031:
032: import com.caucho.log.Log;
033: import com.caucho.vfs.StreamImpl;
034:
035: import java.io.IOException;
036: import java.net.SocketException;
037: import java.util.Iterator;
038: import java.util.logging.Logger;
039:
040: /**
041: * Facade to HmuxStream to properly handle the close.
042: */
043: class HmuxStreamWrapper extends StreamImpl {
044: private static final Logger log = Log.open(HmuxStream.class);
045:
046: private HmuxStream _stream;
047:
048: /**
049: * Create a new HMUX stream.
050: */
051: HmuxStreamWrapper(HmuxStream stream) throws IOException {
052: _stream = stream;
053: }
054:
055: /**
056: * Set if this should be an SSL connection.
057: */
058: public void setSSL(boolean isSSL) {
059: _stream.setSSL(isSSL);
060: }
061:
062: /**
063: * Set if this should be an SSL connection.
064: */
065: public boolean isSSL() {
066: return _stream.isSSL();
067: }
068:
069: /**
070: * Sets the method
071: */
072: public void setMethod(String method) {
073: _stream.setMethod(method);
074: }
075:
076: /**
077: * Sets true if we're only interested in the head.
078: */
079: public void setHead(boolean isHead) {
080: _stream.setHead(isHead);
081: }
082:
083: /**
084: * Returns the stream's host.
085: */
086: public String getHost() {
087: return _stream.getHost();
088: }
089:
090: /**
091: * Returns the stream's port.
092: */
093: public int getPort() {
094: return _stream.getPort();
095: }
096:
097: /**
098: * Returns a header from the response returned from the HMUX server.
099: *
100: * @param name name of the header
101: * @return the header value.
102: */
103: public Object getAttribute(String name) throws IOException {
104: if (_stream != null)
105: return _stream.getAttribute(name);
106: else
107: return null;
108: }
109:
110: /**
111: * Returns an iterator of the returned header names.
112: */
113: public Iterator getAttributeNames() throws IOException {
114: if (_stream != null)
115: return _stream.getAttributeNames();
116: else
117: return null;
118: }
119:
120: /**
121: * Sets a header for the request.
122: */
123: public void setAttribute(String name, Object value) {
124: if (_stream != null)
125: _stream.setAttribute(name, value);
126: }
127:
128: /**
129: * Remove a header for the request.
130: */
131: public void removeAttribute(String name) {
132: if (_stream != null)
133: _stream.removeAttribute(name);
134: }
135:
136: /**
137: * Sets the timeout.
138: */
139: public void setSocketTimeout(long timeout) throws SocketException {
140: if (_stream != null)
141: _stream.setSocketTimeout(timeout);
142: }
143:
144: /**
145: * The stream is always writable (?)
146: */
147: public boolean canWrite() {
148: if (_stream != null)
149: return _stream.canWrite();
150: else
151: return false;
152: }
153:
154: /**
155: * Writes a buffer to the underlying stream.
156: *
157: * @param buffer the byte array to write.
158: * @param offset the offset into the byte array.
159: * @param length the number of bytes to write.
160: * @param isEnd true when the write is flushing a close.
161: */
162: public void write(byte[] buf, int offset, int length, boolean isEnd)
163: throws IOException {
164: if (_stream != null)
165: _stream.write(buf, offset, length, isEnd);
166: }
167:
168: /**
169: * The stream is readable.
170: */
171: public boolean canRead() {
172: if (_stream != null)
173: return _stream.canRead();
174: else
175: return false;
176: }
177:
178: /**
179: * Read data from the connection. If the request hasn't yet been sent
180: * to the server, send it.
181: */
182: public int read(byte[] buf, int offset, int length)
183: throws IOException {
184: if (_stream != null)
185: return _stream.read(buf, offset, length);
186: else
187: return -1;
188: }
189:
190: /**
191: * Returns the bytes still available.
192: */
193: public int getAvailable() throws IOException {
194: if (_stream != null)
195: return _stream.getAvailable();
196: else
197: return -1;
198: }
199:
200: /**
201: * Close the connection.
202: */
203: public void close() throws IOException {
204: HmuxStream stream = _stream;
205: _stream = null;
206:
207: if (stream != null)
208: stream.close();
209: }
210: }
|