01: /*
02: * <copyright>
03: *
04: * Copyright 1997-2004 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: * This test Aspect uses the {@link RMISocketControlService} to set
40: * socket timeouts.
41: */
42: public class RMITestAspect extends StandardAspect
43:
44: {
45: private RMISocketControlService svc;
46:
47: public Object getDelegate(Object object, Class type) {
48: if (type == DestinationLink.class) {
49: return new Delegate((DestinationLink) object);
50: } else {
51: return null;
52: }
53: }
54:
55: public void load() {
56: super .load();
57:
58: svc = (RMISocketControlService) getServiceBroker().getService(
59: this , RMISocketControlService.class, null);
60:
61: }
62:
63: private class Delegate extends DestinationLinkDelegateImplBase {
64: Delegate(DestinationLink link) {
65: super (link);
66: }
67:
68: public MessageAttributes forwardMessage(
69: AttributedMessage message)
70: throws UnregisteredNameException, NameLookupException,
71: CommFailureException, MisdeliveredMessageException {
72: svc.setSoTimeout(getDestination(), 2000);
73: return super.forwardMessage(message);
74: }
75:
76: }
77:
78: }
|