mirror of
https://github.com/tokio-rs/axum.git
synced 2025-01-11 12:31:25 +01:00
Remove sessions example
There are better libraries these days for doing sessions with axum. So think this example was just causing more confusion than helping.
This commit is contained in:
David Pedersen
parent
adb387a370
commit
b1b0e45683
2 changed files with 0 additions and 172 deletions
|
|
@ -1,15 +0,0 @@
|
|||
[package]
|
||||
name = "example-sessions"
|
||||
version = "0.1.0"
|
||||
edition = "2021"
|
||||
publish = false
|
||||
|
||||
[dependencies]
|
||||
async-session = "3.0.0"
|
||||
axum = { path = "../../axum" }
|
||||
axum-extra = { path = "../../axum-extra", features = ["typed-header"] }
|
||||
serde = { version = "1.0", features = ["derive"] }
|
||||
tokio = { version = "1.0", features = ["full"] }
|
||||
tracing = "0.1"
|
||||
tracing-subscriber = { version = "0.3", features = ["env-filter"] }
|
||||
uuid = { version = "1.0", features = ["v4", "serde"] }
|
||||
|
|
@ -1,157 +0,0 @@
|
|||
//! Run with
|
||||
//!
|
||||
//! ```not_rust
|
||||
//! cargo run -p example-sessions
|
||||
//! ```
|
||||
|
||||
use async_session::{MemoryStore, Session, SessionStore as _};
|
||||
use axum::{
|
||||
async_trait,
|
||||
extract::{FromRef, FromRequestParts},
|
||||
http::{
|
||||
self,
|
||||
header::{HeaderMap, HeaderValue},
|
||||
request::Parts,
|
||||
StatusCode,
|
||||
},
|
||||
response::IntoResponse,
|
||||
routing::get,
|
||||
RequestPartsExt, Router,
|
||||
};
|
||||
use axum_extra::{headers::Cookie, TypedHeader};
|
||||
use serde::{Deserialize, Serialize};
|
||||
use std::fmt::Debug;
|
||||
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};
|
||||
use uuid::Uuid;
|
||||
|
||||
const AXUM_SESSION_COOKIE_NAME: &str = "axum_session";
|
||||
|
||||
#[tokio::main]
|
||||
async fn main() {
|
||||
tracing_subscriber::registry()
|
||||
.with(
|
||||
tracing_subscriber::EnvFilter::try_from_default_env()
|
||||
.unwrap_or_else(|_| "example_sessions=debug".into()),
|
||||
)
|
||||
.with(tracing_subscriber::fmt::layer())
|
||||
.init();
|
||||
|
||||
// `MemoryStore` just used as an example. Don't use this in production.
|
||||
let store = MemoryStore::new();
|
||||
|
||||
let app = Router::new().route("/", get(handler)).with_state(store);
|
||||
|
||||
let listener = tokio::net::TcpListener::bind("127.0.0.1:3000")
|
||||
.await
|
||||
.unwrap();
|
||||
tracing::debug!("listening on {}", listener.local_addr().unwrap());
|
||||
axum::serve(listener, app).await.unwrap();
|
||||
}
|
||||
|
||||
async fn handler(user_id: UserIdFromSession) -> impl IntoResponse {
|
||||
let (headers, user_id, create_cookie) = match user_id {
|
||||
UserIdFromSession::FoundUserId(user_id) => (HeaderMap::new(), user_id, false),
|
||||
UserIdFromSession::CreatedFreshUserId(new_user) => {
|
||||
let mut headers = HeaderMap::new();
|
||||
headers.insert(http::header::SET_COOKIE, new_user.cookie);
|
||||
(headers, new_user.user_id, true)
|
||||
}
|
||||
};
|
||||
|
||||
tracing::debug!("handler: user_id={:?} send_headers={:?}", user_id, headers);
|
||||
|
||||
(
|
||||
headers,
|
||||
format!(
|
||||
"user_id={:?} session_cookie_name={} create_new_session_cookie={}",
|
||||
user_id, AXUM_SESSION_COOKIE_NAME, create_cookie
|
||||
),
|
||||
)
|
||||
}
|
||||
|
||||
struct FreshUserId {
|
||||
pub user_id: UserId,
|
||||
pub cookie: HeaderValue,
|
||||
}
|
||||
|
||||
enum UserIdFromSession {
|
||||
FoundUserId(UserId),
|
||||
CreatedFreshUserId(FreshUserId),
|
||||
}
|
||||
|
||||
#[async_trait]
|
||||
impl<S> FromRequestParts<S> for UserIdFromSession
|
||||
where
|
||||
MemoryStore: FromRef<S>,
|
||||
S: Send + Sync,
|
||||
{
|
||||
type Rejection = (StatusCode, &'static str);
|
||||
|
||||
async fn from_request_parts(parts: &mut Parts, state: &S) -> Result<Self, Self::Rejection> {
|
||||
let store = MemoryStore::from_ref(state);
|
||||
|
||||
let cookie: Option<TypedHeader<Cookie>> = parts.extract().await.unwrap();
|
||||
|
||||
let session_cookie = cookie
|
||||
.as_ref()
|
||||
.and_then(|cookie| cookie.get(AXUM_SESSION_COOKIE_NAME));
|
||||
|
||||
// return the new created session cookie for client
|
||||
if session_cookie.is_none() {
|
||||
let user_id = UserId::new();
|
||||
let mut session = Session::new();
|
||||
session.insert("user_id", user_id).unwrap();
|
||||
let cookie = store.store_session(session).await.unwrap().unwrap();
|
||||
return Ok(Self::CreatedFreshUserId(FreshUserId {
|
||||
user_id,
|
||||
cookie: HeaderValue::from_str(
|
||||
format!("{}={}", AXUM_SESSION_COOKIE_NAME, cookie).as_str(),
|
||||
)
|
||||
.unwrap(),
|
||||
}));
|
||||
}
|
||||
|
||||
tracing::debug!(
|
||||
"UserIdFromSession: got session cookie from user agent, {}={}",
|
||||
AXUM_SESSION_COOKIE_NAME,
|
||||
session_cookie.unwrap()
|
||||
);
|
||||
// continue to decode the session cookie
|
||||
let user_id = if let Some(session) = store
|
||||
.load_session(session_cookie.unwrap().to_owned())
|
||||
.await
|
||||
.unwrap()
|
||||
{
|
||||
if let Some(user_id) = session.get::<UserId>("user_id") {
|
||||
tracing::debug!(
|
||||
"UserIdFromSession: session decoded success, user_id={:?}",
|
||||
user_id
|
||||
);
|
||||
user_id
|
||||
} else {
|
||||
return Err((
|
||||
StatusCode::INTERNAL_SERVER_ERROR,
|
||||
"No `user_id` found in session",
|
||||
));
|
||||
}
|
||||
} else {
|
||||
tracing::debug!(
|
||||
"UserIdFromSession: err session not exists in store, {}={}",
|
||||
AXUM_SESSION_COOKIE_NAME,
|
||||
session_cookie.unwrap()
|
||||
);
|
||||
return Err((StatusCode::BAD_REQUEST, "No session found for cookie"));
|
||||
};
|
||||
|
||||
Ok(Self::FoundUserId(user_id))
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Serialize, Deserialize, Debug, Clone, Copy)]
|
||||
struct UserId(Uuid);
|
||||
|
||||
impl UserId {
|
||||
fn new() -> Self {
|
||||
Self(Uuid::new_v4())
|
||||
}
|
||||
}
|
||||
Loading…
Reference in a new issue