in uforwarder-core/src/main/java/com/uber/data/kafka/datatransfer/common/JobUtils.java [159:172]
public static int getJobKey(StoredJob job) throws UnsupportedOperationException {
JobType type = job.getJob().getType();
switch (type) {
case JOB_TYPE_KAFKA_CONSUMER_TO_RPC_DISPATCHER:
case JOB_TYPE_KAFKA_AUDIT:
case JOB_TYPE_KAFKA_REPLICATION:
case JOB_TYPE_LOAD_GEN_CONSUME:
return job.getJob().getKafkaConsumerTask().getPartition();
case JOB_TYPE_LOAD_GEN_PRODUCE:
return job.getJob().getKafkaDispatcherTask().getPartition();
default:
throw new UnsupportedOperationException(type.toString());
}
}