pl.allegro.tech.hermes.common.metric.WorkloadMetrics Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hermes-common Show documentation
Fast and reliable message broker built on top of Kafka.
The newest version!
package pl.allegro.tech.hermes.common.metric;

import io.micrometer.core.instrument.Gauge;
import io.micrometer.core.instrument.MeterRegistry;
import io.micrometer.core.instrument.Tags;
import io.micrometer.core.instrument.search.Search;
import java.util.Collection;
import java.util.Set;
import java.util.function.ToDoubleFunction;
import pl.allegro.tech.hermes.metrics.HermesTimer;

public class WorkloadMetrics {

  private static final String CONSUMER_ID_TAG = "consumer-id";
  private static final String KAFKA_CLUSTER_TAG = "kafka-cluster";

  private final MeterRegistry meterRegistry;
  private final GaugeRegistrar gaugeRegistrar;

  WorkloadMetrics(MeterRegistry meterRegistry) {
    this.meterRegistry = meterRegistry;
    this.gaugeRegistrar = new GaugeRegistrar(meterRegistry);
  }

  public  void registerAllAssignmentsGauge(T obj, String kafkaCluster, ToDoubleFunction f) {
    gaugeRegistrar.registerGauge(
        "workload.all-assignments", obj, f, Tags.of(KAFKA_CLUSTER_TAG, kafkaCluster));
  }

  public  void registerMissingResourcesGauge(T obj, String kafkaCluster, ToDoubleFunction f) {
    gaugeRegistrar.registerGauge(
        "workload.missing-resources", obj, f, Tags.of(KAFKA_CLUSTER_TAG, kafkaCluster));
  }

  public  void registerDeletedAssignmentsGauge(
      T obj, String kafkaCluster, ToDoubleFunction f) {
    gaugeRegistrar.registerGauge(
        "workload.deleted-assignments", obj, f, Tags.of(KAFKA_CLUSTER_TAG, kafkaCluster));
  }

  public  void registerCreatedAssignmentsGauge(
      T obj, String kafkaCluster, ToDoubleFunction f) {
    gaugeRegistrar.registerGauge(
        "workload.created-assignments", obj, f, Tags.of(KAFKA_CLUSTER_TAG, kafkaCluster));
  }

  public HermesTimer rebalanceDurationTimer(String kafkaCluster) {
    return HermesTimer.from(
        meterRegistry.timer(
            "workload.rebalance-duration", Tags.of(KAFKA_CLUSTER_TAG, kafkaCluster)));
  }

  public  void registerRunningSubscriptionsGauge(T obj, ToDoubleFunction f) {
    gaugeRegistrar.registerGauge("workload.subscriptions.running", obj, f);
  }

  public  void registerAssignedSubscriptionsGauge(T obj, ToDoubleFunction f) {
    gaugeRegistrar.registerGauge("workload.subscriptions.assigned", obj, f);
  }

  public  void registerMissingSubscriptionsGauge(T obj, ToDoubleFunction f) {
    gaugeRegistrar.registerGauge("workload.subscriptions.missing", obj, f);
  }

  public  void registerOversubscribedGauge(T obj, ToDoubleFunction f) {
    gaugeRegistrar.registerGauge("workload.subscriptions.oversubscribed", obj, f);
  }

  public  void registerOperationsPerSecondGauge(T obj, ToDoubleFunction f) {
    gaugeRegistrar.registerGauge("workload.weighted.ops", obj, f);
  }

  public  void registerCpuUtilizationGauge(T obj, ToDoubleFunction f) {
    gaugeRegistrar.registerGauge("workload.weighted.cpu-utilization", obj, f);
  }

  public  void registerCurrentScoreGauge(String consumerId, T obj, ToDoubleFunction f) {
    gaugeRegistrar.registerGauge(
        "workload.weighted.current-score", obj, f, Tags.of(CONSUMER_ID_TAG, consumerId));
  }

  public  void registerProposedErrorGauge(String consumerId, T obj, ToDoubleFunction f) {
    gaugeRegistrar.registerGauge(
        "workload.weighted.proposed-error", obj, f, Tags.of(CONSUMER_ID_TAG, consumerId));
  }

  public  void registerScoringErrorGauge(String consumerId, T obj, ToDoubleFunction f) {
    gaugeRegistrar.registerGauge(
        "workload.weighted.scoring-error", obj, f, Tags.of(CONSUMER_ID_TAG, consumerId));
  }

  public  void registerCurrentWeightGauge(String consumerId, T obj, ToDoubleFunction f) {
    gaugeRegistrar.registerGauge(
        "workload.weighted.current-weight.ops", obj, f, Tags.of(CONSUMER_ID_TAG, consumerId));
  }

  public  void registerProposedWeightGauge(String consumerId, T obj, ToDoubleFunction f) {
    gaugeRegistrar.registerGauge(
        "workload.weighted.proposed-weight.ops", obj, f, Tags.of(CONSUMER_ID_TAG, consumerId));
  }

  public void unregisterAllWorkloadWeightedGaugesForConsumerIds(Set consumerIds) {
    Collection gauges =
        Search.in(meterRegistry)
            .tag(CONSUMER_ID_TAG, consumerIds::contains)
            .name(s -> s.startsWith("workload.weighted"))
            .gauges();
    for (Gauge gauge : gauges) {
      meterRegistry.remove(gauge);
    }
  }
}