oadmin/new-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix redis

Browse Source
This commit is contained in:
CalciumIon
2024-12-30 22:05:41 +08:00
parent 1eb706de7a
commit b4f17543cb
2 changed files with 21 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
model/token.go
View File

@@ -133,7 +133,7 @@ func GetTokenById(id int) (*Token, error) {
token := Token{Id: id} token := Token{Id: id}
var err error = nil var err error = nil
err = DB.First(&token, "id = ?", id).Error err = DB.First(&token, "id = ?", id).Error
if err != nil { if shouldUpdateRedis(true, err) {
gopool.Go(func() { gopool.Go(func() {
if err := cacheSetToken(token); err != nil { if err := cacheSetToken(token); err != nil {
common.SysError("failed to update user status cache: " + err.Error()) common.SysError("failed to update user status cache: " + err.Error())
@@ -176,7 +176,7 @@ func (token *Token) Insert() error {
// Update Make sure your token's fields is completed, because this will update non-zero values // Update Make sure your token's fields is completed, because this will update non-zero values
func (token *Token) Update() (err error) { func (token *Token) Update() (err error) {
defer func() { defer func() {
if common.RedisEnabled && err == nil { if shouldUpdateRedis(true, err) {
gopool.Go(func() { gopool.Go(func() {
err := cacheSetToken(*token) err := cacheSetToken(*token)
if err != nil { if err != nil {
@@ -192,7 +192,7 @@ func (token *Token) Update() (err error) {
func (token *Token) SelectUpdate() (err error) { func (token *Token) SelectUpdate() (err error) {
defer func() { defer func() {
if common.RedisEnabled && err == nil { if shouldUpdateRedis(true, err) {
gopool.Go(func() { gopool.Go(func() {
err := cacheSetToken(*token) err := cacheSetToken(*token)
if err != nil { if err != nil {
@@ -207,7 +207,7 @@ func (token *Token) SelectUpdate() (err error) {
func (token *Token) Delete() (err error) { func (token *Token) Delete() (err error) {
defer func() { defer func() {
if common.RedisEnabled && err == nil { if shouldUpdateRedis(true, err) {
gopool.Go(func() { gopool.Go(func() {
err := cacheDeleteToken(token.Key) err := cacheDeleteToken(token.Key)
if err != nil { if err != nil {
@@ -267,12 +267,14 @@ func IncreaseTokenQuota(id int, key string, quota int) (err error) {
if quota < 0 { if quota < 0 {
return errors.New("quota 不能为负数！") return errors.New("quota 不能为负数！")
} }
gopool.Go(func() { if common.RedisEnabled {
err := cacheIncrTokenQuota(key, int64(quota)) gopool.Go(func() {
if err != nil { err := cacheIncrTokenQuota(key, int64(quota))
common.SysError("failed to increase token quota: " + err.Error()) if err != nil {
} common.SysError("failed to increase token quota: " + err.Error())
}) }
})
}
if common.BatchUpdateEnabled { if common.BatchUpdateEnabled {
addNewRecord(BatchUpdateTypeTokenQuota, id, quota) addNewRecord(BatchUpdateTypeTokenQuota, id, quota)
return nil return nil
@@ -295,12 +297,14 @@ func DecreaseTokenQuota(id int, key string, quota int) (err error) {
if quota < 0 { if quota < 0 {
return errors.New("quota 不能为负数！") return errors.New("quota 不能为负数！")
} }
gopool.Go(func() { if common.RedisEnabled {
err := cacheDecrTokenQuota(key, int64(quota)) gopool.Go(func() {
if err != nil { err := cacheDecrTokenQuota(key, int64(quota))
common.SysError("failed to decrease token quota: " + err.Error()) if err != nil {
} common.SysError("failed to decrease token quota: " + err.Error())
}) }
})
}
if common.BatchUpdateEnabled { if common.BatchUpdateEnabled {
addNewRecord(BatchUpdateTypeTokenQuota, id, -quota) addNewRecord(BatchUpdateTypeTokenQuota, id, -quota)
return nil return nil

2
web/src/helpers/render.js
View File

@@ -64,7 +64,7 @@ export function renderRatio(ratio) {
} else if (ratio > 1) { } else if (ratio > 1) {
color = 'blue'; color = 'blue';
} }
return <Tag color={color}>{ratio} {i18next.t('倍率')}</Tag>; return <Tag color={color}>{ratio}x {i18next.t('倍率')}</Tag>;
} }
export const renderGroupOption = (item) => { export const renderGroupOption = (item) => {