Skip to content

Commit 750b761

Browse files
committed
refactor: rename metrics to prometheus metrics
Signed-off-by: Gustavo Inacio <[email protected]>
1 parent 6d800ed commit 750b761

File tree

4 files changed

+21
-21
lines changed

4 files changed

+21
-21
lines changed

crates/service/src/middleware.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,11 +6,11 @@ mod inject_deployment;
66
mod inject_labels;
77
mod inject_receipt;
88
mod inject_sender;
9-
mod metrics;
9+
mod prometheus_metrics;
1010

1111
pub use inject_allocation::{allocation_middleware, Allocation, AllocationState};
1212
pub use inject_deployment::deployment_middleware;
1313
pub use inject_labels::labels_middleware;
1414
pub use inject_receipt::receipt_middleware;
1515
pub use inject_sender::{sender_middleware, Sender, SenderState};
16-
pub use metrics::MetricsMiddlewareLayer;
16+
pub use prometheus_metrics::PrometheusMetricsMiddlewareLayer;

crates/service/src/middleware/inject_labels.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ use thegraph_core::DeploymentId;
1313
use super::{
1414
inject_allocation::Allocation,
1515
inject_sender::Sender,
16-
metrics::{MetricLabelProvider, MetricLabels},
16+
prometheus_metrics::{MetricLabelProvider, MetricLabels},
1717
};
1818

