show folks we can use const to "strongly type" keys

maxcountryman · Sep 22, 2023 · 73ad3da · 73ad3da
1 parent 6c830f6
commit 73ad3da
Show file tree

Hide file tree

Showing 5 changed files with 20 additions and 10 deletions.
diff --git a/examples/counter-extractor.rs b/examples/counter-extractor.rs
@@ -10,6 +10,8 @@ use serde::{Deserialize, Serialize};
 use tower::ServiceBuilder;
 use tower_sessions::{time::Duration, MemoryStore, Session, SessionManagerLayer};
 
+const COUNTER_KEY: &str = "counter";
+
 #[derive(Default, Deserialize, Serialize)]
 struct Counter(usize);
 
@@ -24,12 +26,12 @@ where
         let session = Session::from_request_parts(req, state).await?;
 
         let counter: Counter = session
-            .get("counter")
+            .get(COUNTER_KEY)
             .expect("Could not deserialize.")
             .unwrap_or_default();
 
         session
-            .insert("counter", counter.0 + 1)
+            .insert(COUNTER_KEY, counter.0 + 1)
             .expect("Could not serialize.");
 
         Ok(counter)

diff --git a/examples/counter.rs b/examples/counter.rs
@@ -8,6 +8,8 @@ use serde::{Deserialize, Serialize};
 use tower::ServiceBuilder;
 use tower_sessions::{time::Duration, MemoryStore, Session, SessionManagerLayer};
 
+const COUNTER_KEY: &str = "counter";
+
 #[derive(Default, Deserialize, Serialize)]
 struct Counter(usize);
 
@@ -37,12 +39,12 @@ async fn main() {
 
 async fn handler(session: Session) -> impl IntoResponse {
     let counter: Counter = session
-        .get("counter")
+        .get(COUNTER_KEY)
         .expect("Could not deserialize.")
         .unwrap_or_default();
 
     session
-        .insert("counter", counter.0 + 1)
+        .insert(COUNTER_KEY, counter.0 + 1)
         .expect("Could not serialize.");
 
     format!("Current count: {}", counter.0)

diff --git a/examples/postgres-store.rs b/examples/postgres-store.rs
@@ -8,6 +8,8 @@ use serde::{Deserialize, Serialize};
 use tower::ServiceBuilder;
 use tower_sessions::{sqlx::PgPool, time::Duration, PostgresStore, Session, SessionManagerLayer};
 
+const COUNTER_KEY: &str = "counter";
+
 #[derive(Serialize, Deserialize, Default)]
 struct Counter(usize);
 
@@ -50,12 +52,12 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
 
 async fn handler(session: Session) -> impl IntoResponse {
     let counter: Counter = session
-        .get("counter")
+        .get(COUNTER_KEY)
         .expect("Could not deserialize.")
         .unwrap_or_default();
 
     session
-        .insert("counter", counter.0 + 1)
+        .insert(COUNTER_KEY, counter.0 + 1)
         .expect("Could not serialize.");
 
     format!("Current count: {}", counter.0)

diff --git a/examples/redis-store.rs b/examples/redis-store.rs
@@ -8,6 +8,8 @@ use serde::{Deserialize, Serialize};
 use tower::ServiceBuilder;
 use tower_sessions::{fred::prelude::*, time::Duration, RedisStore, Session, SessionManagerLayer};
 
+const COUNTER_KEY: &str = "counter";
+
 #[derive(Serialize, Deserialize, Default)]
 struct Counter(usize);
 
@@ -43,12 +45,12 @@ async fn main() {
 
 async fn handler(session: Session) -> impl IntoResponse {
     let counter: Counter = session
-        .get("counter")
+        .get(COUNTER_KEY)
         .expect("Could not deserialize.")
         .unwrap_or_default();
 
     session
-        .insert("counter", counter.0 + 1)
+        .insert(COUNTER_KEY, counter.0 + 1)
         .expect("Could not serialize.");
 
     format!("Current count: {}", counter.0)

diff --git a/examples/sqlite-store.rs b/examples/sqlite-store.rs
@@ -8,6 +8,8 @@ use serde::{Deserialize, Serialize};
 use tower::ServiceBuilder;
 use tower_sessions::{sqlx::SqlitePool, time::Duration, Session, SessionManagerLayer, SqliteStore};
 
+const COUNTER_KEY: &str = "counter";
+
 #[derive(Serialize, Deserialize, Default)]
 struct Counter(usize);
 
@@ -49,12 +51,12 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
 
 async fn handler(session: Session) -> impl IntoResponse {
     let counter: Counter = session
-        .get("counter")
+        .get(COUNTER_KEY)
         .expect("Could not deserialize.")
         .unwrap_or_default();
 
     session
-        .insert("counter", counter.0 + 1)
+        .insert(COUNTER_KEY, counter.0 + 1)
         .expect("Could not serialize.");
 
     format!("Current count: {}", counter.0)