1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

use async_graphql::Object;

use crate::{
    api::schema::metrics::{self, MetricsFilter},
    event::Metric,
};

#[derive(Debug, Clone)]
pub struct GenericSinkMetrics(Vec<Metric>);

impl GenericSinkMetrics {
    pub const fn new(metrics: Vec<Metric>) -> Self {
        Self(metrics)
    }
}

#[Object]
impl GenericSinkMetrics {
    /// Total received events for the current sink
    pub async fn received_events_total(&self) -> Option<metrics::ReceivedEventsTotal> {
        self.0.received_events_total()
    }

    /// Total sent bytes for the current sink
    pub async fn sent_bytes_total(&self) -> Option<metrics::SentBytesTotal> {
        self.0.sent_bytes_total()
    }

    /// Total sent events for the current sink
    pub async fn sent_events_total(&self) -> Option<metrics::SentEventsTotal> {
        self.0.sent_events_total()
    }
}