xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-09-26 20:06:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update cache.go

Browse Source
This commit is contained in:
Mary Smiley 2024-03-16 09:36:49 +08:00 committed by GitHub
parent 0ebdd03dd7
commit b8ef5cd34c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 29 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
model/cache.go
View File

@ -18,6 +18,8 @@ var (
UserId2GroupCacheSeconds = common.SyncFrequency UserId2GroupCacheSeconds = common.SyncFrequency
UserId2QuotaCacheSeconds = common.SyncFrequency UserId2QuotaCacheSeconds = common.SyncFrequency
UserId2StatusCacheSeconds = common.SyncFrequency UserId2StatusCacheSeconds = common.SyncFrequency
UserId2timeCacheSeconds = common.SyncFrequency
) )
// 仅用于定时同步缓存 // 仅用于定时同步缓存
@ -160,6 +162,26 @@ func CacheGetUserQuota(id int) (quota int, err error) {
return quota, err return quota, err
} }
func CacheGetUserRefreshTimeStamp(id int) (RefreshTimeStamp int64, err error) {
if !common.RedisEnabled {
return GetUserRefreshTimeStamp(id)
}
stampString, err := common.RedisGet(fmt.Sprintf("user_stamp:%d", id))
if err != nil {
timestamp, err := GetUserRefreshTimeStamp(id)
if err != nil {
return 0, err
}
err = common.RedisSet(fmt.Sprintf("user_stamp:%d", id), fmt.Sprintf("%d", timestamp), time.Duration(UserId2timeCacheSeconds)*time.Second)
if err != nil {
common.SysError("Redis set user timestamp error: " + err.Error())
}
return timestamp, err
}
timestamp, err := strconv.ParseInt(stampString, 10, 64)
return timestamp, err
}
func CacheUpdateUserQuota(id int) error { func CacheUpdateUserQuota(id int) error {
if !common.RedisEnabled { if !common.RedisEnabled {
return nil return nil
@ -171,6 +193,13 @@ func CacheUpdateUserQuota(id int) error {
err = common.RedisSet(fmt.Sprintf("user_quota:%d", id), fmt.Sprintf("%d", quota), time.Duration(UserId2QuotaCacheSeconds)*time.Second) err = common.RedisSet(fmt.Sprintf("user_quota:%d", id), fmt.Sprintf("%d", quota), time.Duration(UserId2QuotaCacheSeconds)*time.Second)
return err return err
} }
func CacheUpdateUserRefreshTimeStamp(id int, timestamp int64) error {
if !common.RedisEnabled {
return nil
}
err := common.RedisSet(fmt.Sprintf("user_stamp:%d", id), fmt.Sprintf("%d", timestamp), time.Duration(UserId2QuotaCacheSeconds)*time.Second)
return err
}
func CacheDecreaseUserQuota(id int, quota int) error { func CacheDecreaseUserQuota(id int, quota int) error {
if !common.RedisEnabled { if !common.RedisEnabled {