You've already forked AstralRinth
forked from didirus/AstralRinth
cf190d86d5
* Update Rust version * Update async-compression 0.4.25 -> 0.4.27 * Update async-tungstenite 0.29.1 -> 0.30.0 * Update bytemuck 1.23.0 -> 1.23.1 * Update clap 4.5.40 -> 4.5.43 * Update deadpool-redis 0.21.1 -> 0.22.0 and redis 0.31.0 -> 0.32.4 * Update enumset 1.1.6 -> 1.1.7 * Update hyper-util 0.1.14 -> 0.1.16 * Update indexmap 2.9.0 -> 2.10.0 * Update indicatif 0.17.11 -> 0.18.0 * Update jemalloc_pprof 0.7.0 -> 0.8.1 * Update lettre 0.11.17 -> 0.11.18 * Update meilisearch-sdk 0.28.0 -> 0.29.1 * Update notify 8.0.0 -> 8.2.0 and notify-debouncer-mini 0.6.0 -> 0.7.0 * Update quick-xml 0.37.5 -> 0.38.1 * Fix theseus lint * Update reqwest 0.12.20 -> 0.12.22 * Cargo fmt in theseus * Update rgb 0.8.50 -> 0.8.52 * Update sentry 0.41.0 -> 0.42.0 and sentry-actix 0.41.0 -> 0.42.0 * Update serde_json 1.0.140 -> 1.0.142 * Update serde_with 3.13.0 -> 3.14.0 * Update spdx 0.10.8 -> 0.10.9 * Update sysinfo 0.35.2 -> 0.36.1 * Update tauri suite * Fix build by updating mappings * Update tokio 1.45.1 -> 1.47.1 and tokio-util 0.7.15 -> 0.7.16 * Update tracing-actix-web 0.7.18 -> 0.7.19 * Update zip 4.2.0 -> 4.3.0 * Misc Cargo.lock updates * Update Dockerfiles
70 lines
1.9 KiB
Rust
70 lines
1.9 KiB
Rust
use crate::database::redis::RedisPool;
|
|
use crate::queue::socket::ActiveSockets;
|
|
use ariadne::ids::UserId;
|
|
use ariadne::users::UserStatus;
|
|
use redis::AsyncCommands;
|
|
|
|
const EXPIRY_TIME_SECONDS: i64 = 60;
|
|
|
|
pub async fn get_user_status(
|
|
user: UserId,
|
|
local_sockets: &ActiveSockets,
|
|
redis: &RedisPool,
|
|
) -> Option<UserStatus> {
|
|
if let Some(friend_status) = local_sockets.get_status(user) {
|
|
return Some(friend_status);
|
|
}
|
|
|
|
if let Ok(mut conn) = redis.pool.get().await
|
|
&& let Ok(mut statuses) =
|
|
conn.sscan::<_, String>(get_field_name(user)).await
|
|
&& let Some(status_json) = statuses.next_item().await
|
|
{
|
|
return serde_json::from_str::<UserStatus>(&status_json).ok();
|
|
}
|
|
|
|
None
|
|
}
|
|
|
|
pub async fn replace_user_status(
|
|
old_status: Option<&UserStatus>,
|
|
new_status: Option<&UserStatus>,
|
|
redis: &RedisPool,
|
|
) -> Result<(), redis::RedisError> {
|
|
let Some(user) = new_status.or(old_status).map(|x| x.user_id) else {
|
|
return Ok(());
|
|
};
|
|
|
|
if let Ok(mut conn) = redis.pool.get().await {
|
|
let field_name = get_field_name(user);
|
|
let mut pipe = redis::pipe();
|
|
pipe.atomic();
|
|
if let Some(status) = old_status {
|
|
pipe.srem(&field_name, serde_json::to_string(&status).unwrap())
|
|
.ignore();
|
|
}
|
|
if let Some(status) = new_status {
|
|
pipe.sadd(&field_name, serde_json::to_string(&status).unwrap())
|
|
.ignore();
|
|
pipe.expire(&field_name, EXPIRY_TIME_SECONDS).ignore();
|
|
}
|
|
return pipe.query_async(&mut conn).await;
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub async fn push_back_user_expiry(
|
|
user: UserId,
|
|
redis: &RedisPool,
|
|
) -> Result<(), redis::RedisError> {
|
|
if let Ok(mut conn) = redis.pool.get().await {
|
|
return conn.expire(get_field_name(user), EXPIRY_TIME_SECONDS).await;
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
fn get_field_name(user: UserId) -> String {
|
|
format!("user_status:{user}")
|
|
}
|