CustomModeISBuilder builder = new CustomModeISBuilder(jobResource);
builder.setRebalancerMode(IdealState.RebalanceMode.TASK);
builder.setNumReplica(1);
builder.setNumPartitions(numPartitions);
builder.setStateModel(TaskConstants.STATE_MODEL_NAME);
IdealState is = builder.build();
for (int i = 0; i < numPartitions; i++) {
is.getRecord().setListField(jobResource + "_" + i, new ArrayList<String>());
is.getRecord().setMapField(jobResource + "_" + i, new HashMap<String, String>());
}
if (taskConfigMap != null && !taskConfigMap.isEmpty()) {