Add Id to tokens.Token struct and func GetID to tokens package
This commit is contained in:
parent
39cecc80d1
commit
2d61d59179
|
@ -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(),
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue
Block a user