anthrove/bkapi
4
0
Fork 0
mirror of https://github.com/Syfaro/bkapi.git synced 2024-11-05 14:44:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add metric for size of tree.

Browse Source
This commit is contained in:
Syfaro 2022-10-13 13:10:35 -04:00
parent f9a0a21f92
commit fa36cf193d
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
bkapi/src/tree.rs
View File

@ -9,6 +9,8 @@ use tracing_unwrap::ResultExt;
use crate::{Config, Error}; use crate::{Config, Error};
lazy_static::lazy_static! { lazy_static::lazy_static! {
static ref TREE_ENTRIES: prometheus::IntCounter = prometheus::register_int_counter!("bkapi_tree_entries", "Total number of entries within tree").unwrap();
static ref TREE_ADD_DURATION: prometheus::Histogram = prometheus::register_histogram!("bkapi_tree_add_duration_seconds", "Duration to add new item to tree").unwrap(); static ref TREE_ADD_DURATION: prometheus::Histogram = prometheus::register_histogram!("bkapi_tree_add_duration_seconds", "Duration to add new item to tree").unwrap();
static ref TREE_DURATION: prometheus::HistogramVec = prometheus::register_histogram_vec!("bkapi_tree_duration_seconds", "Duration of tree search time", &["distance"]).unwrap(); static ref TREE_DURATION: prometheus::HistogramVec = prometheus::register_histogram_vec!("bkapi_tree_duration_seconds", "Duration of tree search time", &["distance"]).unwrap();
} }
@ -64,6 +66,9 @@ impl Tree {
let mut tree = self.tree.write().await; let mut tree = self.tree.write().await;
*tree = new_tree; *tree = new_tree;
TREE_ENTRIES.reset();
TREE_ENTRIES.inc_by(count);
Ok(()) Ok(())
} }
@ -80,6 +85,7 @@ impl Tree {
if is_new_hash { if is_new_hash {
let mut tree = self.tree.write().await; let mut tree = self.tree.write().await;
tree.add(node); tree.add(node);
TREE_ENTRIES.inc();
} }
tracing::info!(is_new_hash, "added hash"); tracing::info!(is_new_hash, "added hash");