Skip to content

Commit

Permalink
Add creating global metrics recorder in file
Browse files Browse the repository at this point in the history
  • Loading branch information
danielle-tfh committed Oct 23, 2024
1 parent e34d687 commit 1200f53
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 16 deletions.
5 changes: 1 addition & 4 deletions deploy/stage/common-values-iris-mpc.yaml
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
image: "ghcr.io/worldcoin/iris-mpc:v0.8.27"
image: "ghcr.io/worldcoin/iris-mpc:v0.8.28"

environment: stage
replicaCount: 1
Expand Down Expand Up @@ -78,6 +78,3 @@ libsDir:
file: "libcublasLt.so.12.2.5.6"
- path: "/usr/local/cuda-12.2/targets/x86_64-linux/lib/libcublas.so.12.2.5.6"
file: "libcublas.so.12.2.5.6"

metrics:
enabled: true
23 changes: 11 additions & 12 deletions iris-mpc/src/bin/server.rs
Original file line number Diff line number Diff line change
Expand Up @@ -34,16 +34,14 @@ use iris_mpc_gpu::{
},
};
use iris_mpc_store::{Store, StoredIrisRef};
use metrics_exporter_statsd::StatsdBuilder;
use std::{
collections::HashMap,
mem,
sync::{Arc, LazyLock, Mutex},
time::{Duration, Instant},
};
use telemetry_batteries::{
metrics::statsd::StatsdBattery,
tracing::{datadog::DatadogBattery, TracingShutdownHandle},
};
use telemetry_batteries::tracing::{datadog::DatadogBattery, TracingShutdownHandle};
use tokio::{
sync::{mpsc, oneshot, Semaphore},
task::spawn_blocking,
Expand Down Expand Up @@ -418,15 +416,16 @@ fn initialize_tracing(config: &Config) -> eyre::Result<TracingShutdownHandle> {
);

if let Some(metrics_config) = &service.metrics {
StatsdBattery::init(
&metrics_config.host,
metrics_config.port,
metrics_config.queue_size,
metrics_config.buffer_size,
Some(&metrics_config.prefix),
)?;
tracing::info!("Initializing metrics using config...");
let recorder = StatsdBuilder::from(&metrics_config.host, metrics_config.port)
.with_queue_size(metrics_config.queue_size)
.with_buffer_size(metrics_config.buffer_size)
.histogram_is_distribution()
.build(Some(&metrics_config.prefix))?;

metrics::set_global_recorder(recorder)?;
}

metrics::counter!("metrics.test").increment(1);
Ok(tracing_shutdown_handle)
} else {
tracing_subscriber::registry()
Expand Down

0 comments on commit 1200f53

Please sign in to comment.