01: /*
02: * <copyright>
03: *
04: * Copyright 1997-2006 BBNT Solutions, LLC
05: * under sponsorship of the Defense Advanced Research Projects
06: * Agency (DARPA).
07: *
08: * You can redistribute this software and/or modify it under the
09: * terms of the Cougaar Open Source License as published on the
10: * Cougaar Open Source Website (www.cougaar.org).
11: *
12: * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
13: * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
14: * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
15: * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
16: * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
17: * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
18: * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
19: * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
20: * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
21: * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
22: * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
23: *
24: * </copyright>
25: */
26: package org.cougaar.mts.rmi;
27:
28: import org.cougaar.core.mts.MessageAttributes;
29: import org.cougaar.mts.base.CommFailureException;
30: import org.cougaar.mts.base.DestinationLink;
31: import org.cougaar.mts.base.DestinationLinkDelegateImplBase;
32: import org.cougaar.mts.base.MisdeliveredMessageException;
33: import org.cougaar.mts.base.NameLookupException;
34: import org.cougaar.mts.base.StandardAspect;
35: import org.cougaar.mts.base.UnregisteredNameException;
36: import org.cougaar.mts.std.AttributedMessage;
37:
38: /**
39: * Sample Aspect that will disable RMI for 30 seconds (by default) after a failure.
40: *
41: */
42: public class EnableRmiSenderSideAspect extends StandardAspect {
43: private long failureTimeout;
44:
45: public void load() {
46: super .load();
47: failureTimeout = getParameter("failure-timeout", 30000);
48: }
49:
50: public Object getDelegate(Object delegatee, Class type) {
51: if (type == DestinationLink.class) {
52: DestinationLink link = (DestinationLink) delegatee;
53: if (link.getProtocolClass() == RMILinkProtocol.class)
54: return new RmiEnableDestinationLink(link);
55: }
56: return null;
57: }
58:
59: private class RmiEnableDestinationLink extends
60: DestinationLinkDelegateImplBase {
61: long last_fail_time = 0;
62:
63: RmiEnableDestinationLink(DestinationLink delegatee) {
64: super (delegatee);
65: }
66:
67: public boolean isValid() {
68: long now = System.currentTimeMillis();
69: if (now - last_fail_time < failureTimeout) {
70: return false;
71: } else {
72: return super .isValid();
73: }
74: }
75:
76: public MessageAttributes forwardMessage(
77: AttributedMessage message) throws NameLookupException,
78: UnregisteredNameException, CommFailureException,
79: MisdeliveredMessageException {
80: long now = System.currentTimeMillis();
81: try {
82: return super .forwardMessage(message);
83: } catch (MisdeliveredMessageException ex) {
84: last_fail_time = now;
85: throw ex;
86: } catch (CommFailureException ex) {
87: last_fail_time = now;
88: throw ex;
89: } catch (UnregisteredNameException ex) {
90: last_fail_time = now;
91: throw ex;
92: } catch (NameLookupException ex) {
93: last_fail_time = now;
94: throw ex;
95: }
96: }
97: }
98: }
|