diff --git a/pkg/modules/keyvalue/memory/memory.go b/pkg/modules/keyvalue/memory/memory.go index 7a2e83b9e..08edc9886 100644 --- a/pkg/modules/keyvalue/memory/memory.go +++ b/pkg/modules/keyvalue/memory/memory.go @@ -72,11 +72,7 @@ func (s *Storage) IncrBy(key string, update int64) (err error) { s.mutex.Lock() defer s.mutex.Unlock() - v, err := s.Get(key) - if err != nil && !e.IsErrValueNotFoundForKey(err) { - return err - } - val, is := v.(int64) + val, is := s.store[key].(int64) if !is { return &e.ErrValueHasWrongType{Key: key, ExpectedValue: "int64"} } @@ -90,11 +86,7 @@ func (s *Storage) DecrBy(key string, update int64) (err error) { s.mutex.Lock() defer s.mutex.Unlock() - v, err := s.Get(key) - if err != nil && !e.IsErrValueNotFoundForKey(err) { - return err - } - val, is := v.(int64) + val, is := s.store[key].(int64) if !is { return &e.ErrValueHasWrongType{Key: key, ExpectedValue: "int64"} }