public static Job withJobKey()

in uforwarder-core/src/main/java/com/uber/data/kafka/datatransfer/common/JobUtils.java [211:224]

• 14 lines of code
• 6 McCabe index (conditional complexity)

  public static Job withJobKey(Job.Builder builder, int jobKey) {
    switch (builder.getType()) {
      case JOB_TYPE_KAFKA_CONSUMER_TO_RPC_DISPATCHER:
      case JOB_TYPE_KAFKA_REPLICATION:
      case JOB_TYPE_KAFKA_AUDIT:
      case JOB_TYPE_LOAD_GEN_CONSUME:
        builder.getKafkaConsumerTaskBuilder().setPartition(jobKey);
        break;
      case JOB_TYPE_LOAD_GEN_PRODUCE:
        builder.getKafkaDispatcherTaskBuilder().setPartition(jobKey);
        break;
    }
    return builder.build();
  }