in src/main/java/com/uber/rss/tools/StreamServerStressToolLongRun.java [109:155]
public static void main(String[] args) {
Thread.setDefaultUncaughtExceptionHandler(new Thread.UncaughtExceptionHandler() {
@Override
public void uncaughtException(Thread t, Throwable e) {
logger.error(String.format("Got exception from thread %s", t.getName()), e);
System.exit(-1);
}
});
StreamServerStressToolLongRun longRun = new StreamServerStressToolLongRun();
int i = 0;
while (i < args.length) {
String argName = args[i++];
if (argName.equalsIgnoreCase("-runMinutes")) {
longRun.runMinutes = Integer.parseInt(args[i++]);
} else if (argName.equalsIgnoreCase("-workDir")) {
longRun.workDir = args[i++];
} else if (argName.equalsIgnoreCase("-compressCodec")) {
longRun.compressCodec = args[i++];
} else if (argName.equalsIgnoreCase("-bufferSize")) {
longRun.bufferSize = Integer.parseInt(args[i++]);
} else if (argName.equalsIgnoreCase("-maxNumServers")) {
longRun.maxNumServers = Integer.parseInt(args[i++]);
} else if (argName.equalsIgnoreCase("-maxNumBytes")) {
longRun.maxNumBytes = com.uber.rss.util.StringUtils.getBytesValue(args[i++]);
} else if (argName.equalsIgnoreCase("-maxNumMaps")) {
longRun.maxNumMaps = Integer.parseInt(args[i++]);
} else if (argName.equalsIgnoreCase("-maxNumMapAttempts")) {
longRun.maxNumMapAttempts = Integer.parseInt(args[i++]);
} else if (argName.equalsIgnoreCase("-maxNumPartitions")) {
longRun.maxNumPartitions = Integer.parseInt(args[i++]);
} else if (argName.equalsIgnoreCase("-maxNumSplits")) {
longRun.maxNumSplits = Integer.parseInt(args[i++]);
} else if (argName.equalsIgnoreCase("-maxNumReplicas")) {
longRun.maxNumReplicas = Integer.parseInt(args[i++]);
} else {
throw new IllegalArgumentException("Unsupported argument: " + argName);
}
}
longRun.run();
M3Stats.closeDefaultScope();
logger.info(String.format("%s finished", StreamServerStressToolLongRun.class.getSimpleName()));
}