KafkaPartition.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package gobblin.source.extractor.extract.kafka;

import com.google.common.net.HostAndPort;


/**
 * A kafka topic partition.
 * Two partitions are considered equivalent if they have the same topic name and partition id. They may have different leaders.
 *
 * @author Ziyang Liu
 *
 */
public final class KafkaPartition {
  private final int id;
  private final String topicName;
  private KafkaLeader leader;

  public static class Builder {
    private int id = 0;
    private String topicName = "";
    private int leaderId = 0;
    private HostAndPort leaderHostAndPort;

    public Builder withId(int id) {
      this.id = id;
      return this;
    }

    public Builder withTopicName(String topicName) {
      this.topicName = topicName;
      return this;
    }

    public Builder withLeaderId(int leaderId) {
      this.leaderId = leaderId;
      return this;
    }

    public Builder withLeaderHostAndPort(String hostPortString) {
      this.leaderHostAndPort = HostAndPort.fromString(hostPortString);
      return this;
    }

    public Builder withLeaderHostAndPort(String host, int port) {
      this.leaderHostAndPort = HostAndPort.fromParts(host, port);
      return this;
    }

    public KafkaPartition build() {
      return new KafkaPartition(this);
    }
  }

  public KafkaPartition(KafkaPartition other) {
    this.topicName = other.topicName;
    this.id = other.id;
    this.leader = new KafkaLeader(other.leader.id, other.leader.hostAndPort);
  }

  private KafkaPartition(Builder builder) {
    this.id = builder.id;
    this.topicName = builder.topicName;
    this.leader = new KafkaLeader(builder.leaderId, builder.leaderHostAndPort);
  }

  public KafkaLeader getLeader() {
    return this.leader;
  }

  public String getTopicName() {
    return this.topicName;
  }

  public int getId() {
    return this.id;
  }

  public void setLeader(int leaderId, String leaderHost, int leaderPort) {
    this.leader = new KafkaLeader(leaderId, HostAndPort.fromParts(leaderHost, leaderPort));
  }

  @Override
  public String toString() {
    return this.getTopicName() + ":" + this.getId();
  }

  @Override
  public int hashCode() {
    final int prime = 31;
    int result = 1;
    result = prime * result + this.id;
    result = prime * result + ((this.topicName == null) ? 0 : this.topicName.hashCode());
    return result;
  }

  @Override
  public boolean equals(Object obj) {
    if (!(obj instanceof KafkaPartition)) {
      return false;
    }
    KafkaPartition other = (KafkaPartition) obj;
    if (this.id != other.id) {
      return false;
    }
    if (this.topicName == null) {
      if (other.topicName != null) {
        return false;
      }
    } else if (!this.topicName.equals(other.topicName)) {
      return false;
    }
    return true;
  }

  public final static class KafkaLeader {
    private final int id;
    private final HostAndPort hostAndPort;

    public int getId() {
      return this.id;
    }

    public HostAndPort getHostAndPort() {
      return this.hostAndPort;
    }

    public KafkaLeader(int id, HostAndPort hostAndPort) {
      this.id = id;
      this.hostAndPort = hostAndPort;
    }
  }

}