rocket_session_gats.rs

use rocket::{error::Error, get, response::content::Html, routes};
use turbopump::nightly::{
    fairing::{config::SameSite, config::SessionConfig, SessionFairing},
    in_memory::InMemory,
    Session,
};

#[derive(Clone, Default)]
struct SessionData {
    hit_counter: u32,
}

#[rocket::main]
async fn main() -> Result<(), Error> {
    rocket::ignite()
        .attach(
            SessionFairing::<SessionData, InMemory<SessionData>>::with_config(SessionConfig {
                max_age: 3600,
                domain: None,
                path: Some("/".to_string()),
                same_site: SameSite::Lax,
                http_only: true,
            }),
        )
        .mount("/", routes![hit_counter])
        .launch()
        .await
}

#[get("/hit-counter")]
async fn hit_counter(session: &Session<SessionData>) -> Html<String> {
    let count = session.tap(|data| {
        data.hit_counter += 1;
        data.hit_counter
    });

    Html(format!(
        r##"
    <h1>Hello, world!</h1>
    <p>You've visited this page {} times</p>"##,
        count
    ))
}