public void disposeWorker(WorkerEntry workerEntry, List<IResponseTO> responses) {
ServiceID serviceID = workerEntry.getServiceID();
responses.add(new LoggerResponseTO("Worker dispose: " + serviceID, LoggerResponseTO.DEBUG));
workerEntry.dispose();
WorkerDAO workerDAO = BrokerDAOFactory.getInstance().getWorkerDAO();
String wAddress = StringUtil.deploymentIDToAddress(workerEntry.getWorkerID());
DisposeWorkerResponseTO disposeTO = new DisposeWorkerResponseTO();
disposeTO.setPeerAddress(StringUtil.deploymentIDToAddress(workerEntry.getPeerID()));
disposeTO.setWorkerAddress(wAddress);
disposeTO.setWorkerPublicKey(workerDAO.getWorkerPublicKey(wAddress));
responses.add(disposeTO);
responses.add(new LoggerResponseTO("Stub to be released: " + wAddress, LoggerResponseTO.INFO));
ReleaseResponseTO releaseTO = new ReleaseResponseTO();
releaseTO.setStubAddress(wAddress);
responses.add(releaseTO);
workerDAO.removeWorker(wAddress);
WorkerInfo.getInstance().removeWorker(workerEntry.getServiceID().getContainerID().toString());
}
public boolean isASaboteurWorker(ServiceID workerServiceID) {
if (this.saboteurs != null) {
for (WorkerEntry entry : this.saboteurs) {
if (entry.getServiceID().equals(workerServiceID)) {
return true;
}
}
}
return false;
}
public boolean isWorkerUnwanted(Job job, String workerID) {
DeploymentID workerDeploymentID = new DeploymentID(workerID);
ServiceID workerServiceID = workerDeploymentID.getServiceID();
for (WorkerEntry entry : job.getBlackListedWorkers()) {
if (entry.getServiceID().equals(workerServiceID)) {
return true;
}
}
return isASaboteurWorker(workerServiceID);
}
protected int getRemainingBlacklistFails(WorkerEntry workerEntry, Job job) {
if (isASaboteurWorker(workerEntry.getServiceID())) {
return 0;
}
if (workerEntry != null) {
/*int jobSize = Math.min(job.getSpec().getTaskSpecs().size(), this.maxBlFails);
return Math.max( 0, jobSize - workerEntry.getNumberOfBlacklistedTasks() );*/
return Math.max(0, this.maxBlFails - workerEntry.getNumberOfBlacklistedTasks());
}
return -1;
}
/**
* @param job
* @param workerEntry
*/
private void unwantWorker(Job job, WorkerEntry workerEntry, List<IResponseTO> responses) {
responses.add(
new LoggerResponseTO(
"Worker unwanted: " + workerEntry.getWorkerID(), LoggerResponseTO.DEBUG));
job.unwantWorker(workerEntry);
workerEntry.unwant();
RequestSpecification spec = workerEntry.getRequestSpecification();
String peerAddress = StringUtil.deploymentIDToAddress(workerEntry.getPeerID());
String workerAddress = StringUtil.deploymentIDToAddress(workerEntry.getWorkerID());
Integer jobId = Integer.parseInt("" + spec.getJobId());
JobSpecification jobSpec = BrokerDAOFactory.getInstance().getJobDAO().getJobSpec(jobId);
UnwantWorkerResponseTO unwantWorkerTO = new UnwantWorkerResponseTO();
unwantWorkerTO.setJobID(jobId);
unwantWorkerTO.setJobSpec(jobSpec);
unwantWorkerTO.setMaxFails(spec.getMaxFails());
unwantWorkerTO.setMaxReplicas(spec.getMaxReplicas());
unwantWorkerTO.setPeerAddress(peerAddress);
unwantWorkerTO.setRequestID(spec.getRequestId());
unwantWorkerTO.setRequiredWorkers(spec.getRequiredWorkers());
unwantWorkerTO.setWorkerAddress(workerAddress);
WorkerDAO workerDAO = BrokerDAOFactory.getInstance().getWorkerDAO();
String workerPublicKey = workerDAO.getWorkerPublicKey(workerAddress);
unwantWorkerTO.setWorkerPublicKey(workerPublicKey);
responses.add(unwantWorkerTO);
ReleaseResponseTO releaseTO = new ReleaseResponseTO();
releaseTO.setStubAddress(workerAddress);
responses.add(releaseTO);
workerDAO.removeWorker(workerAddress);
WorkerInfo.getInstance().removeWorker(workerEntry.getServiceID().getContainerID().toString());
}
private boolean createAndAllocateExecution(Job job, Task task, WorkerEntry chosenWorker) {
GridProcess replica = null;
if (canReplicate(task)) {
replica = job.createAndAllocateExecution(task.getTaskid(), chosenWorker);
replica.setRunningState(stateMachine.getInitialState());
}
if (replica != null) {
chosenWorker.allocate(replica);
WorkerEntry worker =
WorkerInfo.getInstance()
.getWorker(chosenWorker.getServiceID().getContainerID().toString());
worker.allocate(replica);
return true;
}
return false;
}