ConsumerGroupConfig.java

/*
 * Copyright © 2015 Cask Data, Inc.
 *
 * Licensed 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 co.cask.cdap.data2.queue;

import com.google.common.base.Objects;

/**
 * Contains queue consumer group information.
 */
public class ConsumerGroupConfig {

  private final long groupId;
  private final int groupSize;
  private final DequeueStrategy dequeueStrategy;
  private final String hashKey;

  public ConsumerGroupConfig(long groupId, int groupSize, DequeueStrategy dequeueStrategy, String hashKey) {
    this.groupId = groupId;
    this.groupSize = groupSize;
    this.dequeueStrategy = dequeueStrategy;
    this.hashKey = dequeueStrategy == DequeueStrategy.HASH ? hashKey : null;
  }

  public ConsumerGroupConfig(ConsumerGroupConfig other) {
    this(other.getGroupId(), other.getGroupSize(), other.getDequeueStrategy(), other.getHashKey());
  }

  public long getGroupId() {
    return groupId;
  }

  public int getGroupSize() {
    return groupSize;
  }

  public DequeueStrategy getDequeueStrategy() {
    return dequeueStrategy;
  }

  public String getHashKey() {
    return hashKey;
  }

  @Override
  public String toString() {
    return Objects.toStringHelper(this)
      .add("groupId", groupId)
      .add("groupSize", groupSize)
      .add("dequeueStrategy", dequeueStrategy)
      .add("hashKey", hashKey)
      .toString();
  }

  @Override
  public boolean equals(Object o) {
    if (this == o) {
      return true;
    }
    if (o == null || getClass() != o.getClass()) {
      return false;
    }

    ConsumerGroupConfig other = (ConsumerGroupConfig) o;

    return groupId == other.groupId
      && groupSize == other.groupSize
      && dequeueStrategy == other.dequeueStrategy
      && Objects.equal(hashKey, other.hashKey);
  }

  @Override
  public int hashCode() {
    return Objects.hashCode(groupId, groupSize, dequeueStrategy, hashKey);
  }
}