dev.responsive.kafka.internal.utils.SessionKey Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kafka-client Show documentation
artifact for kafka-client
The newest version!
/*
 * Copyright 2024 Responsive Computing, 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 dev.responsive.kafka.internal.utils;

import java.util.Arrays;
import java.util.Objects;
import org.apache.kafka.common.utils.Bytes;

public class SessionKey implements Comparable {

  public final Bytes key;
  public final long sessionStartMs;
  public final long sessionEndMs;

  public SessionKey(final Bytes key, final long sessionStartMs, final long sessionEndMs) {
    this.key = key;
    this.sessionStartMs = sessionStartMs;
    this.sessionEndMs = sessionEndMs;
  }

  public SessionKey(final byte[] key, final long sessionStartMs, final long sessionEndMs) {
    this(Bytes.wrap(key), sessionStartMs, sessionEndMs);
  }

  @Override
  public String toString() {
    return "SessionKey{"
        + "key=" + this.key
        + ", sessionEndMs=" + this.sessionEndMs
        + ", sessionStartMs=" + this.sessionStartMs
        + '}';
  }

  @Override
  public int compareTo(final SessionKey o) {
    final int compareKeys = this.key.compareTo(o.key);
    if (compareKeys != 0) {
      return compareKeys;
    }

    final int compareEnds = Long.compare(this.sessionEndMs, o.sessionEndMs);
    if (compareEnds != 0) {
      return compareEnds;
    }

    return Long.compare(this.sessionStartMs, o.sessionStartMs);
  }

  @Override
  public boolean equals(final Object o) {
    if (this == o) {
      return true;
    }
    if (o == null || getClass() != o.getClass()) {
      return false;
    }
    final SessionKey sessionKey = (SessionKey) o;
    return Arrays.equals(this.key.get(), sessionKey.key.get())
        && this.sessionEndMs == sessionKey.sessionEndMs
        && this.sessionStartMs == sessionKey.sessionStartMs;
  }

  @Override
  public int hashCode() {
    int result = Objects.hash(this.sessionEndMs, this.sessionStartMs);
    result = 31 * result + Arrays.hashCode(this.key.get());
    return result;
  }
}