feat: init

2026-06-07 11:30:56 +08:00
commit 563381c1ca
361 changed files with 41327 additions and 0 deletions
@@ -0,0 +1,93 @@
+use crate::cache::redis::AppRedis;
+use crate::error::{AppError, AppResult};
+use crate::service::im::util::PRESENCE_PREFIX;
+use ::redis::Cmd;
+
+use super::redis_keys::*;
+use super::session::WsSession;
+
+pub fn register_redis_online(redis: &AppRedis, session: &WsSession) -> AppResult<()> {
+    let set_key = format!("{WS_ONLINE_PREFIX}{}", session.user_id);
+    let conn_id = session.connection_id.to_string();
+    let meta_key = format!("{WS_CONNS_PREFIX}{}", session.connection_id);
+    let mut conn = redis.get_connection()?;
+
+    Cmd::new()
+        .arg("SADD")
+        .arg(&set_key)
+        .arg(&conn_id)
+        .query::<i32>(&mut *conn.inner_mut())?;
+    Cmd::new()
+        .arg("EXPIRE")
+        .arg(&set_key)
+        .arg(WS_ONLINE_TTL_SECS)
+        .query::<()>(&mut *conn.inner_mut())?;
+    Cmd::new()
+        .arg("SETEX")
+        .arg(&meta_key)
+        .arg(WS_ONLINE_TTL_SECS)
+        .arg(session.workspace_name.as_str())
+        .query::<()>(&mut *conn.inner_mut())?;
+    Ok(())
+}
+
+pub fn unregister_redis_online(redis: &AppRedis, session: &WsSession) -> AppResult<()> {
+    let set_key = format!("{WS_ONLINE_PREFIX}{}", session.user_id);
+    let conn_id = session.connection_id.to_string();
+    let meta_key = format!("{WS_CONNS_PREFIX}{}", session.connection_id);
+    let mut conn = redis.get_connection()?;
+
+    Cmd::new()
+        .arg("SREM")
+        .arg(&set_key)
+        .arg(&conn_id)
+        .query::<i32>(&mut *conn.inner_mut())?;
+
+    let remaining: i32 = Cmd::new()
+        .arg("SCARD")
+        .arg(&set_key)
+        .query(&mut *conn.inner_mut())
+        .map_err(AppError::Redis)?;
+    if remaining == 0 {
+        Cmd::new()
+            .arg("DEL")
+            .arg(&set_key)
+            .query::<()>(&mut *conn.inner_mut())?;
+        let pk = format!("{PRESENCE_PREFIX}{}", session.user_id);
+        let _ = Cmd::new()
+            .arg("DEL")
+            .arg(&pk)
+            .query::<()>(&mut *conn.inner_mut());
+    }
+    let _ = Cmd::new()
+        .arg("DEL")
+        .arg(&meta_key)
+        .query::<()>(&mut *conn.inner_mut());
+    Ok(())
+}
+
+pub fn heartbeat_redis(redis: &AppRedis, session: &WsSession) -> AppResult<()> {
+    let set_key = format!("{WS_ONLINE_PREFIX}{}", session.user_id);
+    let meta_key = format!("{WS_CONNS_PREFIX}{}", session.connection_id);
+    let pk = format!("{PRESENCE_PREFIX}{}", session.user_id);
+    let mut conn = redis.get_connection()?;
+
+    let _ = Cmd::new()
+        .arg("EXPIRE")
+        .arg(&set_key)
+        .arg(WS_ONLINE_TTL_SECS)
+        .query::<()>(&mut *conn.inner_mut());
+    let _ = Cmd::new()
+        .arg("SETEX")
+        .arg(&meta_key)
+        .arg(WS_ONLINE_TTL_SECS)
+        .arg(session.workspace_name.as_str())
+        .query::<()>(&mut *conn.inner_mut());
+    let _ = Cmd::new()
+        .arg("SETEX")
+        .arg(&pk)
+        .arg(WS_ONLINE_TTL_SECS)
+        .arg("online")
+        .query::<()>(&mut *conn.inner_mut());
+    Ok(())
+}