rocket-sessions 0.0.1

trait for managing sessions on a rocket application.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76

#![warn(clippy::all)]

use std::time::Duration;

use redis::{AsyncCommands, Client, RedisError as Error};

use crate::SessionManager;

#[async_trait]
impl SessionManager for Client {
    type Error = Error;
    async fn set(&self, id: &str, key: &str, value: &str, time: Duration) -> Result<(), Error> {
        let key = format!("{key}:{id}");
        let secs = time
            .as_secs()
            .try_into()
            .unwrap_or(usize::MAX);
        #[cfg(feature = "log")]
        log::debug!("set \"{key}\" to \"{value}\" expiring in {secs} seconds.");
        self.get_async_connection()
            .await?
            .set_ex(key, &*value, secs)
            .await?;
        Ok(())
    }

    async fn delete(&self, id: &str, key: &str) -> Result<(), Error> {
        #[cfg(feature = "log")]
        log::debug!("delete \"{key}:{id}\"");
        self.get_async_connection()
            .await?
            .del(format!("{key}:{id}"))
            .await?;

        Ok(())
    }

    async fn get(&self, id: &str, key: &str) -> Result<Option<String>, Error> {
        #[cfg(feature = "log")]
        log::debug!("get {key}");
        let option: Option<String> = self
            .get_async_connection()
            .await?
            .get(format!("{key}:{id}"))
            .await?;
        Ok(option)
    }

    async fn expire_in(&self, id: &str, key: &str, time: Duration) -> Result<(), Error> {
        let key = format!("{key}:{id}");
        let secs = time
            .as_secs()
            .try_into()
            .unwrap_or(usize::MAX);
        #[cfg(feature = "log")]
        log::debug!("set expiration for \"{key}\" in {secs} seconds.");
        self.get_async_connection()
            .await?
            .expire(key, secs)
            .await?;
        Ok(())
    }

    async fn clear_all(&self) -> Result<(), Error> {
        #[cfg(feature = "log")]
        log::warn!("flushing all redis keys.");
        let mut cnn = self
            .get_async_connection()
            .await?;
        redis::Cmd::new()
            .arg("FLUSHDB")
            .query_async::<_, ()>(&mut cnn)
            .await?;
        Ok(())
    }
}