Use gob encoding for struct marshalling in redis cache
This commit is contained in:
parent
3366a71ac1
commit
00b041fbb0
2
cache/cache.go
vendored
2
cache/cache.go
vendored
@ -5,7 +5,7 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
var ErrNilValue = errors.New("value is empty")
|
var ErrNotFound = errors.New("key doesn't exist in cache")
|
||||||
|
|
||||||
type Cache interface {
|
type Cache interface {
|
||||||
Store(key string, value interface{}) error
|
Store(key string, value interface{}) error
|
||||||
|
2
cache/inmemory_cache.go
vendored
2
cache/inmemory_cache.go
vendored
@ -35,7 +35,7 @@ func (imc *InMemoryCache) StoreWithTTL(key string, value interface{}, ttl time.D
|
|||||||
func (imc *InMemoryCache) Get(key string, value interface{}) error {
|
func (imc *InMemoryCache) Get(key string, value interface{}) error {
|
||||||
v, exists := imc.cache.Get(key)
|
v, exists := imc.cache.Get(key)
|
||||||
if !exists {
|
if !exists {
|
||||||
return ErrNilValue
|
return ErrNotFound
|
||||||
}
|
}
|
||||||
value = v
|
value = v
|
||||||
|
|
||||||
|
39
cache/redis_cache.go
vendored
39
cache/redis_cache.go
vendored
@ -1,11 +1,13 @@
|
|||||||
package cache
|
package cache
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
"context"
|
"context"
|
||||||
|
"encoding/gob"
|
||||||
|
"errors"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/Secured-Finance/dione/config"
|
"github.com/Secured-Finance/dione/config"
|
||||||
"github.com/fxamacker/cbor/v2"
|
|
||||||
"github.com/go-redis/redis/v8"
|
"github.com/go-redis/redis/v8"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -28,41 +30,52 @@ func NewRedisCache(config *config.Config) *RedisCache {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (rc *RedisCache) Store(key string, value interface{}) error {
|
func (rc *RedisCache) Store(key string, value interface{}) error {
|
||||||
mRes, err := cbor.Marshal(value)
|
data, err := gobMarshal(value)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
rc.Client.Set(rc.ctx, key, mRes, 0)
|
rc.Client.Set(rc.ctx, key, data, 0)
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (rc *RedisCache) StoreWithTTL(key string, value interface{}, ttl time.Duration) error {
|
func (rc *RedisCache) StoreWithTTL(key string, value interface{}, ttl time.Duration) error {
|
||||||
mRes, err := cbor.Marshal(value)
|
data, err := gobMarshal(value)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
rc.Client.Set(rc.ctx, key, data, ttl)
|
||||||
rc.Client.Set(rc.ctx, key, mRes, ttl)
|
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func gobMarshal(val interface{}) ([]byte, error) {
|
||||||
|
buf := bytes.NewBuffer(nil)
|
||||||
|
enc := gob.NewEncoder(buf)
|
||||||
|
err := enc.Encode(val)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return buf.Bytes(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func gobUnmarshal(data []byte, val interface{}) error {
|
||||||
|
buf := bytes.NewBuffer(data)
|
||||||
|
dec := gob.NewDecoder(buf)
|
||||||
|
return dec.Decode(&val)
|
||||||
|
}
|
||||||
|
|
||||||
func (rc *RedisCache) Get(key string, value interface{}) error {
|
func (rc *RedisCache) Get(key string, value interface{}) error {
|
||||||
data, err := rc.Client.Get(rc.ctx, key).Bytes()
|
data, err := rc.Client.Get(rc.ctx, key).Bytes()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if err == redis.Nil {
|
if errors.Is(err, redis.Nil) {
|
||||||
return ErrNilValue
|
return ErrNotFound
|
||||||
}
|
}
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
err = cbor.Unmarshal(data, &value)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
return gobUnmarshal(data, &value)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (rc *RedisCache) Delete(key string) {
|
func (rc *RedisCache) Delete(key string) {
|
||||||
|
Loading…
Reference in New Issue
Block a user