Mercurial > hg > blitz_stable

comparison src/com/go/trove/net/SocketConnector.java @ 0:3dc0c5604566

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
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
comparison
equal deleted inserted replaced
-1:000000000000 0:3dc0c5604566
1 /* ====================================================================
2 * Trove - Copyright (c) 1997-2000 Walt Disney Internet Group
3 * ====================================================================
4 * The Tea Software License, Version 1.1
5 *
6 * Copyright (c) 2000 Walt Disney Internet Group. All rights reserved.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 *
12 * 1. Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
14 *
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in
17 * the documentation and/or other materials provided with the
18 * distribution.
19 *
20 * 3. The end-user documentation included with the redistribution,
21 * if any, must include the following acknowledgment:
22 * "This product includes software developed by the
23 * Walt Disney Internet Group (http://opensource.go.com/)."
24 * Alternately, this acknowledgment may appear in the software itself,
25 * if and wherever such third-party acknowledgments normally appear.
26 *
27 * 4. The names "Tea", "TeaServlet", "Kettle", "Trove" and "BeanDoc" must
28 * not be used to endorse or promote products derived from this
29 * software without prior written permission. For written
30 * permission, please contact opensource@dig.com.
31 *
32 * 5. Products derived from this software may not be called "Tea",
33 * "TeaServlet", "Kettle" or "Trove", nor may "Tea", "TeaServlet",
34 * "Kettle", "Trove" or "BeanDoc" appear in their name, without prior
35 * written permission of the Walt Disney Internet Group.
36 *
37 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
38 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
39 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
40 * DISCLAIMED. IN NO EVENT SHALL THE WALT DISNEY INTERNET GROUP OR ITS
41 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
42 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
43 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
44 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
45 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
46 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
47 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
48 * ====================================================================
49 *
50 * For more information about Tea, please see http://opensource.go.com/.
51 */
52
53 package com.go.trove.net;
54
55 import java.io.*;
56 import java.net.*;
57 import java.util.*;
58 import com.go.trove.util.*;
59
60 /******************************************************************************
61 * Allows client socket connections to be established with a timeout.
62 *
63 * @author Brian S O'Neill
64 * @version
65 * <!--$$Revision: 1.1 $-->, <!--$$JustDate:--> 00/12/05 <!-- $-->
66 */
67 public class SocketConnector {
68 // Limit the number of threads that may simultaneously connect to a
69 // specific destination.
70 private static final int CONNECT_THREAD_MAX = 5;
71
72 // Maps address:port pairs to ThreadPools for connecting.
73 private static Map mConnectors =
74 Collections.synchronizedMap(new SoftHashMap());
75
76 /**
77 * @param timeout Max time to wait for new connection. If negative, wait
78 * is infinite.
79 * @return null if couldn't connect in time.
80 */
81 public static Socket connect(String host, int port, long timeout)
82 throws SocketException
83 {
84 return connect((Object)host, port, timeout);
85 }
86
87
88 /**
89 * @param timeout Max time to wait for new connection. If negative, wait
90 * is infinite.
91 * @return null if couldn't connect in time.
92 */
93 public static Socket connect(InetAddress address, int port, long timeout)
94 throws SocketException
95 {
96 return connect((Object)address, port, timeout);
97 }
98
99 /**
100 * @param address either a string or InetAddress.
101 * @param timeout Max time to wait for new connection. If negative, wait
102 * is infinite.
103 * @return null if couldn't connect in time
104 */
105 private static Socket connect(Object address, int port, long timeout)
106 throws SocketException
107 {
108 Key key = new Key(address, port);
109 ThreadPool pool;
110 synchronized (mConnectors) {
111 pool = (ThreadPool)mConnectors.get(key);
112 if (pool == null) {
113 pool = new ThreadPool
114 ("SocketConnector[" + key + ']', CONNECT_THREAD_MAX);
115 pool.setIdleTimeout(10000);
116 mConnectors.put(key, pool);
117 }
118 }
119
120 Connector connector = new Connector(key);
121 Thread thread;
122
123 long start;
124 if (timeout > 0) {
125 start = System.currentTimeMillis();
126 }
127 else {
128 start = 0;
129 }
130
131 try {
132 thread = pool.start(connector, timeout);
133 }
134 catch (InterruptedException e) {
135 return null;
136 }
137
138 if (timeout > 0) {
139 timeout = timeout - (System.currentTimeMillis() - start);
140 if (timeout < 0) {
141 timeout = 0;
142 }
143 }
144
145 try {
146 Socket socket = connector.connect(timeout);
147 if (socket != null) {
148 return socket;
149 }
150 }
151 catch (InterruptedException e) {
152 }
153
154 thread.interrupt();
155 return null;
156 }
157
158 private SocketConnector() {
159 }
160
161 private static class Key {
162 final Object mAddress;
163 final int mPort;
164
165 Key(Object address, int port) {
166 mAddress = address;
167 mPort = port;
168 }
169
170 public boolean equals(Object obj) {
171 if (obj instanceof Key) {
172 Key key = (Key)obj;
173 return key.mAddress.equals(mAddress) && key.mPort == mPort;
174 }
175 return false;
176 }
177
178 public int hashCode() {
179 return mAddress.hashCode() + mPort;
180 }
181
182 public String toString() {
183 if (mAddress instanceof InetAddress) {
184 return ((InetAddress)mAddress).getHostAddress() + ':' + mPort;
185 }
186 else {
187 return String.valueOf(mAddress) + ':' + mPort;
188 }
189 }
190 }
191
192 private static class Connector implements Runnable {
193 private final Key mKey;
194 private Object mSocketOrException;
195 private boolean mDoneWaiting;
196
197 public Connector(Key key) {
198 mKey = key;
199 }
200
201 public synchronized Socket connect(long timeout)
202 throws SocketException, InterruptedException
203 {
204 try {
205 if (mSocketOrException == null) {
206 if (timeout < 0) {
207 wait();
208 }
209 else if (timeout > 0) {
210 wait(timeout);
211 }
212 else {
213 return null;
214 }
215 }
216 }
217 finally {
218 mDoneWaiting = true;
219 }
220
221 if (mSocketOrException instanceof Socket) {
222 return (Socket)mSocketOrException;
223 }
224 else if (mSocketOrException instanceof InterruptedIOException) {
225 throw new InterruptedException();
226 }
227 else if (mSocketOrException instanceof Exception) {
228 throw new SocketException
229 ("Unable to connect to " + mKey + ", " +
230 ((Exception)mSocketOrException).getMessage());
231 }
232
233 return null;
234 }
235
236 public void run() {
237 try {
238 Socket socket;
239 Object address = mKey.mAddress;
240 if (address instanceof InetAddress) {
241 socket = new Socket((InetAddress)address, mKey.mPort);
242 }
243 else {
244 socket = new Socket(String.valueOf(address), mKey.mPort);
245 }
246
247 synchronized (this) {
248 if (mDoneWaiting) {
249 try {
250 socket.close();
251 }
252 catch (IOException e) {
253 }
254 }
255 else {
256 mSocketOrException = socket;
257 notify();
258 }
259 }
260 }
261 catch (Exception e) {
262 synchronized (this) {
263 mSocketOrException = e;
264 notify();
265 }
266 }
267 }
268 }
269 }