mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-15 03:11:02 +01:00
25 lines
413 B
Go
25 lines
413 B
Go
package hmac
|
|
|
|
import (
|
|
"sync"
|
|
)
|
|
|
|
// TokenStore is a simple in-memory store for token
|
|
// With this can update the token in thread safe way
|
|
type TokenStore struct {
|
|
mu sync.Mutex
|
|
token Token
|
|
}
|
|
|
|
func (a *TokenStore) UpdateToken(token Token) {
|
|
a.mu.Lock()
|
|
defer a.mu.Unlock()
|
|
a.token = token
|
|
}
|
|
|
|
func (a *TokenStore) Token() ([]byte, error) {
|
|
a.mu.Lock()
|
|
defer a.mu.Unlock()
|
|
return marshalToken(a.token)
|
|
}
|