Shield Tower: Rename to session storage

RustForWeb · Dec 25, 2024 · 724538b · 724538b
1 parent fd573b9
commit 724538b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 8 deletions.
diff --git a/packages/integrations/shield-tower/src/service.rs b/packages/integrations/shield-tower/src/service.rs
@@ -8,7 +8,7 @@ use http::{Request, Response};
 use shield::{Session, Shield};
 use tower_service::Service;
 
-use crate::session::TowerSession;
+use crate::session::TowerSessionStorage;
 
 #[derive(Clone)]
 pub struct ShieldService<S> {
@@ -66,10 +66,11 @@ where
                 }
             };
 
-            let session_storage = match TowerSession::load(session.clone(), session_key).await {
-                Ok(session_storage) => session_storage,
-                Err(_err) => return Ok(Self::internal_server_error()),
-            };
+            let session_storage =
+                match TowerSessionStorage::load(session.clone(), session_key).await {
+                    Ok(session_storage) => session_storage,
+                    Err(_err) => return Ok(Self::internal_server_error()),
+                };
             let shield_session = Session::new(session_storage);
 
             req.extensions_mut().insert(shield);

diff --git a/packages/integrations/shield-tower/src/session.rs b/packages/integrations/shield-tower/src/session.rs
@@ -5,13 +5,13 @@ use shield::{SessionData, SessionError, SessionStorage};
 use tower_sessions::Session;
 
 #[derive(Clone, Debug)]
-pub struct TowerSession {
+pub struct TowerSessionStorage {
     session: Session,
     session_key: &'static str,
     session_data: Arc<Mutex<SessionData>>,
 }
 
-impl TowerSession {
+impl TowerSessionStorage {
     pub async fn load(session: Session, session_key: &'static str) -> Result<Self, SessionError> {
         let data = Self::load_data(&session, session_key).await?;
 
@@ -35,7 +35,7 @@ impl TowerSession {
 }
 
 #[async_trait]
-impl SessionStorage for TowerSession {
+impl SessionStorage for TowerSessionStorage {
     fn data(&self) -> Arc<Mutex<SessionData>> {
         self.session_data.clone()
     }