session-android/src/org/thoughtcrime/securesms/util/StableIdGenerator.java

package org.thoughtcrime.securesms.util;

import androidx.annotation.MainThread;
import androidx.annotation.NonNull;

import androidx.recyclerview.widget.RecyclerView;

import java.util.HashMap;
import java.util.Map;

/**
 * Useful for generate ID's to be used with
 * {@link RecyclerView.Adapter#getItemId(int)} when you otherwise don't
 * have a good way to generate an ID.
 */
public class StableIdGenerator<E> {

  private final Map<E, Long> keys = new HashMap<>();

  private long index = 1;

  @MainThread
  public long getId(@NonNull E item) {
    if (keys.containsKey(item)) {
      return keys.get(item);
    }

    long key = index++;
    keys.put(item, key);

    return key;
  }
}