app/im/src/jwt.rs · zhenyi/gitdata-core

194c44c

ZhenYi refactor: replace Uuid with i64 for user identifiers across multiple files 194c44c 12 days ago

29 lines 918 B Rust

Blame Raw

use export::AppRedis;
pub async fn consume_jti(redis: &AppRedis, jti: &str, ttl_secs: i64) -> bool {
    use export::RedisConnection;
    let mut conn = redis.connection_clone();
    let result: Result<Option<String>, redis::RedisError> = match &mut conn {
        RedisConnection::Standalone(c) => {
            redis::cmd("SET")
                .arg(format!("chat:jwt_used:{jti}"))
                .arg("1")
                .arg("NX")
                .arg("EX")
                .arg(ttl_secs.max(1))
                .query_async(c)
                .await
        }
        RedisConnection::Cluster(c) => {
            redis::cmd("SET")
                .arg(format!("chat:jwt_used:{jti}"))
                .arg("1")
                .arg("NX")
                .arg("EX")
                .arg(ttl_secs.max(1))
                .query_async(c)
                .await
        }
    };
    matches!(result, Ok(Some(_)))
}