frostfs-node/lib/placement/store.go
alexvanin dadfd90dcd Initial commit
Initial public review release v0.10.0
2020-07-10 17:45:00 +03:00

66 lines
1,007 B
Go

package placement
import (
"sync"
"github.com/nspcc-dev/neofs-node/lib/netmap"
)
type (
// NetMap is a type alias of
// NetMap from netmap package.
NetMap = netmap.NetMap
netMapStore struct {
*sync.RWMutex
items map[uint64]*NetMap
curEpoch uint64
}
)
func newNetMapStore() *netMapStore {
return &netMapStore{
RWMutex: new(sync.RWMutex),
items: make(map[uint64]*NetMap),
}
}
func (s *netMapStore) put(epoch uint64, nm *NetMap) {
s.Lock()
s.items[epoch] = nm
s.curEpoch = epoch
s.Unlock()
}
func (s *netMapStore) get(epoch uint64) *NetMap {
s.RLock()
nm := s.items[epoch]
s.RUnlock()
return nm
}
// trim cleans all network states elder than epoch.
func (s *netMapStore) trim(epoch uint64) {
s.Lock()
m := make(map[uint64]struct{}, len(s.items))
for e := range s.items {
if e < epoch {
m[e] = struct{}{}
}
}
for e := range m {
delete(s.items, e)
}
s.Unlock()
}
func (s *netMapStore) epoch() uint64 {
s.RLock()
defer s.RUnlock()
return s.curEpoch
}