public void register() throws DTFException {
String id = Action.getLocalID();
Connect connect;
connect = new Connect(id);
/*
* Put default Attribs into connect message.
*/
Iterator objs = _attribs.values().iterator();
while (objs.hasNext()) {
Attrib attrib = (Attrib) objs.next();
connect.addAttrib(attrib);
}
connect.setAddress(_server.getAddress());
connect.setPort(_server.getPort());
connect.setBuildid(DTFNode.getBuildID());
if (_logger.isDebugEnabled()) {
_logger.debug("Registering " + connect);
}
try {
_node.register(connect).execute();
} catch (RemoteException e) {
convertException(e, "Unable to register.");
}
}
public void execute() throws DTFException {
/*
* For the case in which we're tunneling to this agent then we should
* communicate with the DTFX using the controller. It is slower but its
* the only guaranteed method.
*/
if (!getTunneled()) {
CommClient client = new CommRMIClient(getAddress(), getPort());
if (client.heartbeat().booleanValue()) {
if (getLogger().isDebugEnabled())
getLogger().debug("Direction connection to component being used.");
Comm.addClient(getOwner(), client);
}
}
/*
* Recreate the Agent base config from the main config that the agent
* has when it starts for the first time and connects to the DTFC.
*/
ActionState as = ActionState.getInstance();
DTFState state = as.getState("main").duplicate();
state.setGlobalContext((Hashtable) state.getGlobalContext().clone());
as.setState(Node.BASE_CONFIG, state);
DTFNode.setOwner(this);
getLogger().info("This agent [" + getId() + "] in use by [" + getOwner() + "]");
}
public void stop() throws RecorderException {
try {
_recorder.stop();
TxtQuery query = new TxtQuery();
query.open(_uri, null, null, null, null, DTFConstants.DEFAULT_ENCODING);
Cursor cursor = new Cursor(query);
int count = 0;
if (DTFNode.getOwner() != null) {
String ownerId = DTFNode.getOwner().getOwner();
Sequence seq = new Sequence();
seq.setThreadID(_threadId);
while (true) {
HashMap<String, String> map = cursor.next(false);
if (map == null) break;
Event event = CLIUtil.hashMapToEvent(map);
seq.addAction(event);
count++;
if (count > 100) {
Action.getComm().sendAction(ownerId, seq);
seq.clearChildren();
count = 0;
}
}
if (seq.children().size() != 0) Action.getComm().sendAction(ownerId, seq);
}
} catch (StorageException e) {
throw new RecorderException("Error handling remote events.", e);
} catch (DTFException e) {
throw new RecorderException("Error handling remote events.", e);
} finally {
try {
RemoteStorage.delete(_uri);
} catch (StorageException e) {
throw new RecorderException("Error cleaning up event file.", e);
}
}
}