TrueCloudLab/neoneo-go
7
0
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity

storage: add locking into (*MemoryStore).Seek

Browse source 
It was missing there.
This commit is contained in:
Roman Khimov 2020-02-24 17:51:50 +03:00
parent cad1f074d4
commit ed9e9e8590
2 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
pkg/core/storage/memcached_store.go
View file

@ -75,7 +75,7 @@ func (s *MemCachedStore) GetBatch() *MemBatch {
func (s *MemCachedStore) Seek(key []byte, f func(k, v []byte)) { func (s *MemCachedStore) Seek(key []byte, f func(k, v []byte)) {
s.mut.RLock() s.mut.RLock()
defer s.mut.RUnlock() defer s.mut.RUnlock()
s.MemoryStore.Seek(key, f) s.MemoryStore.seek(key, f)
s.ps.Seek(key, func(k, v []byte) { s.ps.Seek(key, func(k, v []byte) {
elem := string(k) elem := string(k)
// If it's in mem, we already called f() for it in MemoryStore.Seek(). // If it's in mem, we already called f() for it in MemoryStore.Seek().

7
pkg/core/storage/memory_store.go
View file

@ -97,6 +97,13 @@ func (s *MemoryStore) PutBatch(batch Batch) error {
// Seek implements the Store interface. // Seek implements the Store interface.
func (s *MemoryStore) Seek(key []byte, f func(k, v []byte)) { func (s *MemoryStore) Seek(key []byte, f func(k, v []byte)) {
s.mut.RLock()
s.seek(key, f)
s.mut.RUnlock()
}
// seek is an internal unlocked implementation of Seek.
func (s *MemoryStore) seek(key []byte, f func(k, v []byte)) {
for k, v := range s.mem { for k, v := range s.mem {
if strings.HasPrefix(k, string(key)) { if strings.HasPrefix(k, string(key)) {
f([]byte(k), v) f([]byte(k), v)