From d18989582d7d05ff796305893daccbc952ac88cf Mon Sep 17 00:00:00 2001
From: Tglman <tglman@tglman.com>
Date: Mon, 18 Nov 2019 22:12:21 +0000
Subject: [PATCH] feat: add access to UserSession from RequestHead

---
 actix-session/src/lib.rs | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

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,