1
0
mirror of https://github.com/actix/examples synced 2024-11-23 22:41:07 +01:00

Update session/redis-session to v4 (#494)

Co-authored-by: Rob Ede <robjtede@icloud.com>
This commit is contained in:
Luca Palmieri 2022-02-02 15:24:23 +00:00 committed by GitHub
parent 6e5b04c570
commit 7ed15d05fe
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 22 additions and 25 deletions

View File

@ -4,16 +4,13 @@ version = "1.0.0"
edition = "2021" edition = "2021"
[dependencies] [dependencies]
actix-web = "3" actix-web = "4.0.0-beta.21"
actix-session = "0.4" actix-session = "0.5.0-beta.7"
actix-redis = "0.9" actix-redis = "0.10.0-beta.5"
env_logger = "0.8" env_logger = "0.9.0"
serde = { version = "1", features = ["derive"] } serde = { version = "1", features = ["derive"] }
serde_json = "1" serde_json = "1"
actix-service = "1"
actix-http = "2"
rand = "0.7.3"
time = "0.2"
[dev-dependencies] [dev-dependencies]
actix-rt = "1" actix-test = "0.1.0-beta.11"
time = "0.3.7"

View File

@ -11,7 +11,6 @@ use actix_web::{
web::{get, post, resource}, web::{get, post, resource},
App, HttpResponse, HttpServer, Result, App, HttpResponse, HttpServer, Result,
}; };
use rand::Rng;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
#[derive(Serialize, Deserialize, Debug, PartialEq)] #[derive(Serialize, Deserialize, Debug, PartialEq)]
@ -36,7 +35,7 @@ async fn do_something(session: Session) -> Result<HttpResponse> {
.get::<i32>("counter") .get::<i32>("counter")
.unwrap_or(Some(0)) .unwrap_or(Some(0))
.map_or(1, |inner| inner + 1); .map_or(1, |inner| inner + 1);
session.set("counter", counter)?; session.insert("counter", counter)?;
Ok(HttpResponse::Ok().json(IndexResponse { user_id, counter })) Ok(HttpResponse::Ok().json(IndexResponse { user_id, counter }))
} }
@ -48,7 +47,7 @@ struct Identity {
async fn login(user_id: web::Json<Identity>, session: Session) -> Result<HttpResponse> { async fn login(user_id: web::Json<Identity>, session: Session) -> Result<HttpResponse> {
let id = user_id.into_inner().user_id; let id = user_id.into_inner().user_id;
session.set("user_id", &id)?; session.insert("user_id", &id)?;
session.renew(); session.renew();
let counter: i32 = session let counter: i32 = session
@ -62,11 +61,11 @@ async fn login(user_id: web::Json<Identity>, session: Session) -> Result<HttpRes
})) }))
} }
async fn logout(session: Session) -> Result<HttpResponse> { async fn logout(session: Session) -> Result<String> {
let id: Option<String> = session.get("user_id")?; let id: Option<String> = session.get("user_id")?;
if let Some(x) = id { if let Some(x) = id {
session.purge(); session.purge();
Ok(format!("Logged out: {}", x).into()) Ok(format!("Logged out: {}", x))
} else { } else {
Ok("Could not log out anonymous user".into()) Ok("Could not log out anonymous user".into())
} }
@ -80,12 +79,12 @@ async fn main() -> std::io::Result<()> {
// Generate a random 32 byte key. Note that it is important to use a unique // Generate a random 32 byte key. Note that it is important to use a unique
// private key for every project. Anyone with access to the key can generate // private key for every project. Anyone with access to the key can generate
// authentication cookies for any user! // authentication cookies for any user!
let private_key = rand::thread_rng().gen::<[u8; 32]>(); let private_key = actix_web::cookie::Key::generate();
HttpServer::new(move || { HttpServer::new(move || {
App::new() App::new()
// redis session middleware // redis session middleware
.wrap(RedisSession::new("127.0.0.1:6379", &private_key)) .wrap(RedisSession::new("127.0.0.1:6379", &private_key.master()))
// enable logger - always register actix-web Logger middleware last // enable logger - always register actix-web Logger middleware last
.wrap(middleware::Logger::default()) .wrap(middleware::Logger::default())
.service(resource("/").route(get().to(index))) .service(resource("/").route(get().to(index)))
@ -101,15 +100,14 @@ async fn main() -> std::io::Result<()> {
#[cfg(test)] #[cfg(test)]
mod test { mod test {
use super::*; use super::*;
use actix_http::httpmessage::HttpMessage;
use actix_web::{ use actix_web::{
middleware, test, middleware,
web::{get, post, resource}, web::{get, post, resource},
App, App,
}; };
use serde_json::json; use serde_json::json;
#[actix_rt::test] #[actix_web::test]
async fn test_workflow() { async fn test_workflow() {
// Step 1: GET index // Step 1: GET index
// - set-cookie actix-session will be in response (session cookie #1) // - set-cookie actix-session will be in response (session cookie #1)
@ -141,11 +139,11 @@ mod test {
// - set-cookie actix-session will be in response (session cookie #3) // - set-cookie actix-session will be in response (session cookie #3)
// - response should be: {"counter": 0, "user_id": None} // - response should be: {"counter": 0, "user_id": None}
let private_key = rand::thread_rng().gen::<[u8; 32]>(); let private_key = actix_web::cookie::Key::generate();
let srv = test::start(move || { let srv = actix_test::start(move || {
App::new() App::new()
.wrap( .wrap(
RedisSession::new("127.0.0.1:6379", &private_key) RedisSession::new("127.0.0.1:6379", &private_key.master())
.cookie_name("test-session"), .cookie_name("test-session"),
) )
.wrap(middleware::Logger::default()) .wrap(middleware::Logger::default())
@ -158,8 +156,8 @@ mod test {
// Step 1: GET index // Step 1: GET index
// - set-cookie actix-session will be in response (session cookie #1) // - set-cookie actix-session will be in response (session cookie #1)
// - response should be: {"counter": 0, "user_id": None} // - response should be: {"counter": 0, "user_id": None}
let req_1a = srv.get("/").send(); let request = srv.get("/").send();
let mut resp_1 = req_1a.await.unwrap(); let mut resp_1 = request.await.unwrap();
let cookie_1 = resp_1 let cookie_1 = resp_1
.cookies() .cookies()
.unwrap() .unwrap()
@ -306,7 +304,9 @@ mod test {
.unwrap(); .unwrap();
let now = time::OffsetDateTime::now_utc(); let now = time::OffsetDateTime::now_utc();
assert!(now.year() != cookie_4.expires().map(|t| t.year()).unwrap()); assert_ne!(
now.year(), cookie_4.expires().unwrap().datetime().unwrap().year()
);
// Step 10: GET index, including session cookie #2 in request // Step 10: GET index, including session cookie #2 in request
// - set-cookie actix-session will be in response (session cookie #3) // - set-cookie actix-session will be in response (session cookie #3)