diff --git a/actix-session/src/lib.rs b/actix-session/src/lib.rs index def35a1e9..8fd6bd0e0 100644 --- a/actix-session/src/lib.rs +++ b/actix-session/src/lib.rs @@ -45,7 +45,9 @@ use std::cell::RefCell; use std::rc::Rc; -use actix_web::dev::{Extensions, Payload, ServiceRequest, ServiceResponse}; +use actix_web::dev::{ + Extensions, Payload, RequestHead, ServiceRequest, ServiceResponse, +}; use actix_web::{Error, FromRequest, HttpMessage, HttpRequest}; use futures::future::{ok, Ready}; use hashbrown::HashMap; @@ -99,6 +101,12 @@ impl UserSession for ServiceRequest { } } +impl UserSession for RequestHead { + fn get_session(&mut self) -> Session { + Session::get_session(&mut *self.extensions_mut()) + } +} + #[derive(PartialEq, Clone, Debug)] pub enum SessionStatus { Changed,