5 releases (3 breaking)

0.4.0 Sep 2, 2021
0.3.0 Sep 2, 2021
0.2.1 Aug 26, 2021
0.2.0 Aug 26, 2021
0.1.0 Jul 14, 2021

#191 in Profiling

Apache-2.0

26KB
440 lines

Metrics Catalogue

github crates.io docs.rs build status

This library provides a way to automatically derive a catalogue of metrics from a struct definition. It generates a hierarchical set of modules & metric keys following the same hierarchy from the root struct. This allows users to interact with the metrics without the risk of typos or having to maintain an external list of available metric keys. Additionally, the metrics can also be interacted with directly, without using the metrics framework.

[dependencies]
metrics-catalogue = "0.1"

Example

use metric_catalogue::{Metrics, Counter, Gauge};

#[derive(Catalogue)]
#[metric(root)]
struct Foo {
  my_counter: Counter,
  my_gauge: Gauge,
  my_bar: Bar,
}

#[derive(Catalogue)]
struct Bar {
  my_counter: Counter,
}

generates the following catalogue of metric keys:

#[allow(non_camel_case_types)]
pub mod catalogue {
  pub const MY_COUNTER: &str = "my_counter";
  pub const MY_GAUGE: &str = "my_gauge";
  
  #[allow(non_camel_case_types)]
  pub mod bar {
    pub const MY_COUNTER: &str = "bar.my_counter";
  }
}

allowing updates to the relevant metrics without potential typos:

fn my_function() {
  metrics::increment_counter!(catalogue::my_sub::MY_COUNTER);
}

Supported metric types

Currently, the following metric types are supported:

  • Counter A monotonically incrementing AtomicU64 metric
  • Gauge An AtomicU64 based metric allowing arbitrary updated, increments & decrements using real units.
  • DiscreteGauge An AtomicU64 based metric allowing arbitrary updated, increments & decrements using discrete units.

Metrics Registry

Adding the Catalogue derivation will implement the Registry trait for all relevant structs:

pub trait Registry {
    /// Find a registered counter under the provided name
    fn find_counter(&self, name: &str) -> Option<&Counter>;
    /// Find a registered gauge under the provided name
    fn find_gauge(&self, name: &str) -> Option<&Gauge>;
}

allowing an automatic hierarchical look-up of the generated catalogue.

Metrics recorder

The Catalogue derivation will also implement the Recorder trait for the root struct:

impl Recorder for Foo {
        fn register_counter(&self, _key: &Key, _unit: Option<Unit>, _desc: Option<&'static str>) {}
        fn register_gauge(&self, _key: &Key, _unit: Option<Unit>, _desc: Option<&'static str>) {}
        fn register_histogram(&self, _key: &Key, _unit: Option<Unit>, _desc: Option<&'static str>) {}
        fn record_histogram(&self, _key: &Key, _value: f64) {}
        fn increment_counter(&self, key: &Key, value: u64) {
            if let Some(metric) = self.find_counter(key.name()) {
                metric.increment(value);
            }
        }
        fn update_gauge(&self, key: &Key, value: GaugeValue) {
            if let Some(metric) = self.find_gauge(key.name()) {
                match value {
                    GaugeValue::Increment(val) => metric.increase(val),
                    GaugeValue::Decrement(val) => metric.decrease(val),
                    GaugeValue::Absolute(val) => metric.set(val),
                }
            }
        }
    }

Details

  • A single root structure must be declared by using the root attribute. Without a root, no catalogue will be generated.

  • fields can be hidden from the catalogue by using the skip attribute. e.g.

    #[derive(Catalogue)]
    struct Foo {
      #[metric(skip)]
      my_hidden_field: Counter,
    }
    

    which will prevent generating the MY_HIDDEN_FIELD -> "my_hidden_field" key & the associated registry entries.

  • the Catalogue macro is limited to structs only.

License

Copyright 2021 Sam De Roeck

Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at

http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License.

Dependencies

~5–16MB
~187K SLoC