diff options
author | ethersphere <thesw@rm.eth> | 2018-06-20 20:06:27 +0800 |
---|---|---|
committer | ethersphere <thesw@rm.eth> | 2018-06-22 03:10:31 +0800 |
commit | e187711c6545487d4cac3701f0f506bb536234e2 (patch) | |
tree | d2f6150f70b84b36e49a449082aeda267b4b9046 /swarm/state/inmemorystore.go | |
parent | 574378edb50c907b532946a1d4654dbd6701b20a (diff) | |
download | go-tangerine-e187711c6545487d4cac3701f0f506bb536234e2.tar go-tangerine-e187711c6545487d4cac3701f0f506bb536234e2.tar.gz go-tangerine-e187711c6545487d4cac3701f0f506bb536234e2.tar.bz2 go-tangerine-e187711c6545487d4cac3701f0f506bb536234e2.tar.lz go-tangerine-e187711c6545487d4cac3701f0f506bb536234e2.tar.xz go-tangerine-e187711c6545487d4cac3701f0f506bb536234e2.tar.zst go-tangerine-e187711c6545487d4cac3701f0f506bb536234e2.zip |
swarm: network rewrite merge
Diffstat (limited to 'swarm/state/inmemorystore.go')
-rw-r--r-- | swarm/state/inmemorystore.go | 94 |
1 files changed, 94 insertions, 0 deletions
diff --git a/swarm/state/inmemorystore.go b/swarm/state/inmemorystore.go new file mode 100644 index 000000000..1ca25404a --- /dev/null +++ b/swarm/state/inmemorystore.go @@ -0,0 +1,94 @@ +// Copyright 2018 The go-ethereum Authors +// This file is part of the go-ethereum library. +// +// The go-ethereum library is free software: you can redistribute it and/or modify +// it under the terms of the GNU Lesser General Public License as published by +// the Free Software Foundation, either version 3 of the License, or +// (at your option) any later version. +// +// The go-ethereum library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +// GNU Lesser General Public License for more details. +// +// You should have received a copy of the GNU Lesser General Public License +// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>. + +package state + +import ( + "encoding" + "encoding/json" + "sync" +) + +// InmemoryStore is the reference implementation of Store interface that is supposed +// to be used in tests. +type InmemoryStore struct { + db map[string][]byte + mu sync.RWMutex +} + +// NewInmemoryStore returns a new instance of InmemoryStore. +func NewInmemoryStore() *InmemoryStore { + return &InmemoryStore{ + db: make(map[string][]byte), + } +} + +// Get retrieves a value stored for a specific key. If there is no value found, +// ErrNotFound is returned. +func (s *InmemoryStore) Get(key string, i interface{}) (err error) { + s.mu.RLock() + defer s.mu.RUnlock() + + bytes, ok := s.db[key] + if !ok { + return ErrNotFound + } + + unmarshaler, ok := i.(encoding.BinaryUnmarshaler) + if !ok { + return json.Unmarshal(bytes, i) + } + + return unmarshaler.UnmarshalBinary(bytes) +} + +// Put stores a value for a specific key. +func (s *InmemoryStore) Put(key string, i interface{}) (err error) { + s.mu.Lock() + defer s.mu.Unlock() + bytes := []byte{} + + marshaler, ok := i.(encoding.BinaryMarshaler) + if !ok { + if bytes, err = json.Marshal(i); err != nil { + return err + } + } else { + if bytes, err = marshaler.MarshalBinary(); err != nil { + return err + } + } + + s.db[key] = bytes + return nil +} + +// Delete removes value stored under a specific key. +func (s *InmemoryStore) Delete(key string) (err error) { + s.mu.Lock() + defer s.mu.Unlock() + + if _, ok := s.db[key]; !ok { + return ErrNotFound + } + delete(s.db, key) + return nil +} + +// Close does not do anything. +func (s *InmemoryStore) Close() error { + return nil +} |