From e3311a768f4f60c40a67ec239c79962eb070a915 Mon Sep 17 00:00:00 2001 From: Asim Aslam Date: Wed, 12 Jun 2019 07:48:31 +0100 Subject: [PATCH] Move data to store --- redis.go | 88 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 88 insertions(+) create mode 100644 redis.go diff --git a/redis.go b/redis.go new file mode 100644 index 0000000..b386313 --- /dev/null +++ b/redis.go @@ -0,0 +1,88 @@ +package redis + +import ( + "github.com/micro/go-micro/options" + "github.com/micro/go-micro/store" + redis "gopkg.in/redis.v3" +) + +type rkv struct { + options.Options + Client *redis.Client +} + +func (r *rkv) Read(key string) (*store.Record, error) { + val, err := r.Client.Get(key).Bytes() + + if err != nil && err == redis.Nil { + return nil, store.ErrNotFound + } else if err != nil { + return nil, err + } + + if val == nil { + return nil, store.ErrNotFound + } + + d, err := r.Client.TTL(key).Result() + if err != nil { + return nil, err + } + + return &store.Record{ + Key: key, + Value: val, + Expiry: d, + }, nil +} + +func (r *rkv) Delete(key string) error { + return r.Client.Del(key).Err() +} + +func (r *rkv) Write(record *store.Record) error { + return r.Client.Set(record.Key, record.Value, record.Expiry).Err() +} + +func (r *rkv) Dump() ([]*store.Record, error) { + keys, err := r.Client.Keys("*").Result() + if err != nil { + return nil, err + } + var vals []*store.Record + for _, k := range keys { + i, err := r.Read(k) + if err != nil { + return nil, err + } + vals = append(vals, i) + } + return vals, nil +} + +func (r *rkv) String() string { + return "redis" +} + +func NewStore(opts ...options.Option) store.Store { + options := options.NewOptions(opts...) + + var nodes []string + + if n, ok := options.Values().Get("store.nodes"); ok { + nodes = n.([]string) + } + + if len(nodes) == 0 { + nodes = []string{"127.0.0.1:6379"} + } + + return &rkv{ + Options: options, + Client: redis.NewClient(&redis.Options{ + Addr: nodes[0], + Password: "", // no password set + DB: 0, // use default DB + }), + } +}