forked from TrueCloudLab/frostfs-sdk-go
48 lines
799 B
Go
48 lines
799 B
Go
|
package pool
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/bluele/gcache"
|
||
|
|
"github.com/nspcc-dev/neofs-api-go/pkg/session"
|
||
|
|
)
|
||
|
|
|
||
|
|
type SessionCache struct {
|
||
|
|
cache gcache.Cache
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewCache() *SessionCache {
|
||
|
|
return &SessionCache{
|
||
|
|
cache: gcache.New(100).Build(),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *SessionCache) Get(key string) *session.Token {
|
||
|
|
tokenRaw, err := c.cache.Get(key)
|
||
|
|
if err != nil {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
token, ok := tokenRaw.(*session.Token)
|
||
|
|
if !ok {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
return token
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *SessionCache) Put(key string, token *session.Token) error {
|
||
|
|
return c.cache.Set(key, token)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *SessionCache) DeleteByPrefix(prefix string) {
|
||
|
|
for _, key := range c.cache.Keys(false) {
|
||
|
|
keyStr, ok := key.(string)
|
||
|
|
if !ok {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
if strings.HasPrefix(keyStr, prefix) {
|
||
|
|
c.cache.Remove(key)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|