org.evosuite.rmi.service.MasterNodeImpl Maven / Gradle / Ivy
The newest version!
/**
* Copyright (C) 2010-2018 Gordon Fraser, Andrea Arcuri and EvoSuite
* contributors
*
* This file is part of EvoSuite.
*
* EvoSuite is free software: you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as published
* by the Free Software Foundation, either version 3.0 of the License, or
* (at your option) any later version.
*
* EvoSuite is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with EvoSuite. If not, see clients;
protected final Collection> listeners = Collections.synchronizedList(new ArrayList>());
/**
* It is important to keep track of client states for debugging reasons. For
* example, if client crash, could be useful to know in which state it was.
* We cannot query the client directly in those cases, because it is
* crashed... The "key" is the RMI identifier of the client
*/
private final Map clientStates;
private final Map clientStateInformation;
public MasterNodeImpl(Registry registry) {
clients = new ConcurrentHashMap();
clientStates = new ConcurrentHashMap();
clientStateInformation = new ConcurrentHashMap();
this.registry = registry;
}
@Override
public void evosuite_registerClientNode(String clientRmiIdentifier) throws RemoteException {
/*
* The client should first register its node, and then inform MasterNode
* by calling this method
*/
ClientNodeRemote node = null;
try {
node = (ClientNodeRemote) registry.lookup(clientRmiIdentifier);
} catch (Exception e) {
logger.error("Error when client " + clientRmiIdentifier
+ " tries to register to master", e);
return;
}
synchronized (clients) {
clients.put(clientRmiIdentifier, node);
clients.notifyAll();
}
}
@Override
public void evosuite_informChangeOfStateInClient(String clientRmiIdentifier,
ClientState state, ClientStateInformation information) throws RemoteException {
clientStates.put(clientRmiIdentifier, state);
// To be on the safe side
information.setState(state);
clientStateInformation.put(clientRmiIdentifier, information);
fireEvent(information);
}
@Override
public Collection getCurrentState() {
return clientStates.values();
}
@Override
public ClientState getCurrentState(String clientId) {
return clientStates.get(clientId);
}
@Override
public Collection getCurrentStateInformation() {
return clientStateInformation.values();
}
@Override
public String getSummaryOfClientStatuses() {
if (clientStates.isEmpty()) {
return "No client has registered";
}
String summary = "";
for (String id : clientStates.keySet()) {
ClientState state = clientStates.get(id);
summary += id + ": " + state + "\n";
}
return summary;
}
@Override
public Map getClientsOnceAllConnected(long timeoutInMs)
throws InterruptedException {
long start = System.currentTimeMillis();
int numberOfExpectedClients = Properties.NUM_PARALLEL_CLIENTS;
synchronized (clients) {
while (clients.size() != numberOfExpectedClients) {
long elapsed = System.currentTimeMillis() - start;
long timeRemained = timeoutInMs - elapsed;
if (timeRemained <= 0) {
return null;
}
clients.wait(timeRemained);
}
return Collections.unmodifiableMap(clients);
}
}
@Override
public void cancelAllClients() {
for (ClientNodeRemote client : clients.values()) {
try {
LoggingUtils.getEvoLogger().info("Trying to kill client " + client);
client.cancelCurrentSearch();
} catch (RemoteException e) {
logger.warn("Error while trying to cancel client: " + e);
e.printStackTrace();
}
}
}
@Override
public void evosuite_collectStatistics(String clientRmiIdentifier, Chromosome individual) {
SearchStatistics.getInstance(clientRmiIdentifier).currentIndividual(individual);
}
@Override
public void evosuite_collectStatistics(String clientRmiIdentifier, RuntimeVariable variable, Object value)
throws RemoteException {
SearchStatistics.getInstance(clientRmiIdentifier).setOutputVariable(variable, value);
}
@Override
public void evosuite_collectTestGenerationResult(
String clientRmiIdentifier, List results)
throws RemoteException {
SearchStatistics.getInstance(clientRmiIdentifier).addTestGenerationResult(results);
}
@Override
public void evosuite_flushStatisticsForClassChange(String clientRmiIdentifier)
throws RemoteException {
SearchStatistics.getInstance(clientRmiIdentifier).writeStatisticsForAnalysis();
}
@Override
public void evosuite_updateProperty(String clientRmiIdentifier, String propertyName, Object value)
throws RemoteException, IllegalArgumentException, IllegalAccessException, NoSuchParameterException {
Properties.getInstance().setValue(propertyName, value);
}
@Override
public void evosuite_migrate(String clientRmiIdentifier, Set migrants)
throws RemoteException {
//implements ring topology
int idSender = Integer.parseInt(clientRmiIdentifier.replaceAll("[^0-9]", ""));
int idNeighbour = (idSender + 1) % Properties.NUM_PARALLEL_CLIENTS;
while (!ClientState.SEARCH.equals(clientStates.get("ClientNode" + idNeighbour)) && idNeighbour != idSender) {
idNeighbour = (idNeighbour + 1) % Properties.NUM_PARALLEL_CLIENTS;
}
if (idNeighbour != idSender) {
ClientNodeRemote node = clients.get("ClientNode" + idNeighbour);
node.immigrate(migrants);
}
}
@Override
public void evosuite_collectBestSolutions(String clientRmiIdentifier, Set solutions) {
try {
ClientNodeRemote node = clients.get(ClientProcess.DEFAULT_CLIENT_NAME);
node.collectBestSolutions(solutions);
} catch (RemoteException e) {
logger.error("Cannot send best solutions to client 0", e);
}
}
@Override
public void addListener(Listener listener) {
listeners.add(listener);
}
@Override
public void deleteListener(Listener listener) {
listeners.remove(listener);
}
/**
*
* fireEvent
*
*
* @param event
* a T object.
*/
public void fireEvent(ClientStateInformation event) {
for (Listener listener : listeners) {
listener.receiveEvent(event);
}
}
}