chore: add server example (#2)

Author: andylokandy

Cargo.lock

@@ -17,5 +17,7 @@ fastrace = "0.7"
 reqwest = "0.12"
 
 [dev-dependencies]
+axum = "0.8"
 fastrace = { version = "0.7", features = ["enable"] }
+fastrace-axum = { git = "https://github.com/fast/fastrace-axum" }
 tokio = { version = "1.44", features = ["full"] }

Cargo.toml

@@ -1,49 +1,22 @@
-//! Example demonstrating how to use fastrace-reqwest to propagate trace context in HTTP client
-//! requests.
-//!
-//! This example shows:
-//! 1. Setting up a fastrace reporter.
-//! 2. Creating a root span.
-//! 3. Sending an HTTP request with trace context propagation.
-//!
-//! To run this example:
-//! First start the server example: `cargo run --example server`
-//! Then run this client: `cargo run --example client`
-
 use fastrace::collector::Config;
 use fastrace::collector::ConsoleReporter;
-use fastrace::prelude::*;
-use fastrace_reqwest::traceparent_headers;
-use reqwest::Client;
 
 #[tokio::main]
 async fn main() {
-    // Initialize fastrace with the console reporter.
+    // Configurate fastrace reporter.
     fastrace::set_reporter(ConsoleReporter, Config::default());
 
-    {
-        // Create a root span for the client operation.
-        let root = Span::root("client".to_string(), SpanContext::random());
-        let _g = root.set_local_parent();
-
-        send_request().await;
-    }
+    let app = axum::Router::new()
+        .route("/ping", axum::routing::get(ping))
+        // Add a the FastraceLayer to routes.
+        // The layer extracts trace context from incoming requests.
+        .layer(fastrace_axum::FastraceLayer);
 
-    // Flush any remaining traces before the program exits.
-    fastrace::flush();
+    let listener = tokio::net::TcpListener::bind("0.0.0.0:8080").await.unwrap();
+    axum::serve(listener, app).await.unwrap();
 }
 
-/// Send an HTTP request to the server with trace context propagation.
-/// The traceparent_headers() function adds the trace context to the request headers.
-#[fastrace::trace]
-async fn send_request() {
-    let client = Client::new();
-    let response = client
-        .get("http://localhost:8080/ping")
-        .headers(traceparent_headers())
-        .send()
-        .await
-        .unwrap();
-    assert!(response.status().is_success());
-    println!("{:?}", response.text().await.unwrap());
+#[fastrace::trace] // Trace individual handlers.
+async fn ping() -> &'static str {
+    "pong"
 }

examples/client.rs

@@ -0,0 +1,22 @@
+use fastrace::collector::Config;
+use fastrace::collector::ConsoleReporter;
+
+#[tokio::main]
+async fn main() {
+    // Initialize fastrace with the console reporter.
+    fastrace::set_reporter(ConsoleReporter, Config::default());
+
+    let app = axum::Router::new()
+        .route("/ping", axum::routing::get(ping))
+        // Add a the FastraceLayer to routes.
+        // The layer extracts trace context from incoming requests.
+        .layer(fastrace_axum::FastraceLayer);
+
+    let listener = tokio::net::TcpListener::bind("0.0.0.0:8080").await.unwrap();
+    axum::serve(listener, app).await.unwrap();
+}
+
+#[fastrace::trace]
+async fn ping() -> &'static str {
+    "pong"
+}