Mercurial > hg > blitz_condensed
diff src/com/go/trove/net/SocketConnector.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/SocketConnector.java Sat Mar 21 11:00:06 2009 +0000 @@ -0,0 +1,269 @@ +/* ==================================================================== + * 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.io.*; +import java.net.*; +import java.util.*; +import com.go.trove.util.*; + +/****************************************************************************** + * Allows client socket connections to be established with a timeout. + * + * @author Brian S O'Neill + * @version + * <!--$$Revision: 1.1 $-->, <!--$$JustDate:--> 00/12/05 <!-- $--> + */ +public class SocketConnector { + // Limit the number of threads that may simultaneously connect to a + // specific destination. + private static final int CONNECT_THREAD_MAX = 5; + + // Maps address:port pairs to ThreadPools for connecting. + private static Map mConnectors = + Collections.synchronizedMap(new SoftHashMap()); + + /** + * @param timeout Max time to wait for new connection. If negative, wait + * is infinite. + * @return null if couldn't connect in time. + */ + public static Socket connect(String host, int port, long timeout) + throws SocketException + { + return connect((Object)host, port, timeout); + } + + + /** + * @param timeout Max time to wait for new connection. If negative, wait + * is infinite. + * @return null if couldn't connect in time. + */ + public static Socket connect(InetAddress address, int port, long timeout) + throws SocketException + { + return connect((Object)address, port, timeout); + } + + /** + * @param address either a string or InetAddress. + * @param timeout Max time to wait for new connection. If negative, wait + * is infinite. + * @return null if couldn't connect in time + */ + private static Socket connect(Object address, int port, long timeout) + throws SocketException + { + Key key = new Key(address, port); + ThreadPool pool; + synchronized (mConnectors) { + pool = (ThreadPool)mConnectors.get(key); + if (pool == null) { + pool = new ThreadPool + ("SocketConnector[" + key + ']', CONNECT_THREAD_MAX); + pool.setIdleTimeout(10000); + mConnectors.put(key, pool); + } + } + + Connector connector = new Connector(key); + Thread thread; + + long start; + if (timeout > 0) { + start = System.currentTimeMillis(); + } + else { + start = 0; + } + + try { + thread = pool.start(connector, timeout); + } + catch (InterruptedException e) { + return null; + } + + if (timeout > 0) { + timeout = timeout - (System.currentTimeMillis() - start); + if (timeout < 0) { + timeout = 0; + } + } + + try { + Socket socket = connector.connect(timeout); + if (socket != null) { + return socket; + } + } + catch (InterruptedException e) { + } + + thread.interrupt(); + return null; + } + + private SocketConnector() { + } + + private static class Key { + final Object mAddress; + final int mPort; + + Key(Object address, int port) { + mAddress = address; + mPort = port; + } + + public boolean equals(Object obj) { + if (obj instanceof Key) { + Key key = (Key)obj; + return key.mAddress.equals(mAddress) && key.mPort == mPort; + } + return false; + } + + public int hashCode() { + return mAddress.hashCode() + mPort; + } + + public String toString() { + if (mAddress instanceof InetAddress) { + return ((InetAddress)mAddress).getHostAddress() + ':' + mPort; + } + else { + return String.valueOf(mAddress) + ':' + mPort; + } + } + } + + private static class Connector implements Runnable { + private final Key mKey; + private Object mSocketOrException; + private boolean mDoneWaiting; + + public Connector(Key key) { + mKey = key; + } + + public synchronized Socket connect(long timeout) + throws SocketException, InterruptedException + { + try { + if (mSocketOrException == null) { + if (timeout < 0) { + wait(); + } + else if (timeout > 0) { + wait(timeout); + } + else { + return null; + } + } + } + finally { + mDoneWaiting = true; + } + + if (mSocketOrException instanceof Socket) { + return (Socket)mSocketOrException; + } + else if (mSocketOrException instanceof InterruptedIOException) { + throw new InterruptedException(); + } + else if (mSocketOrException instanceof Exception) { + throw new SocketException + ("Unable to connect to " + mKey + ", " + + ((Exception)mSocketOrException).getMessage()); + } + + return null; + } + + public void run() { + try { + Socket socket; + Object address = mKey.mAddress; + if (address instanceof InetAddress) { + socket = new Socket((InetAddress)address, mKey.mPort); + } + else { + socket = new Socket(String.valueOf(address), mKey.mPort); + } + + synchronized (this) { + if (mDoneWaiting) { + try { + socket.close(); + } + catch (IOException e) { + } + } + else { + mSocketOrException = socket; + notify(); + } + } + } + catch (Exception e) { + synchronized (this) { + mSocketOrException = e; + notify(); + } + } + } + } +}