Draft: feat: bump versions and add basic OTLP metrics

Cargo.toml

@@ -6,18 +6,23 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
-async-graphql = { version = "6.0.11", features = ["bson", "chrono", "uuid", "log"] }
-async-graphql-axum = "6.0.11"
-tokio = { version = "1.8", features = ["macros", "rt-multi-thread"] }
-axum = { version = "0.6.0", features = ["headers", "macros"] }
-mongodb = "2.8.0"
-serde = "1.0.193"
-futures = "0.3.30"
-bson = "2.8.1"
-clap = { version = "4.4.13", features = ["derive"] }
-uuid = { version = "1.6.1", features = ["v4", "serde"] }
+async-graphql = { version = "7.0.16", features = ["bson", "chrono", "uuid", "log"] }
+async-graphql-axum = "7.0.16"
+tokio = { version = "1.44.2", features = ["macros", "rt-multi-thread"] }
+axum = { version = "0.8.3", features = ["macros"] }
+mongodb = "2.8.2"
+serde = "1.0.219"
+futures = "0.3.31"
+bson = "2.14.0"
+clap = { version = "4.5.37", features = ["derive"] }
+uuid = { version = "1.16.0", features = ["v4", "serde"] }
 mongodb-cursor-pagination = "0.3.2"
 json = "0.12.4"
-log = "0.4.20"
-simple_logger = "4.3.3"
-serde_json = "1.0.113"
+log = "0.4.27"
+simple_logger = "5.0.0"
+serde_json = "1.0.140"
+opentelemetry = "0.30.0"
+opentelemetry_sdk = { version = "0.30.0", features = ["rt-tokio"]}
+opentelemetry-otlp = "0.30.0"
+axum-otel-metrics = { version = "0.12.0" }
+once_cell = "1.21.3"

src/main.rs

@@ -11,7 +11,7 @@ use axum::{
     http::{header::HeaderMap, StatusCode},
     response::{self, IntoResponse},
     routing::{get, post},
-    Router, Server,
+    Router,
 };
 use clap::{arg, command, Parser};
 
@@ -22,6 +22,15 @@ use mongodb::{bson::DateTime, options::ClientOptions, Client, Collection, Databa
 
 use bson::Uuid;
 
+use once_cell::sync::Lazy;
+use axum_otel_metrics::HttpMetricsLayerBuilder;
+use axum_otel_metrics::HttpMetricsLayer;
+
+use opentelemetry::global;
+use opentelemetry_sdk::metrics::{PeriodicReader, SdkMeterProvider, Temporality};
+use opentelemetry_sdk::Resource;
+use opentelemetry_otlp::WithExportConfig;
+
 mod authorization;
 use authorization::AuthorizedUserHeader;
 
@@ -125,6 +134,37 @@ async fn graphql_handler(
     schema.execute(request).await.into()
 }
 
+static RESOURCE: Lazy<Resource> = Lazy::new(|| {
+    Resource::builder()
+        .with_service_name("wishlist")
+        .build()
+});
+
+/// Initializes OpenTelemetry metrics exporter and sets the global meter provider.
+fn init_otlp() -> HttpMetricsLayer {
+    let exporter = opentelemetry_otlp::MetricExporter::builder()
+        .with_http()
+        .with_endpoint("http://otel-collector:4318/v1/metrics")
+        .with_temporality(Temporality::default())
+        .build()
+        .unwrap();
+
+    let reader = PeriodicReader::builder(exporter)
+        .with_interval(std::time::Duration::from_secs(5))
+        .build();
+
+    let provider = SdkMeterProvider::builder()
+        .with_reader(reader)
+        .with_resource(RESOURCE.clone())
+        .build();
+
+    global::set_meter_provider(provider.clone());
+
+    HttpMetricsLayerBuilder::new()
+        .with_provider(provider.clone())
+        .build()
+}
+
 /// Starts wishlist service on port 8000.
 async fn start_service() {
     let client = db_connection().await;
@@ -141,11 +181,17 @@ async fn start_service() {
         .route("/health", get(StatusCode::OK))
         .with_state(schema);
     let dapr_router = build_dapr_router(db_client).await;
-    let app = Router::new().merge(graphiql).merge(dapr_router);
+    let metrics = init_otlp();
+
+    let app = Router::new()
+        .merge(graphiql)
+        .merge(dapr_router)
+        .layer(metrics);
 
     info!("GraphiQL IDE: http://0.0.0.0:8080");
-    Server::bind(&"0.0.0.0:8080".parse().unwrap())
-        .serve(app.into_make_service())
+
+    let listener = tokio::net::TcpListener::bind("0.0.0.0:8080").await.unwrap();
+    axum::serve(listener, app)
         .await
         .unwrap();
 }