io.opentelemetry.sdk.metrics.internal.view.AdviceAttributesProcessor Maven / Gradle / Ivy

Go to download
/*
 * Copyright The OpenTelemetry Authors
 * SPDX-License-Identifier: Apache-2.0
 */

package io.opentelemetry.sdk.metrics.internal.view;

import io.opentelemetry.api.common.AttributeKey;
import io.opentelemetry.api.common.Attributes;
import io.opentelemetry.api.common.AttributesBuilder;
import io.opentelemetry.context.Context;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

final class AdviceAttributesProcessor extends AttributesProcessor {

  private final Set> attributeKeys;

  AdviceAttributesProcessor(List> adviceAttributeKeys) {
    this.attributeKeys = new HashSet<>(adviceAttributeKeys);
  }

  @Override
  public Attributes process(Attributes incoming, Context context) {
    // Exit early to avoid allocations if the incoming attributes do not have extra keys to be
    // filtered
    if (!hasExtraKeys(incoming)) {
      return incoming;
    }
    AttributesBuilder builder = incoming.toBuilder();
    builder.removeIf(key -> !attributeKeys.contains(key));
    return builder.build();
  }

  /** Returns true if {@code attributes} has keys not contained in {@link #attributeKeys}. */
  private boolean hasExtraKeys(Attributes attributes) {
    boolean[] result = {false};
    attributes.forEach(
        (key, value) -> {
          if (!result[0] && !attributeKeys.contains(key)) {
            result[0] = true;
          }
        });
    return result[0];
  }

  @Override
  public boolean usesContext() {
    return false;
  }

  @Override
  public String toString() {
    return "AdviceAttributesProcessor{attributeKeys=" + attributeKeys + '}';
  }
}