forked from TrueCloudLab/frostfs-api-go
22265a9f03
This commit: * changes the set and signatures of TokenStore interface methods; * adds unit test for map storage.
53 lines
1 KiB
Go
53 lines
1 KiB
Go
package session
|
|
|
|
import (
|
|
"crypto/elliptic"
|
|
"sync"
|
|
)
|
|
|
|
type mapTokenStore struct {
|
|
*sync.RWMutex
|
|
|
|
tokens map[TokenID]PrivateToken
|
|
}
|
|
|
|
// TODO get curve from neofs-crypto
|
|
func defaultCurve() elliptic.Curve {
|
|
return elliptic.P256()
|
|
}
|
|
|
|
// NewMapTokenStore creates new PrivateTokenStore instance.
|
|
//
|
|
// The elements of the instance are stored in the map.
|
|
func NewMapTokenStore() PrivateTokenStore {
|
|
return &mapTokenStore{
|
|
RWMutex: new(sync.RWMutex),
|
|
tokens: make(map[TokenID]PrivateToken),
|
|
}
|
|
}
|
|
|
|
// Store adds passed token to the map.
|
|
//
|
|
// Resulting error is always nil.
|
|
func (s *mapTokenStore) Store(id TokenID, token PrivateToken) error {
|
|
s.Lock()
|
|
s.tokens[id] = token
|
|
s.Unlock()
|
|
|
|
return nil
|
|
}
|
|
|
|
// Fetch returns the map element corresponding to the given key.
|
|
//
|
|
// Returns ErrPrivateTokenNotFound is there is no element in map.
|
|
func (s *mapTokenStore) Fetch(id TokenID) (PrivateToken, error) {
|
|
s.RLock()
|
|
defer s.RUnlock()
|
|
|
|
t, ok := s.tokens[id]
|
|
if !ok {
|
|
return nil, ErrPrivateTokenNotFound
|
|
}
|
|
|
|
return t, nil
|
|
}
|