forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-08-18 09:53:48 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

swarm/storage: fix setCapacity for 0

Browse source
This commit is contained in:
Anton Evangelatov 2018-04-03 14:52:58 +02:00
parent 9a40655aa6
commit a43c8a9c36
1 changed files with 20 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
swarm/storage/memstore_lrucache.go
View file

@ -31,10 +31,17 @@ const (
type MemStore struct { type MemStore struct {
//cache *lru.ARCCache //cache *lru.ARCCache
cache *lru.Cache cache *lru.Cache
disabled bool
} }
func NewMemStore(_ *LDBStore, capacity uint) (m *MemStore) { func NewMemStore(_ *LDBStore, capacity uint) (m *MemStore) {
if capacity == 0 {
return &MemStore{
disabled: true,
}
}
onEvicted := func(key interface{}, value interface{}) { onEvicted := func(key interface{}, value interface{}) {
v := value.(*Chunk) v := value.(*Chunk)
<-v.dbStoredC <-v.dbStoredC
@ -51,6 +58,10 @@ func NewMemStore(_ *LDBStore, capacity uint) (m *MemStore) {
} }
func (m *MemStore) Get(key Key) (*Chunk, error) { func (m *MemStore) Get(key Key) (*Chunk, error) {
if m.disabled {
return nil, ErrChunkNotFound
}
c, ok := m.cache.Get(string(key)) c, ok := m.cache.Get(string(key))
if !ok { if !ok {
return nil, ErrChunkNotFound return nil, ErrChunkNotFound
@ -63,11 +74,18 @@ func (m *MemStore) Get(key Key) (*Chunk, error) {
} }
func (m *MemStore) Put(c *Chunk) { func (m *MemStore) Put(c *Chunk) {
if m.disabled {
return
}
m.cache.Add(string(c.Key), c) m.cache.Add(string(c.Key), c)
} }
func (m *MemStore) setCapacity(n int) { func (m *MemStore) setCapacity(n int) {
//no-op if n <= 0 {
m.disabled = true
} else {
m = NewMemStore(nil, uint(n))
}
} }
// Close memstore // Close memstore