/*
 *  Copyright (c) 2018-present, Facebook, Inc.
 *
 *  This source code is licensed under the MIT license found in the LICENSE
 *  file in the root directory of this source tree.
 *
 */

package com.facebook.flipper.plugins.inspector;

import java.lang.ref.WeakReference;
import java.util.HashMap;
import java.util.Map;
import javax.annotation.Nullable;

public class ObjectTracker {
  ObjectTracker() {}

  private final Map<String, WeakReference<Object>> mObjects = new HashMap<>();

  void put(String id, Object obj) {
    mObjects.put(id, new WeakReference<>(obj));
  }

  @Nullable
  public Object get(String id) {
    final WeakReference<Object> weakObj = mObjects.get(id);
    if (weakObj == null) {
      return null;
    }

    final Object obj = weakObj.get();
    if (obj == null) {
      mObjects.remove(id);
    }

    return obj;
  }

  void clear() {
    mObjects.clear();
  }

  boolean contains(String id) {
    return mObjects.containsKey(id);
  }
}

/*
 *  Copyright (c) 2018-present, Facebook, Inc.
 *
 *  This source code is licensed under the MIT license found in the LICENSE
 *  file in the root directory of this source tree.
 *
 */

package com.facebook.flipper.plugins.inspector;

import java.lang.ref.SoftReference;
import java.util.HashMap;
import java.util.Map;
import javax.annotation.Nullable;

public class ObjectTracker {
  ObjectTracker() {}

  private final Map<String, SoftReference<Object>> mObjects = new HashMap<>();

  void put(String id, Object obj) {
    mObjects.put(id, new SoftReference<>(obj));
  }

  @Nullable
  public Object get(String id) {
    final SoftReference<Object> softRef = mObjects.get(id);
    if (softRef == null) {
      return null;
    }

    final Object obj = softRef.get();
    if (obj == null) {
      mObjects.remove(id);
    }

    return obj;
  }

  void clear() {
    mObjects.clear();
  }

  boolean contains(String id) {
    return mObjects.containsKey(id);
  }
}

Use SoftReference in ObjectTracker