Add Id to tokens.Token struct and func GetID to tokens package

This commit is contained in:
qowevisa 2024-08-03 08:48:33 +03:00
parent 39cecc80d1
commit 2d61d59179

View File

@ -11,6 +11,7 @@ import (
) )
type Token struct { type Token struct {
Id uint
Val string Val string
LastActive time.Time LastActive time.Time
} }
@ -77,6 +78,16 @@ func GetToken(id uint) (*Token, error) {
return val, nil return val, nil
} }
func GetID(token string) (uint, error) {
toks.Mu.RLock()
val, exists := toks.TokmapRev[token]
toks.Mu.RUnlock()
if !exists {
return 0, ERROR_DONT_HAVE_TOKEN
}
return val.Id, nil
}
func haveToken(id uint) bool { func haveToken(id uint) bool {
toks.Mu.RLock() toks.Mu.RLock()
_, exists := toks.Tokmap[id] _, exists := toks.Tokmap[id]
@ -130,6 +141,7 @@ func AddToken(id uint) (*Token, error) {
} }
val := generateTokenVal() val := generateTokenVal()
token := &Token{ token := &Token{
Id: id,
Val: val, Val: val,
LastActive: time.Now(), LastActive: time.Now(),
} }