reth_provider/providers/database/
metrics.rs

1use metrics::Histogram;
2use reth_metrics::Metrics;
3use std::time::{Duration, Instant};
4
5#[derive(Debug)]
6pub(crate) struct DurationsRecorder {
7    start: Instant,
8    current_metrics: DatabaseProviderMetrics,
9    pub(crate) actions: Vec<(Action, Duration)>,
10    latest: Option<Duration>,
11}
12
13impl Default for DurationsRecorder {
14    fn default() -> Self {
15        Self {
16            start: Instant::now(),
17            actions: Vec::new(),
18            latest: None,
19            current_metrics: DatabaseProviderMetrics::default(),
20        }
21    }
22}
23
24impl DurationsRecorder {
25    /// Records the duration since last record, saves it for future logging and instantly reports as
26    /// a metric with `action` label.
27    pub(crate) fn record_relative(&mut self, action: Action) {
28        let elapsed = self.start.elapsed();
29        let duration = elapsed - self.latest.unwrap_or_default();
30
31        self.actions.push((action, duration));
32        self.current_metrics.record_duration(action, duration);
33        self.latest = Some(elapsed);
34    }
35}
36
37#[derive(Debug, Copy, Clone)]
38pub(crate) enum Action {
39    InsertBlock,
40    InsertState,
41    InsertHashes,
42    InsertHistoryIndices,
43    UpdatePipelineStages,
44    InsertHeaderNumbers,
45    InsertBlockBodyIndices,
46    InsertTransactionBlocks,
47    GetNextTxNum,
48    GetParentTD,
49}
50
51/// Database provider metrics
52#[derive(Metrics)]
53#[metrics(scope = "storage.providers.database")]
54struct DatabaseProviderMetrics {
55    /// Duration of insert block
56    insert_block: Histogram,
57    /// Duration of insert state
58    insert_state: Histogram,
59    /// Duration of insert hashes
60    insert_hashes: Histogram,
61    /// Duration of insert history indices
62    insert_history_indices: Histogram,
63    /// Duration of update pipeline stages
64    update_pipeline_stages: Histogram,
65    /// Duration of insert canonical headers
66    /// Duration of insert header numbers
67    insert_header_numbers: Histogram,
68    /// Duration of insert block body indices
69    insert_block_body_indices: Histogram,
70    /// Duration of insert transaction blocks
71    insert_tx_blocks: Histogram,
72    /// Duration of get next tx num
73    get_next_tx_num: Histogram,
74    /// Duration of get parent TD
75    get_parent_td: Histogram,
76}
77
78impl DatabaseProviderMetrics {
79    /// Records the duration for the given action.
80    pub(crate) fn record_duration(&self, action: Action, duration: Duration) {
81        match action {
82            Action::InsertBlock => self.insert_block.record(duration),
83            Action::InsertState => self.insert_state.record(duration),
84            Action::InsertHashes => self.insert_hashes.record(duration),
85            Action::InsertHistoryIndices => self.insert_history_indices.record(duration),
86            Action::UpdatePipelineStages => self.update_pipeline_stages.record(duration),
87            Action::InsertHeaderNumbers => self.insert_header_numbers.record(duration),
88            Action::InsertBlockBodyIndices => self.insert_block_body_indices.record(duration),
89            Action::InsertTransactionBlocks => self.insert_tx_blocks.record(duration),
90            Action::GetNextTxNum => self.get_next_tx_num.record(duration),
91            Action::GetParentTD => self.get_parent_td.record(duration),
92        }
93    }
94}