liyunjia/chuanyue-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'main' of http://47.95.198.7:30000/gitea/liyunjia/chuanyue-service

Browse Source
This commit is contained in:
李运家 2025-03-09 10:57:19 +08:00
commit d1b0bb70ac
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
library/src/cache/inner_cache.rs vendored
View File

@ -12,12 +12,17 @@ use serde::{de::DeserializeOwned, Deserialize, Serialize};
use crate::cache::redis_cache::REDIS_CACHE; use crate::cache::redis_cache::REDIS_CACHE;
/// 缓存账号信息
#[derive(Debug, Clone, Serialize, Deserialize)] #[derive(Debug, Clone, Serialize, Deserialize)]
pub struct CacheAccount { pub struct CacheAccount {
pub account: Arc<Account>, pub account: Arc<Account>,
pub token: Arc<String>, pub token: Arc<String>,
} }
/// 内存缓存数据结构，包含缓存类型和缓存数据
/// 缓存类型用于区分不同的缓存数据，缓存数据用于存储缓存数据
/// S: 键值类型
/// D: 缓存数据类型
pub struct ServerCache<S, D> { pub struct ServerCache<S, D> {
pub cache: Cache<S, D>, pub cache: Cache<S, D>,
pub cache_type: Arc<String>, pub cache_type: Arc<String>,
@ -30,7 +35,7 @@ impl<D> ServerCache<String, D>
where where
D: Debug + Send + Sync + Clone + Serialize + DeserializeOwned + 'static, D: Debug + Send + Sync + Clone + Serialize + DeserializeOwned + 'static,
{ {
fn default(cache_type: &str) -> Self { fn default(cache_type: &str, expire: i64) -> Self {
let cache_type = Arc::new(cache_type.to_string()); let cache_type = Arc::new(cache_type.to_string());
let cache_type_raw = cache_type.clone(); let cache_type_raw = cache_type.clone();
let cache_eviction_listener: CacheEvictionListener<D> = let cache_eviction_listener: CacheEvictionListener<D> =
@ -60,7 +65,7 @@ where
let cache = CacheBuilder::new(20480) let cache = CacheBuilder::new(20480)
.name("login_cache") .name("login_cache")
.eviction_policy(EvictionPolicy::lru()) .eviction_policy(EvictionPolicy::lru())
.time_to_live(Duration::from_secs(config!().jwt.expires as u64)) .time_to_live(Duration::from_secs(expire as u64))
.async_eviction_listener(cache_eviction_listener) .async_eviction_listener(cache_eviction_listener)
.build(); .build();
ServerCache { cache, cache_type } ServerCache { cache, cache_type }
@ -120,17 +125,20 @@ struct RedisCache {
pub cache_key: String, pub cache_key: String,
// 缓存数据 // 缓存数据
pub data: String, pub data: String,
// 缓存过期时间，单位秒
pub expires: Option<i64>,
} }
impl RedisCache { impl RedisCache {
/// 向redis缓存Hash中插入数据 /// 向redis缓存Hash中插入数据
pub async fn insert(&self) { pub async fn insert(&self) {
let expire = self.expires.unwrap_or(60 * 30);
match REDIS_CACHE match REDIS_CACHE
.hset_ex( .hset_ex(
&self.cache_type, &self.cache_type,
&self.cache_key, &self.cache_key,
&self.data, &self.data,
config!().jwt.expires, expire
) )
.await .await
{ {
@ -151,6 +159,7 @@ impl RedisCache {
cache_type: cache_type.clone(), cache_type: cache_type.clone(),
cache_key: key, cache_key: key,
data: value, data: value,
expires: None,
}; };
result.push(cache_data); result.push(cache_data);
} }
@ -180,6 +189,7 @@ pub async fn init_cache() {
} }
lazy_static! { lazy_static! {
/// 登录账号缓存
pub static ref LOGIN_ACCOUNT_CACHE: ServerCache<String, Arc<CacheAccount>> = pub static ref LOGIN_ACCOUNT_CACHE: ServerCache<String, Arc<CacheAccount>> =
ServerCache::default("CACHE_ACCOUNT"); ServerCache::default("CACHE_ACCOUNT", config!().jwt.expires);
} }