Initial commit

Initial public review release v0.10.0

lib/placement/store.go

@@ -0,0 +1,66 @@
+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
+}