rifts 0.3.10

Rift Realtime Protocol / 1.0 — server + client implementation
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

//! Redis-backed message deduplication using per-key TTL.
//! All methods are async.

use std::time::Duration;

use async_trait::async_trait;

use crate::redis::connection::RedisPool;
use crate::storage::DedupeStore;

#[derive(Clone)]
pub struct RedisDedupeStore {
    pool: RedisPool,
}

impl RedisDedupeStore {
    pub fn new(pool: RedisPool) -> Self {
        Self { pool }
    }

    fn member_key(&self, topic: &str, message_id: &str) -> String {
        format!("{}:dedupe:{topic}:{message_id}", self.pool.prefix())
    }
}

#[async_trait]
impl DedupeStore for RedisDedupeStore {
    async fn check_and_record(&self, topic: &str, key: &str, window: Duration) -> bool {
        let member_key = self.member_key(topic, key);
        let window_secs = window.as_secs().max(1) as usize;
        let mut conn = self.pool.conn().clone();
        let result: Option<String> = redis::cmd("SET")
            .arg(&member_key)
            .arg("1")
            .arg("NX")
            .arg("EX")
            .arg(window_secs)
            .query_async(&mut conn)
            .await
            .unwrap_or(None);
        result.is_some()
    }

    async fn sweep(&self) -> usize {
        0
    }
}