@Override
public CompoundStatus checkTargetTTsSuccess(
String opType, String[] affectedTTs, int totalTargetEnabled, HadoopCluster cluster) {
CompoundStatus status = new CompoundStatus("checkTargetTTsSuccess");
String scriptFileName = CHECK_SCRIPT_FILE_NAME;
String scriptRemoteFilePath = DEFAULT_SCRIPT_DEST_PATH + scriptFileName;
String listRemoteFilePath = null;
String opDesc = "checkTargetTTsSuccess";
_log.log(Level.INFO, "AffectedTTs:");
for (String tt : affectedTTs) {
_log.log(Level.INFO, tt);
}
HadoopConnection connection = getConnectionForCluster(cluster);
setErrorParamsForCommand(opDesc, scriptRemoteFilePath, listRemoteFilePath);
int rc = -1;
int iterations = 0;
do {
if (iterations > 0) {
_log.log(Level.INFO, "Target TTs not yet achieved...checking again - " + iterations);
}
OutputStream out = new ByteArrayOutputStream();
rc =
executeScriptWithCopyRetryOnFailure(
connection,
scriptFileName,
new String[] {"" + totalTargetEnabled, connection.getHadoopHome()},
out);
try {
out.flush();
} catch (IOException e) {
String errorMsg = "Unexpected exception in SSH OutputStream ";
_log.log(Level.WARNING, errorMsg, e);
status.registerTaskFailed(false, errorMsg + e.getMessage());
}
// _log.log(Level.INFO, "Output from SSH script execution:\n"+out.toString());
/* Convert to String array and "nullify" last element (which happens to be "@@@..." or empty line) */
String[] allActiveTTs = out.toString().split("\n");
allActiveTTs[allActiveTTs.length - 1] = null;
if (checkOpSuccess(opType, affectedTTs, allActiveTTs)) {
_log.log(Level.INFO, "All selected TTs correctly %sed", opType.toLowerCase());
rc = SUCCESS;
break;
}
// TODO: out.close()?
} while ((rc == ERROR_FEWER_TTS || rc == ERROR_EXCESS_TTS)
&& (++iterations <= MAX_CHECK_RETRY_ITERATIONS));
status.addStatus(_errorCodes.interpretErrorCode(_log, rc, _errorParamValues));
return status;
}
private CompoundStatus decomRecomTTs(
String opDesc,
String[] tts,
HadoopCluster cluster,
String scriptFileName,
String listFileName) {
CompoundStatus status = new CompoundStatus("decomRecomTTs");
if (!isValidTTList(tts)) {
String errorMsg = opDesc + " failed due to bad TT list";
_log.log(Level.SEVERE, opDesc + " failed due to bad TT list");
status.registerTaskFailed(false, errorMsg);
return status;
}
String scriptRemoteFilePath = DEFAULT_SCRIPT_DEST_PATH + scriptFileName;
String listRemoteFilePath = DEFAULT_SCRIPT_DEST_PATH + listFileName;
HadoopConnection connection = getConnectionForCluster(cluster);
setErrorParamsForCommand(opDesc.toLowerCase(), scriptRemoteFilePath, listRemoteFilePath);
OutputStream out = new ByteArrayOutputStream();
String operationList = createVMList(tts);
int rc =
connection.copyDataToJobTracker(
operationList.getBytes(), DEFAULT_SCRIPT_DEST_PATH, listFileName, false);
if (rc == 0) {
rc =
executeScriptWithCopyRetryOnFailure(
connection,
scriptFileName,
new String[] {
listRemoteFilePath, connection.getExcludeFilePath(), connection.getHadoopHome()
},
out);
}
status.addStatus(_errorCodes.interpretErrorCode(_log, rc, _errorParamValues));
return status;
}