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:
27: package org.cougaar.core.agent;
28:
29: import org.cougaar.core.component.Component;
30: import org.cougaar.core.component.ServiceBroker;
31: import org.cougaar.core.component.ServiceRevokedListener;
32: import org.cougaar.core.mts.MessageAddress;
33: import org.cougaar.core.service.AgentIdentificationService;
34: import org.cougaar.core.service.LoggingService;
35: import org.cougaar.util.GenericStateModelAdapter;
36:
37: /**
38: * This component registers the agent name in the
39: * {@link ClusterContextTable}, which is required for serialization
40: * and deserialization of persistence snapshots and messages.
41: */
42: public final class RegisterContext extends GenericStateModelAdapter
43: implements Component {
44:
45: private ServiceBroker sb;
46:
47: private LoggingService log;
48: private MessageAddress localAgent;
49:
50: public void setServiceBroker(ServiceBroker sb) {
51: this .sb = sb;
52: }
53:
54: public void load() {
55: super .load();
56:
57: log = (LoggingService) sb.getService(this ,
58: LoggingService.class, null);
59:
60: // get our local agent's address
61: AgentIdentificationService ais = (AgentIdentificationService) sb
62: .getService(this , AgentIdentificationService.class,
63: null);
64: if (ais != null) {
65: localAgent = ais.getMessageAddress();
66: sb.releaseService(this , AgentIdentificationService.class,
67: ais);
68: }
69:
70: // add our address to our VM's table
71: if (log.isDebugEnabled()) {
72: log.debug("Adding to the cluster context table");
73: }
74: ClusterContextTable.addContext(localAgent);
75: }
76:
77: public void unload() {
78: super .unload();
79: // remove ourselves from the VM-local context
80: if (log.isDebugEnabled()) {
81: log.debug("Removing from the cluster context table");
82: }
83: ClusterContextTable.removeContext(localAgent);
84: }
85: }
|