forked from Deuxfleurs/garage
Allow building without Prometheus exporter (/metrics endpoint)
prometheus and opentelemetry-prometheus add 7 extra dependencies in total and increases the size of the garage binary by ~7 % (with fat LTO).
This commit is contained in:
parent
e7af006c1c
commit
ea36b9ff90
3 changed files with 38 additions and 24 deletions
|
@ -54,10 +54,11 @@ quick-xml = { version = "0.21", features = [ "serialize" ] }
|
||||||
url = "2.1"
|
url = "2.1"
|
||||||
|
|
||||||
opentelemetry = "0.17"
|
opentelemetry = "0.17"
|
||||||
opentelemetry-prometheus = "0.10"
|
opentelemetry-prometheus = { version = "0.10", optional = true }
|
||||||
opentelemetry-otlp = { version = "0.10", optional = true }
|
opentelemetry-otlp = { version = "0.10", optional = true }
|
||||||
prometheus = "0.13"
|
prometheus = { version = "0.13", optional = true }
|
||||||
|
|
||||||
[features]
|
[features]
|
||||||
k2v = [ "garage_util/k2v", "garage_model/k2v" ]
|
k2v = [ "garage_util/k2v", "garage_model/k2v" ]
|
||||||
|
metrics = [ "opentelemetry-prometheus", "prometheus" ]
|
||||||
telemetry-otlp = ["opentelemetry-otlp"]
|
telemetry-otlp = ["opentelemetry-otlp"]
|
||||||
|
|
|
@ -3,13 +3,14 @@ use std::sync::Arc;
|
||||||
use async_trait::async_trait;
|
use async_trait::async_trait;
|
||||||
|
|
||||||
use futures::future::Future;
|
use futures::future::Future;
|
||||||
use http::header::{
|
use http::header::{ACCESS_CONTROL_ALLOW_METHODS, ACCESS_CONTROL_ALLOW_ORIGIN, ALLOW};
|
||||||
ACCESS_CONTROL_ALLOW_METHODS, ACCESS_CONTROL_ALLOW_ORIGIN, ALLOW, CONTENT_TYPE,
|
|
||||||
};
|
|
||||||
use hyper::{Body, Request, Response};
|
use hyper::{Body, Request, Response};
|
||||||
|
|
||||||
use opentelemetry::trace::{SpanRef, Tracer};
|
use opentelemetry::trace::SpanRef;
|
||||||
|
|
||||||
|
#[cfg(feature = "metrics")]
|
||||||
use opentelemetry_prometheus::PrometheusExporter;
|
use opentelemetry_prometheus::PrometheusExporter;
|
||||||
|
#[cfg(feature = "metrics")]
|
||||||
use prometheus::{Encoder, TextEncoder};
|
use prometheus::{Encoder, TextEncoder};
|
||||||
|
|
||||||
use garage_model::garage::Garage;
|
use garage_model::garage::Garage;
|
||||||
|
@ -25,6 +26,7 @@ use crate::admin::router::{Authorization, Endpoint};
|
||||||
|
|
||||||
pub struct AdminApiServer {
|
pub struct AdminApiServer {
|
||||||
garage: Arc<Garage>,
|
garage: Arc<Garage>,
|
||||||
|
#[cfg(feature = "metrics")]
|
||||||
exporter: PrometheusExporter,
|
exporter: PrometheusExporter,
|
||||||
metrics_token: Option<String>,
|
metrics_token: Option<String>,
|
||||||
admin_token: Option<String>,
|
admin_token: Option<String>,
|
||||||
|
@ -32,7 +34,6 @@ pub struct AdminApiServer {
|
||||||
|
|
||||||
impl AdminApiServer {
|
impl AdminApiServer {
|
||||||
pub fn new(garage: Arc<Garage>) -> Self {
|
pub fn new(garage: Arc<Garage>) -> Self {
|
||||||
let exporter = opentelemetry_prometheus::exporter().init();
|
|
||||||
let cfg = &garage.config.admin;
|
let cfg = &garage.config.admin;
|
||||||
let metrics_token = cfg
|
let metrics_token = cfg
|
||||||
.metrics_token
|
.metrics_token
|
||||||
|
@ -44,7 +45,8 @@ impl AdminApiServer {
|
||||||
.map(|tok| format!("Bearer {}", tok));
|
.map(|tok| format!("Bearer {}", tok));
|
||||||
Self {
|
Self {
|
||||||
garage,
|
garage,
|
||||||
exporter,
|
#[cfg(feature = "metrics")]
|
||||||
|
exporter: opentelemetry_prometheus::exporter().init(),
|
||||||
metrics_token,
|
metrics_token,
|
||||||
admin_token,
|
admin_token,
|
||||||
}
|
}
|
||||||
|
@ -71,22 +73,31 @@ impl AdminApiServer {
|
||||||
}
|
}
|
||||||
|
|
||||||
fn handle_metrics(&self) -> Result<Response<Body>, Error> {
|
fn handle_metrics(&self) -> Result<Response<Body>, Error> {
|
||||||
let mut buffer = vec![];
|
#[cfg(feature = "metrics")]
|
||||||
let encoder = TextEncoder::new();
|
{
|
||||||
|
use opentelemetry::trace::Tracer;
|
||||||
|
|
||||||
let tracer = opentelemetry::global::tracer("garage");
|
let mut buffer = vec![];
|
||||||
let metric_families = tracer.in_span("admin/gather_metrics", |_| {
|
let encoder = TextEncoder::new();
|
||||||
self.exporter.registry().gather()
|
|
||||||
});
|
|
||||||
|
|
||||||
encoder
|
let tracer = opentelemetry::global::tracer("garage");
|
||||||
.encode(&metric_families, &mut buffer)
|
let metric_families = tracer.in_span("admin/gather_metrics", |_| {
|
||||||
.ok_or_internal_error("Could not serialize metrics")?;
|
self.exporter.registry().gather()
|
||||||
|
});
|
||||||
|
|
||||||
Ok(Response::builder()
|
encoder
|
||||||
.status(200)
|
.encode(&metric_families, &mut buffer)
|
||||||
.header(CONTENT_TYPE, encoder.format_type())
|
.ok_or_internal_error("Could not serialize metrics")?;
|
||||||
.body(Body::from(buffer))?)
|
|
||||||
|
Ok(Response::builder()
|
||||||
|
.status(200)
|
||||||
|
.header(http::header::CONTENT_TYPE, encoder.format_type())
|
||||||
|
.body(Body::from(buffer))?)
|
||||||
|
}
|
||||||
|
#[cfg(not(feature = "metrics"))]
|
||||||
|
Err(Error::bad_request(
|
||||||
|
"Garage was built without the metrics feature".to_string(),
|
||||||
|
))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -55,9 +55,9 @@ tokio = { version = "1.0", default-features = false, features = ["rt", "rt-multi
|
||||||
netapp = "0.4"
|
netapp = "0.4"
|
||||||
|
|
||||||
opentelemetry = { version = "0.17", features = [ "rt-tokio" ] }
|
opentelemetry = { version = "0.17", features = [ "rt-tokio" ] }
|
||||||
opentelemetry-prometheus = "0.10"
|
opentelemetry-prometheus = { version = "0.10", optional = true }
|
||||||
opentelemetry-otlp = { version = "0.10", optional = true }
|
opentelemetry-otlp = { version = "0.10", optional = true }
|
||||||
prometheus = "0.13"
|
prometheus = { version = "0.13", optional = true }
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
aws-sdk-s3 = "0.8"
|
aws-sdk-s3 = "0.8"
|
||||||
|
@ -74,9 +74,11 @@ base64 = "0.13"
|
||||||
|
|
||||||
|
|
||||||
[features]
|
[features]
|
||||||
default = [ "bundled-libs" ]
|
default = [ "bundled-libs", "metrics" ]
|
||||||
kubernetes-discovery = [ "garage_rpc/kubernetes-discovery" ]
|
kubernetes-discovery = [ "garage_rpc/kubernetes-discovery" ]
|
||||||
k2v = [ "garage_util/k2v", "garage_api/k2v" ]
|
k2v = [ "garage_util/k2v", "garage_api/k2v" ]
|
||||||
|
# Prometheus exporter (/metrics endpoint).
|
||||||
|
metrics = [ "garage_api/metrics", "opentelemetry-prometheus", "prometheus" ]
|
||||||
# Exporter for the OpenTelemetry Collector.
|
# Exporter for the OpenTelemetry Collector.
|
||||||
telemetry-otlp = [ "opentelemetry-otlp", "garage_api/telemetry-otlp" ]
|
telemetry-otlp = [ "opentelemetry-otlp", "garage_api/telemetry-otlp" ]
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue