Mercurial > hg > blitz_condensed
diff src/com/go/trove/net/LazySocket.java @ 0:3dc0c5604566
Initial checkin of blitz 2.0 fcs - no installer yet.
author | Dan Creswell <dan.creswell@gmail.com> |
---|---|
date | Sat, 21 Mar 2009 11:00:06 +0000 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/com/go/trove/net/LazySocket.java Sat Mar 21 11:00:06 2009 +0000 @@ -0,0 +1,518 @@ +/* ==================================================================== + * Trove - Copyright (c) 1997-2000 Walt Disney Internet Group + * ==================================================================== + * The Tea Software License, Version 1.1 + * + * Copyright (c) 2000 Walt Disney Internet Group. All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in + * the documentation and/or other materials provided with the + * distribution. + * + * 3. The end-user documentation included with the redistribution, + * if any, must include the following acknowledgment: + * "This product includes software developed by the + * Walt Disney Internet Group (http://opensource.go.com/)." + * Alternately, this acknowledgment may appear in the software itself, + * if and wherever such third-party acknowledgments normally appear. + * + * 4. The names "Tea", "TeaServlet", "Kettle", "Trove" and "BeanDoc" must + * not be used to endorse or promote products derived from this + * software without prior written permission. For written + * permission, please contact opensource@dig.com. + * + * 5. Products derived from this software may not be called "Tea", + * "TeaServlet", "Kettle" or "Trove", nor may "Tea", "TeaServlet", + * "Kettle", "Trove" or "BeanDoc" appear in their name, without prior + * written permission of the Walt Disney Internet Group. + * + * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED + * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES + * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE + * DISCLAIMED. IN NO EVENT SHALL THE WALT DISNEY INTERNET GROUP OR ITS + * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, + * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, + * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR + * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY + * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE + * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + * ==================================================================== + * + * For more information about Tea, please see http://opensource.go.com/. + */ + +package com.go.trove.net; + +import java.net.*; +import java.io.*; + +/****************************************************************************** + * A socket implementation that lazily establishs a connection. It only + * connects when actually needed. Setting options and getting I/O streams will + * not force a connection to be established. As soon as a read or write + * operation is performed, a connection is established. + * <p> + * If the first write operation requires a connection to be established, then a + * recycled connection is requested. The connection is tested by writing the + * data to it. If this fails, a new connection is requested and the operation + * is tried again. + * + * @author Brian S O'Neill + * @version + * <!--$$Revision: 1.1 $-->, <!--$$JustDate:--> 00/12/05 <!-- $--> + */ +class LazySocket extends Socket { + private final Impl mImpl; + + public LazySocket(SocketFactory factory) throws SocketException { + this(factory, null, factory.getDefaultTimeout()); + } + + public LazySocket(SocketFactory factory, Object session) + throws SocketException + { + this(factory, session, factory.getDefaultTimeout()); + } + + public LazySocket(SocketFactory factory, long timeout) + throws SocketException + { + this(factory, null, timeout); + } + + public LazySocket(SocketFactory factory, Object session, long timeout) + throws SocketException + { + this(new Impl(factory, session, timeout)); + } + + private LazySocket(Impl impl) throws SocketException { + super(impl); + mImpl = impl; + } + + /** + * Returns the internal wrapped socket or null if not connected. After + * calling recycle, this LazySocket instance is closed. + */ + CheckedSocket recycle() { + CheckedSocket s; + if (mImpl.mClosed) { + s = null; + } + else { + s = mImpl.mSocket; + mImpl.mSocket = null; + try { + mImpl.close(); + } + catch (IOException e) { + } + } + return s; + } + + private static class Impl extends SocketImpl { + private final SocketFactory mFactory; + private final Object mSession; + private final long mTimeout; + + private boolean mClosed; + private CheckedSocket mSocket; + + private Object[] mOptions; + + private InputStream mIn; + private OutputStream mOut; + + public Impl(SocketFactory factory, Object session, long timeout) { + mFactory = factory; + mSession = session; + mTimeout = timeout; + } + + public void setOption(int optId, Object value) throws SocketException { + int optionIndex; + + switch (optId) { + case TCP_NODELAY: + optionIndex = 0; + break; + case SO_LINGER: + optionIndex = 1; + break; + case SO_TIMEOUT: + optionIndex = 2; + break; + case SO_SNDBUF: + optionIndex = 3; + break; + case SO_RCVBUF: + optionIndex = 4; + break; + case SO_BINDADDR: + case SO_REUSEADDR: + case IP_MULTICAST_IF: + default: + throw new SocketException("Invalid option: " + optId); + } + + if (mOptions == null) { + mOptions = new Object[5]; + } + + mOptions[optionIndex] = value; + + if (mSocket == null) { + return; + } + + switch (optId) { + case TCP_NODELAY: + mSocket.setTcpNoDelay(((Boolean)value).booleanValue()); + break; + case SO_LINGER: + if (value instanceof Boolean) { + mSocket.setSoLinger(((Boolean)value).booleanValue(), 0); + } + else { + mSocket.setSoLinger(true, ((Integer)value).intValue()); + } + break; + case SO_TIMEOUT: + mSocket.setSoTimeout(((Integer)value).intValue()); + break; + case SO_SNDBUF: + mSocket.setSendBufferSize(((Integer)value).intValue()); + break; + case SO_RCVBUF: + mSocket.setReceiveBufferSize(((Integer)value).intValue()); + break; + } + } + + public Object getOption(int optId) throws SocketException { + Socket socket = createSocket(); + + switch (optId) { + case TCP_NODELAY: + return socket.getTcpNoDelay() ? Boolean.TRUE : Boolean.FALSE; + case SO_BINDADDR: + return socket.getLocalAddress(); + case SO_LINGER: + return new Integer(socket.getSoLinger()); + case SO_TIMEOUT: + return new Integer(socket.getSoTimeout()); + case SO_SNDBUF: + return new Integer(socket.getSendBufferSize()); + case SO_RCVBUF: + return new Integer(socket.getReceiveBufferSize()); + case SO_REUSEADDR: + case IP_MULTICAST_IF: + default: + throw new SocketException("Invalid option: " + optId); + } + } + + protected InputStream getInputStream() throws IOException { + if (mIn == null) { + mIn = new In(); + } + return mIn; + } + + protected OutputStream getOutputStream() throws IOException { + if (mOut == null) { + mOut = new Out(); + } + return mOut; + } + + protected int available() throws IOException { + return getInputStream().available(); + } + + protected void close() throws IOException { + if (!mClosed) { + mClosed = true; + if (mSocket != null) { + mSocket.close(); + } + } + } + + protected InetAddress getInetAddress() { + if (mSocket != null) { + return mSocket.getInetAddress(); + } + else { + return mFactory.getInetAddressAndPort(mSession).getInetAddress(); + } + } + + protected int getPort() { + if (mSocket != null) { + return mSocket.getPort(); + } + else { + return mFactory.getInetAddressAndPort(mSession).getPort(); + } + } + + protected int getLocalPort() { + if (mSocket != null) { + return mSocket.getLocalPort(); + } + else { + return -1; + } + } + + Socket createSocket() throws SocketException { + if (mSocket != null) { + return mSocket; + } + + if (mClosed) { + throw new SocketException("Socket is closed"); + } + + mSocket = mFactory.createSocket(mSession, mTimeout); + applyOptions(); + + return mSocket; + } + + Socket getSocket(byte[] data, int off, int len) + throws SocketException + { + if (mSocket != null) { + return mSocket; + } + + if (mClosed) { + throw new SocketException("Socket is closed"); + } + + long timeout = mTimeout; + long start; + if (timeout > 0) { + start = System.currentTimeMillis(); + } + else { + start = 0; + } + + try { + mSocket = mFactory.getSocket(mSession, timeout); + applyOptions(); + OutputStream out = mSocket.getOutputStream(); + out.write(data, off, len); + out.flush(); + } + catch (Exception e) { + if (mSocket != null) { + try { + mSocket.close(); + } + catch (Exception e2) { + } + } + + if (timeout > 0) { + timeout = timeout - (System.currentTimeMillis() - start); + if (timeout < 0) { + timeout = 0; + } + } + + mSocket = mFactory.createSocket(mSession, timeout); + applyOptions(); + try { + OutputStream out = mSocket.getOutputStream(); + out.write(data, off, len); + out.flush(); + } + catch (IOException e2) { + throw new SocketException(e2.getMessage()); + } + } + + return mSocket; + } + + private void applyOptions() throws SocketException { + if (mOptions == null || mSocket == null) { + return; + } + + Object[] options = mOptions; + Object value; + + if ((value = options[0]) != null) { + mSocket.setTcpNoDelay(((Boolean)value).booleanValue()); + } + if ((value = options[1]) != null) { + if (value instanceof Boolean) { + mSocket.setSoLinger(((Boolean)value).booleanValue(), 0); + } + else { + mSocket.setSoLinger(true, ((Integer)value).intValue()); + } + } + if ((value = options[2]) != null) { + mSocket.setSoTimeout(((Integer)value).intValue()); + } + if ((value = options[3]) != null) { + mSocket.setSendBufferSize(((Integer)value).intValue()); + } + if ((value = options[4]) != null) { + mSocket.setReceiveBufferSize(((Integer)value).intValue()); + } + } + + protected void create(boolean stream) throws IOException { + error(); + } + + protected void connect(String host, int port) throws IOException { + error(); + } + + protected void connect(InetAddress host, int port) throws IOException { + error(); + } + + protected void connect(SocketAddress host, + int port) throws IOException { + error(); + } + + protected void bind(InetAddress host, int port) throws IOException { + error(); + } + + protected void listen(int backlog) throws IOException { + error(); + } + + protected void accept(SocketImpl s) throws IOException { + error(); + } + + protected void sendUrgentData(int aByte) throws IOException { + throw new UnsupportedOperationException(); + } + + private void error() throws IOException { + throw new IOException("Unsupported operation"); + } + + private class In extends InputStream { + private InputStream mStream; + + public int read() throws IOException { + return getStream().read(); + } + + public int read(byte[] b) throws IOException { + return getStream().read(b); + } + + public int read(byte[] b, int off, int len) throws IOException { + return getStream().read(b, off, len); + } + + public long skip(long n) throws IOException { + return getStream().skip(n); + } + + public int available() throws IOException { + return getStream().available(); + } + + public void close() throws IOException { + if (mStream != null) { + mStream.close(); + } + Impl.this.close(); + } + + public void mark(int readlimit) { + try { + getStream().mark(readlimit); + } + catch (IOException e) { + } + } + + public void reset() throws IOException { + if (mStream == null) { + throw new IOException("Stream not marked"); + } + else { + mStream.reset(); + } + } + + public boolean markSupported() { + try { + return getStream().markSupported(); + } + catch (IOException e) { + return false; + } + } + + private InputStream getStream() throws IOException { + if (mStream == null) { + mStream = createSocket().getInputStream(); + } + return mStream; + } + } + + private class Out extends OutputStream { + private OutputStream mStream; + + public void write(int b) throws IOException { + write(new byte[] {(byte)b}, 0, 1); + } + + public void write(byte[] b) throws IOException { + write(b, 0, b.length); + } + + public void write(byte[] b, int off, int len) throws IOException { + if (mStream == null) { + mStream = getSocket(b, off, len).getOutputStream(); + } + else { + mStream.write(b, off, len); + } + } + + public void flush() throws IOException { + if (mStream != null) { + mStream.flush(); + } + } + + public void close() throws IOException { + if (mStream != null) { + mStream.close(); + } + Impl.this.close(); + } + } + } +}