1919
const NO_DEPLOYMENT_ID: &str = "no-deployment";
@@ -78,7 +78,7 @@ pub async fn labels_middleware(mut request: Request, next: Next) -> Response {
7878
#[cfg(test)]
7979
mod tests {
8080
use crate::middleware::{
81-
inject_allocation::Allocation, inject_sender::Sender, metrics::MetricLabels,
81+
inject_allocation::Allocation, inject_sender::Sender, prometheus_metrics::MetricLabels,
8282
};
8383

8484
use super::labels_middleware;

crates/service/src/middleware/metrics.rs renamed to crates/service/src/middleware/prometheus_metrics.rs

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -21,52 +21,52 @@ pub trait MetricLabelProvider {
2121
}
2222

2323
#[derive(Clone)]
24-
pub struct MetricsMiddleware<S> {
24+
pub struct PrometheusMetricsMiddleware<S> {
2525
inner: S,
2626
histogram: prometheus::HistogramVec,
2727
failure: prometheus::CounterVec,
2828
}
2929

3030
#[derive(Clone)]
31-
pub struct MetricsMiddlewareLayer {
31+
pub struct PrometheusMetricsMiddlewareLayer {
3232
histogram: prometheus::HistogramVec,
3333
failure: prometheus::CounterVec,
3434
}
3535

36-
impl MetricsMiddlewareLayer {
36+
impl PrometheusMetricsMiddlewareLayer {
3737
pub fn new(histogram: prometheus::HistogramVec, failure: prometheus::CounterVec) -> Self {
3838
Self { histogram, failure }
3939
}
4040
}
4141

42-
impl<S> Layer<S> for MetricsMiddlewareLayer {
43-
type Service = MetricsMiddleware<S>;
42+
impl<S> Layer<S> for PrometheusMetricsMiddlewareLayer {
43+
type Service = PrometheusMetricsMiddleware<S>;
4444

4545
fn layer(&self, inner: S) -> Self::Service {
46-
MetricsMiddleware {
46+
PrometheusMetricsMiddleware {
4747
inner,
4848
histogram: self.histogram.clone(),
4949
failure: self.failure.clone(),
5050
}
5151
}
5252
}
5353

54-
impl<S, ReqBody> Service<Request<ReqBody>> for MetricsMiddleware<S>
54+
impl<S, ReqBody> Service<Request<ReqBody>> for PrometheusMetricsMiddleware<S>
5555
where
5656
S: Service<Request<ReqBody>> + Clone + 'static,
5757
ReqBody: 'static,
5858
{
5959
type Response = S::Response;
6060
type Error = S::Error;
61-
type Future = MetricsFuture<S::Future>;
61+
type Future = PrometheusMetricsFuture<S::Future>;
6262

6363
fn poll_ready(&mut self, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
6464
self.inner.poll_ready(cx)
6565
}
6666

67-
fn call(&mut self, request: Request<ReqBody>) -> MetricsFuture<S::Future> {
67+
fn call(&mut self, request: Request<ReqBody>) -> PrometheusMetricsFuture<S::Future> {
6868
let labels = request.extensions().get::<MetricLabels>().cloned();
69-
MetricsFuture {
69+
PrometheusMetricsFuture {
7070
timer: None,
7171
histogram: self.histogram.clone(),
7272
failure: self.failure.clone(),
@@ -77,7 +77,7 @@ where
7777
}
7878

7979
#[pin_project]
80-
pub struct MetricsFuture<F> {
80+
pub struct PrometheusMetricsFuture<F> {
8181
/// Instant at which we started the requst.
8282
timer: Option<HistogramTimer>,
8383

@@ -90,7 +90,7 @@ pub struct MetricsFuture<F> {
9090
fut: F,
9191
}
9292

93-
impl<F, R, E> Future for MetricsFuture<F>
93+
impl<F, R, E> Future for PrometheusMetricsFuture<F>
9494
where
9595
F: Future<Output = Result<R, E>>,
9696
{
@@ -137,7 +137,7 @@ mod tests {
137137
use prometheus::core::Collector;
138138
use tower::{Service, ServiceBuilder, ServiceExt};
139139

140-
use crate::middleware::metrics::{MetricLabels, MetricsMiddlewareLayer};
140+
use crate::middleware::prometheus_metrics::{MetricLabels, PrometheusMetricsMiddlewareLayer};
141141

142142
use super::MetricLabelProvider;
143143

@@ -190,7 +190,7 @@ mod tests {
190190
);
191191

192192
let metrics_layer =
193-
MetricsMiddlewareLayer::new(histogram_metric.clone(), failure_metric.clone());
193+
PrometheusMetricsMiddlewareLayer::new(histogram_metric.clone(), failure_metric.clone());
194194
let mut service = ServiceBuilder::new()
195195
.layer(metrics_layer)
196196
.service_fn(handle);
@@ -219,7 +219,7 @@ mod tests {
219219
);
220220

221221
let metrics_layer =
222-
MetricsMiddlewareLayer::new(histogram_metric.clone(), failure_metric.clone());
222+
PrometheusMetricsMiddlewareLayer::new(histogram_metric.clone(), failure_metric.clone());
223223
let mut service = ServiceBuilder::new()
224224
.layer(metrics_layer)
225225
.service_fn(handle_err);

crates/service/src/service/indexer_service.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ use crate::{
4040
metrics::{HANDLER_FAILURE, HANDLER_HISTOGRAM},
4141
middleware::{
4242
allocation_middleware, deployment_middleware, labels_middleware, receipt_middleware,
43-
sender_middleware, AllocationState, MetricsMiddlewareLayer, SenderState,
43+
sender_middleware, AllocationState, PrometheusMetricsMiddlewareLayer, SenderState,
4444
},
4545
routes::{health, request_handler, static_subgraph_request_handler},
4646
tap::IndexerTapContext,
@@ -382,7 +382,7 @@ pub async fn run(options: IndexerServiceOptions) -> Result<(), anyhow::Error> {
382382
// inject metrics labels
383383
.layer(from_fn(labels_middleware))
384384
// metrics for histogram and failure
385-
.layer(MetricsMiddlewareLayer::new(
385+
.layer(PrometheusMetricsMiddlewareLayer::new(
386386
HANDLER_HISTOGRAM.clone(),
387387
HANDLER_FAILURE.clone(),
388388
));

0 commit comments

Comments
 (0)