001: /*
002: License $Id: JoBalancer.java,v 1.4 2003/09/13 04:59:58 hendriks73 Exp $
003:
004: Copyright (c) 2001-2005 tagtraum industries.
005:
006: LGPL
007: ====
008:
009: jo! is free software; you can redistribute it and/or
010: modify it under the terms of the GNU Lesser General Public
011: License as published by the Free Software Foundation; either
012: version 2.1 of the License, or (at your option) any later version.
013:
014: jo! 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. See the GNU
017: Lesser General Public License for more details.
018:
019: You should have received a copy of the GNU Lesser General Public
020: License along with this library; if not, write to the Free Software
021: Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
022:
023: For LGPL see <http://www.fsf.org/copyleft/lesser.txt>
024:
025:
026: Sun license
027: ===========
028:
029: This release contains software by Sun Microsystems. Therefore
030: the following conditions have to be met, too. They apply to the
031: files
032:
033: - lib/mail.jar
034: - lib/activation.jar
035: - lib/jsse.jar
036: - lib/jcert.jar
037: - lib/jaxp.jar
038: - lib/crimson.jar
039: - lib/servlet.jar
040: - lib/jnet.jar
041: - lib/jaas.jar
042: - lib/jaasmod.jar
043:
044: contained in this release.
045:
046: a. Licensee may not modify the Java Platform
047: Interface (JPI, identified as classes contained within the javax
048: package or any subpackages of the javax package), by creating additional
049: classes within the JPI or otherwise causing the addition to or modification
050: of the classes in the JPI. In the event that Licensee creates any
051: Java-related API and distribute such API to others for applet or
052: application development, you must promptly publish broadly, an accurate
053: specification for such API for free use by all developers of Java-based
054: software.
055:
056: b. Software is confidential copyrighted information of Sun and
057: title to all copies is retained by Sun and/or its licensors. Licensee
058: shall not modify, decompile, disassemble, decrypt, extract, or otherwise
059: reverse engineer Software. Software may not be leased, assigned, or
060: sublicensed, in whole or in part. Software is not designed or intended
061: for use in on-line control of aircraft, air traffic, aircraft navigation
062: or aircraft communications; or in the design, construction, operation or
063: maintenance of any nuclear facility. Licensee warrants that it will not
064: use or redistribute the Software for such purposes.
065:
066: c. Software is provided "AS IS," without a warranty
067: of any kind. ALL EXPRESS OR IMPLIED REPRESENTATIONS AND WARRANTIES,
068: INCLUDING ANY IMPLIED WARRANTY OF MERCHANTABILITY, FITNESS FOR A
069: PARTICULAR PURPOSE OR NON-INFRINGEMENT, ARE HEREBY EXCLUDED.
070:
071: d. This License is effective until terminated. Licensee may
072: terminate this License at any time by destroying all copies of Software.
073: This License will terminate immediately without notice from Sun if Licensee
074: fails to comply with any provision of this License. Upon such termination,
075: Licensee must destroy all copies of Software.
076:
077: e. Software, including technical data, is subject to U.S.
078: export control laws, including the U.S. Export Administration Act and its
079: associated regulations, and may be subject to export or import regulations
080: in other countries. Licensee agrees to comply strictly with all such
081: regulations and acknowledges that it has the responsibility to obtain
082: licenses to export, re-export, or import Software. Software may not be
083: downloaded, or otherwise exported or re-exported (i) into, or to a national
084: or resident of, Cuba, Iraq, Iran, North Korea, Libya, Sudan, Syria or any
085: country to which the U.S. has embargoed goods; or (ii) to anyone on the
086: U.S. Treasury Department's list of Specially Designated Nations or the U.S.
087: Commerce Department's Table of Denial Orders.
088:
089:
090: Feedback
091: ========
092:
093: We encourage your feedback and suggestions and want to use your feedback to
094: improve the Software. Send all such feedback to:
095: <feedback@tagtraum.com>
096:
097: For more information on tagtraum industries and jo!
098: please see <http://www.tagtraum.com/>.
099:
100:
101: */
102: package com.tagtraum.jobalancer;
103:
104: import com.tagtraum.framework.log.C_Log;
105: import com.tagtraum.framework.log.Log;
106: import com.tagtraum.framework.server.Service;
107: import com.tagtraum.framework.util.InetAddressPort;
108:
109: import java.io.IOException;
110: import java.net.InetAddress;
111: import java.net.Socket;
112: import java.util.Date;
113: import java.util.HashMap;
114:
115: /**
116: *
117: * @author Hendrik Schreiber
118: * @version 1.1beta1 $Id: JoBalancer.java,v 1.4 2003/09/13 04:59:58 hendriks73 Exp $
119: */
120: public class JoBalancer extends Service {
121:
122: /**
123: * Source-Version
124: */
125: public static String vcid = "$Id: JoBalancer.java,v 1.4 2003/09/13 04:59:58 hendriks73 Exp $";
126:
127: private StickyIPMap stickyIPMap;
128: private RingIPList ringIPList;
129: private HashMap availableMap;
130: private long unavailableTimeout;
131: // 10 min
132: public final static long DEFAULT_TIMEOUT = 1000L * 60L * 10L;
133: public final static String DEFAULT_HANDLERCLASSNAME = "com.tagtraum.jobalancer.JoBalancerHandler";
134:
135: public JoBalancer() {
136: super ();
137: stickyIPMap = new StickyIPMap();
138: ringIPList = new RingIPList();
139: availableMap = new HashMap();
140: unavailableTimeout = DEFAULT_TIMEOUT;
141: setName("JoBalancer");
142: setHandlerClassname(DEFAULT_HANDLERCLASSNAME);
143: }
144:
145: public void setUnavailableTimeout(long timeout) {
146: unavailableTimeout = timeout;
147: if (Log.isLog(C_Log.METHOD, getName())) {
148: Log.getLog(getName()).log(
149: "Set unavailable timeout to: " + timeout,
150: C_Log.METHOD);
151: }
152: }
153:
154: public long getUnavailableTimeout() {
155: return unavailableTimeout;
156: }
157:
158: public void setStickyTimeout(long timeout) {
159: stickyIPMap.setStickyTimeout(timeout);
160: if (Log.isLog(C_Log.METHOD, getName())) {
161: Log.getLog(getName()).log(
162: "Set sticky timeout to: " + timeout, C_Log.METHOD);
163: }
164: }
165:
166: public long getStickyTimeout() {
167: return stickyIPMap.getStickyTimeout();
168: }
169:
170: public void addServer(InetAddressPort iap) {
171: ringIPList.add(iap);
172: if (Log.isLog(C_Log.METHOD, getName())) {
173: Log.getLog(getName()).log("Added server: " + iap,
174: C_Log.METHOD);
175: }
176: }
177:
178: public boolean removeServer(InetAddressPort iap) {
179: boolean ret = ringIPList.remove(iap);
180: if (Log.isLog(C_Log.METHOD, getName())) {
181: Log.getLog(getName()).log(
182: "Tried to remove server: " + iap + " (success: "
183: + ret + ")", C_Log.METHOD);
184: }
185: return ret;
186: }
187:
188: public InetAddressPort[] getServers() {
189: return ringIPList.toArray();
190: }
191:
192: public Socket getSocket(InetAddress address) throws IOException {
193: InetAddressPort iap = getStickyIAP(address);
194: if (iap == null) {
195: if (Log.isLog(C_Log.METHOD, getName())) {
196: Log.getLog(getName()).log(
197: "Didn't find sticky server for client: "
198: + address, C_Log.METHOD);
199: }
200: iap = getRingIAP(address);
201: }
202: if (iap == null) {
203: if (Log.isLog(C_Log.ERROR, getName())) {
204: Log.getLog(getName()).log(
205: "No server available for " + address,
206: C_Log.ERROR);
207: }
208: throw new IOException("No server available for " + address);
209: }
210: InetAddressPort first = iap;
211: Socket s = null;
212: while (s == null) {
213: if (Log.isLog(C_Log.METHOD, getName())) {
214: Log.getLog(getName()).log(
215: "Trying to connect to server " + iap,
216: C_Log.METHOD);
217: }
218: try {
219: s = new Socket(iap.getInetAddress(), iap.getPort());
220: } catch (IOException ioe) {
221: if (Log.isLog(C_Log.METHOD, getName())) {
222: Log.getLog(getName()).log(
223: "Failed to instantiate socket " + iap
224: + " for client " + address,
225: C_Log.METHOD);
226: }
227: markNotAvailable(iap);
228: iap = getRingIAP(address);
229: if (iap == null) {
230: if (Log.isLog(C_Log.ERROR, getName())) {
231: Log.getLog(getName()).log(
232: "No server available for " + address,
233: C_Log.ERROR);
234: }
235: throw new IOException("No server available.");
236: }
237: }
238: }
239: stickyIPMap.add(address, iap);
240: if (Log.isLog(C_Log.MODULE, getName())) {
241: Log.getLog(getName()).log(
242: "Client " + address + " got stuck with server "
243: + iap, C_Log.MODULE);
244: }
245: return s;
246: }
247:
248: private InetAddressPort getStickyIAP(InetAddress address) {
249: InetAddressPort iap = (InetAddressPort) stickyIPMap
250: .get(address);
251: if (iap != null && isAvailable(iap))
252: return iap;
253: return null;
254: }
255:
256: private InetAddressPort getRingIAP(InetAddress address) {
257: int i = 0;
258: InetAddressPort iap = null;
259: do {
260: iap = ringIPList.next();
261: if (isAvailable(iap))
262: return iap;
263: } while (!isAvailable(iap) && i < ringIPList.size());
264: return null;
265: }
266:
267: private synchronized boolean isAvailable(InetAddressPort iap) {
268: Long timestamp = (Long) availableMap.get(iap);
269: if (timestamp == null)
270: return true;
271: if (timestamp.longValue() < System.currentTimeMillis()) {
272: availableMap.remove(iap);
273: return true;
274: }
275: if (Log.isLog(C_Log.FORTYTWO, getName())) {
276: Log.getLog(getName()).log(
277: "Server " + iap + " is marked not available till "
278: + new Date(timestamp.longValue()) + ".",
279: C_Log.FORTYTWO);
280: }
281: return false;
282: }
283:
284: private synchronized void markNotAvailable(InetAddressPort iap) {
285: Long till = new Long(System.currentTimeMillis()
286: + unavailableTimeout);
287: availableMap.put(iap, till);
288: if (Log.isLog(C_Log.MODULE, getName())) {
289: Log.getLog(getName()).log(
290: "Server " + iap + " got marked unavailable till "
291: + new Date(till.longValue()) + ".",
292: C_Log.MODULE);
293: }
294: }
295:
296: }
|