rama 0.3.0-rc1

modular service framework
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

//! An example to showcase how you can classify the User-Agent of incoming requests.
//!
//! # Run the example
//!
//! ```sh
//! cargo run --example http_user_agent_classifier --features=http-full
//! ```
//!
//! # Expected output
//!
//! The server will start and listen on `:62015`. You can use `curl` to interact with the service:
//!
//! ```sh
//! curl -v http://127.0.0.1:62015
//! ```
//!
//! You should see a response with `HTTP/1.1 200 OK` and a JSON body with the user agent info exposed by Rama.

#![expect(
    clippy::unwrap_used,
    reason = "example/test/bench: panic-on-error and print-for-output are the standard patterns for demos and harnesses"
)]

use rama::{
    Layer,
    extensions::Extensions,
    http::{
        HeaderName,
        layer::error_handling::ErrorHandlerLayer,
        server::HttpServer,
        service::web::{
            IntoEndpointService,
            response::{IntoResponse, Json},
        },
    },
    ua::{UserAgent, layer::classifier::UserAgentClassifierLayer},
};

use serde_json::json;

#[tokio::main]
async fn main() {
    HttpServer::default()
        .listen(
            "127.0.0.1:62015",
            (
                UserAgentClassifierLayer::new()
                    .with_overwrite_header(HeaderName::from_static("x-proxy-ua")),
                ErrorHandlerLayer::new(),
            )
                .into_layer(handle.into_endpoint_service()),
        )
        .await
        .unwrap();
}

async fn handle(ext: Extensions) -> impl IntoResponse {
    let ua = ext.get_ref::<UserAgent>().unwrap();
    Json(json!({
        "ua": ua.header_str(),
        "kind": ua.info().map(|info| info.kind.to_string()),
        "version": ua.info().and_then(|info| info.version),
        "platform": ua.platform().map(|p| p.to_string()),
        "http_agent": ua.http_agent().as_ref().map(ToString::to_string),
        "tls_agent": ua.tls_agent().as_ref().map(ToString::to_string),
    }))
}