🐛 fix: playground gets token without verifying user's identity

This commit is contained in:
Martial BE
2024-04-28 10:25:33 +08:00
parent 490fb4c749
commit 317e0a6370
2 changed files with 5 additions and 4 deletions

View File

@@ -115,13 +115,13 @@ func GetTokenById(id int) (*Token, error) {
return &token, err
}
func GetTokenByName(name string) (*Token, error) {
func GetTokenByName(name string, user_id int) (*Token, error) {
if name == "" {
return nil, errors.New("name 为空!")
}
token := Token{Name: name}
var err error = nil
err = DB.First(&token, "name = ?", name).Error
err = DB.First(&token, "user_id = ? and name = ?", user_id, name).Error
return &token, err
}