io.opentelemetry.instrumentation.logback.mdc.v1_0.internal.UnionMap Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of opentelemetry-logback-mdc-1.0 Show documentation
Instrumentation of Java libraries using OpenTelemetry.
There is a newer version: 2.8.0-alpha
Show newest version
/*
 * Copyright The OpenTelemetry Authors
 * SPDX-License-Identifier: Apache-2.0
 */

package io.opentelemetry.instrumentation.logback.mdc.v1_0.internal;

import java.io.Serializable;
import java.util.AbstractMap;
import java.util.AbstractSet;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.Iterator;
import java.util.LinkedHashSet;
import java.util.Map;
import java.util.Set;

/**
 * An immutable view over two maps, with keys resolving from the first map first, or otherwise the
 * second if not present in the first.
 *
 * This class is internal and is hence not for public use. Its APIs are unstable and can change
 * at any time.
 */
public final class UnionMap extends AbstractMap implements Serializable {
  private static final long serialVersionUID = 1L;

  private final Map first;
  private final Map second;
  private int size = -1;
  private Set> entrySet;

  public UnionMap(Map first, Map second) {
    this.first = first;
    this.second = second;
  }

  @Override
  public int size() {
    if (size >= 0) {
      return size;
    }

    Map a;
    Map b;
    if (first.size() >= second.size()) {
      a = first;
      b = second;
    } else {
      a = second;
      b = first;
    }

    int size = a.size();
    if (!b.isEmpty()) {
      for (K k : b.keySet()) {
        if (!a.containsKey(k)) {
          size++;
        }
      }
    }

    return this.size = size;
  }

  @Override
  public boolean isEmpty() {
    return first.isEmpty() && second.isEmpty();
  }

  @Override
  public boolean containsKey(Object key) {
    return first.containsKey(key) || second.containsKey(key);
  }

  @Override
  public boolean containsValue(Object value) {
    return first.containsValue(value) || second.containsValue(value);
  }

  @Override
  public V get(Object key) {
    V value = first.get(key);
    return value != null ? value : second.get(key);
  }

  @Override
  public V put(K key, V value) {
    throw new UnsupportedOperationException();
  }

  @Override
  public V remove(Object key) {
    throw new UnsupportedOperationException();
  }

  @Override
  public void clear() {
    throw new UnsupportedOperationException();
  }

  @Override
  public Set> entrySet() {
    if (entrySet != null) {
      return entrySet;
    }

    // Check for dupes first to reduce allocations on the vastly more common case where there aren't
    // any.
    boolean secondHasDupes = false;
    for (Entry entry : second.entrySet()) {
      if (first.containsKey(entry.getKey())) {
        secondHasDupes = true;
        break;
      }
    }

    Set> filteredSecond;
    if (!secondHasDupes) {
      filteredSecond = second.entrySet();
    } else {
      filteredSecond = new LinkedHashSet<>();
      for (Entry entry : second.entrySet()) {
        if (!first.containsKey(entry.getKey())) {
          filteredSecond.add(entry);
        }
      }
    }
    return entrySet =
        Collections.unmodifiableSet(new ConcatenatedSet<>(first.entrySet(), filteredSecond));
  }

  private Object writeReplace() {
    // serialize this object as HashMap
    return new HashMap<>(this);
  }

  // Member sets must be deduped by caller.
  static final class ConcatenatedSet extends AbstractSet {

    private final Set first;
    private final Set second;

    private final int size;

    ConcatenatedSet(Set first, Set second) {
      this.first = first;
      this.second = second;

      size = first.size() + second.size();
    }

    @Override
    public int size() {
      return size;
    }

    @Override
    public boolean add(T t) {
      throw new UnsupportedOperationException();
    }

    @Override
    public boolean remove(Object o) {
      throw new UnsupportedOperationException();
    }

    @Override
    public boolean addAll(Collection c) {
      throw new UnsupportedOperationException();
    }

    @Override
    public boolean retainAll(Collection c) {
      throw new UnsupportedOperationException();
    }

    @Override
    public void clear() {
      throw new UnsupportedOperationException();
    }

    @Override
    public Iterator iterator() {
      return new ConcatenatedSetIterator();
    }

    class ConcatenatedSetIterator implements Iterator {
      final Iterator firstItr = first.iterator();
      final Iterator secondItr = second.iterator();

      ConcatenatedSetIterator() {}

      @Override
      public boolean hasNext() {
        return firstItr.hasNext() || secondItr.hasNext();
      }

      @Override
      public T next() {
        if (firstItr.hasNext()) {
          return firstItr.next();
        }
        return secondItr.next();
      }

      @Override
      public void remove() {
        throw new UnsupportedOperationException();
      }
    }
  }
}