trace_id

A lightweight, high-performance trace ID library for Rust applications, designed for seamless integration with web frameworks and the tracing ecosystem. Currently supports Axum with Actix Web support planned for future releases.

🚀 Features

• Zero-overhead trace ID generation - Optimized for high-performance applications
• Web framework integration - Currently supports Axum, with Actix Web planned
• Seamless Axum integration - Drop-in middleware with extractor support
• Automatic request correlation - Links all logs within a request lifecycle
• Header-based propagation - Supports x-trace-id header for distributed tracing
• Thread-safe context management - Built on tokio::task_local! for async safety
• Tracing ecosystem integration - Automatic span attachment for structured logging
• Framework-agnostic core - Extensible architecture for future framework support

📦 Installation

Add this to your Cargo.toml:

[dependencies]
trace_id = "0.1.1"

# For Axum integration
trace_id = { version = "0.1.1", features = ["axum"] }

🎯 Quick Start

Basic Axum Integration

use axum::{routing::get, Router};
use trace_id::{TraceId, TraceIdLayer};
use tracing::info;

async fn handler(trace_id: TraceId) -> String {
    info!("Processing request"); // Automatically includes trace_id
    format!("Hello! Trace ID: {}", trace_id)
}

#[tokio::main]
async fn main() {
    let app = Router::new()
        .route("/", get(handler))
        .layer(TraceIdLayer::new());
    
    let listener = tokio::net::TcpListener::bind("0.0.0.0:3000")
        .await
        .unwrap();
    
    axum::serve(listener, app).await.unwrap();
}

Manual Trace ID Access

use trace_id::get_trace_id;

async fn business_logic() {
    let trace_id = get_trace_id();
    
    // Use trace_id for external system integration
    external_api_call(&trace_id).await;
    
    tracing::info!("Business logic completed");
}

📚 Documentation

Core Components

TraceId

A lightweight wrapper around a high-performance trace identifier that represents a unique trace ID compliant with W3C TraceContext specification.

use trace_id::TraceId;

// Generate a new trace ID
let trace_id = TraceId::new();

// Parse from string (with validation)
let trace_id = TraceId::from_string_validated("550e8400-e29b-41d4-a716-446655440000")?;

// Convert to string
let id_string = trace_id.to_string();

TraceIdLayer

Axum middleware that automatically manages trace ID lifecycle:

• Extracts trace ID from x-trace-id header
• Generates new ID if header is missing
• Adds trace ID to response headers
• Sets up tracing context for the request

use axum::Router;
use trace_id::TraceIdLayer;

let app = Router::new()
    .route("/api/users", get(get_users))
    .layer(TraceIdLayer::new());

Context Management

Access the current trace ID from anywhere in your async call stack:

use trace_id::get_trace_id;

async fn deep_function() {
    let trace_id = get_trace_id();
    tracing::info!(trace_id = %trace_id, "Deep in the call stack");
}

🔧 Advanced Usage

Proper Tracing Configuration

To see trace_id in your logs, you must properly configure your tracing subscriber:

use tracing_subscriber::{EnvFilter, fmt, prelude::*};

// Basic configuration
tracing_subscriber::fmt::init();

// Or more detailed configuration
tracing_subscriber::registry()
    .with(fmt::layer())
    .with(EnvFilter::from_default_env())
    .init();

Without proper tracing subscriber configuration, the trace_id will not appear in your logs, even though it's correctly propagated through the context.

Custom Header Names (Planned Feature)

use trace_id::TraceIdLayer;

// Use custom header name (planned for future release)
// let layer = TraceIdLayer::with_header("x-request-id");

Error Handling

use trace_id::TraceId;

match TraceId::from_string_validated("invalid-trace-id") {
    Some(trace_id) => println!("Valid: {}", trace_id),
    None => println!("Invalid trace ID format"),
}

🏗️ Architecture

┌─────────────────┐    ┌──────────────────┐    ┌─────────────────┐
│   HTTP Request  │───▶│  TraceIdLayer    │───▶│   Your Handler  │
│  (x-trace-id)   │    │  - Extract/Gen   │    │  (TraceId)      │
└─────────────────┘    │  - Set Context   │    └─────────────────┘
                       │  - Add to Span   │
                       └──────────────────┘
                                │
                                ▼
                       ┌──────────────────┐
                       │  Tracing Context │
                       │  - All logs      │
                       │  - Automatic ID  │
                       └──────────────────┘

🚦 Examples

Check out the examples directory for comprehensive usage patterns. See EXAMPLES.md for detailed instructions on running and testing the examples.

Available examples:

• tracing_example.rs - Basic Axum integration with TraceId extractor
• tracing_configurations.rs - Different tracing subscriber configurations

🧪 Testing

# Run all tests
cargo test

# Run with Axum features
cargo test --features axum

# Run benchmarks
cargo bench

# Check code quality
cargo clippy --all-features -- -D warnings
cargo fmt --check

📊 Performance

Optimized for high-throughput applications:

• ID Generation: ~50ns per ID
• String Validation: ~100ns per validation
• Context Access: ~10ns per access
• Memory Overhead: 16 bytes per trace ID

Run benchmarks with cargo bench to see performance on your system.

🤝 Contributing

We welcome contributions! Please see our Contributing Guide for details.

Development Setup

git clone https://github.com/OpenMindOpenWorld/trace_id.git
cd trace_id
cargo test --all-features

📄 License

This project is licensed under either of

• Apache License, Version 2.0, (LICENSE-APACHE or http://www.apache.org/licenses/LICENSE-2.0)
• MIT license (LICENSE or http://opensource.org/licenses/MIT)

at your option.

🔗 Related Projects

• tracing - Structured logging and diagnostics
• axum - Ergonomic web framework (currently supported)
• actix-web - Powerful web framework (planned support)