github
/
alist
mirror of https://github.com/Xhofe/alist
1
0
Fork 0
Code Issues Releases Wiki Activity

chore: keep guest in memory

refactor/fs
Noah Hsu 2022-06-27 14:29:36 +08:00
parent 490df4f5fe
commit e71aff9d94
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
internal/db/user.go
View File

@ -11,6 +11,7 @@ import (
var userCache = cache.NewMemCache(cache.WithShards[*model.User](2)) var userCache = cache.NewMemCache(cache.WithShards[*model.User](2))
var userG singleflight.Group[*model.User] var userG singleflight.Group[*model.User]
var guest *model.User
func GetAdmin() (*model.User, error) { func GetAdmin() (*model.User, error) {
user := model.User{Role: model.ADMIN} user := model.User{Role: model.ADMIN}
@ -21,10 +22,14 @@ func GetAdmin() (*model.User, error) {
} }
func GetGuest() (*model.User, error) { func GetGuest() (*model.User, error) {
if guest != nil {
return guest, nil
}
user := model.User{Role: model.GUEST} user := model.User{Role: model.GUEST}
if err := db.Where(user).Take(&user).Error; err != nil { if err := db.Where(user).Take(&user).Error; err != nil {
return nil, err return nil, err
} }
guest = &user
return &user, nil return &user, nil
} }
@ -65,6 +70,9 @@ func UpdateUser(u *model.User) error {
return err return err
} }
userCache.Del(old.Username) userCache.Del(old.Username)
if u.IsGuest() {
guest = nil
}
return errors.WithStack(db.Save(u).Error) return errors.WithStack(db.Save(u).Error)
} }