83 lines
1.4 KiB
Go
83 lines
1.4 KiB
Go
|
package redis
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/micro/go-micro/sync/data"
|
||
|
|
redis "gopkg.in/redis.v3"
|
||
|
|
)
|
||
|
|
|
||
|
|
type rkv struct {
|
||
|
|
Client *redis.Client
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *rkv) Read(key string) (*data.Record, error) {
|
||
|
|
val, err := r.Client.Get(key).Bytes()
|
||
|
|
|
||
|
|
if err != nil && err == redis.Nil {
|
||
|
|
return nil, data.ErrNotFound
|
||
|
|
} else if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
if val == nil {
|
||
|
|
return nil, data.ErrNotFound
|
||
|
|
}
|
||
|
|
|
||
|
|
d, err := r.Client.TTL(key).Result()
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return &data.Record{
|
||
|
|
Key: key,
|
||
|
|
Value: val,
|
||
|
|
Expiration: d,
|
||
|
|
}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *rkv) Delete(key string) error {
|
||
|
|
return r.Client.Del(key).Err()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *rkv) Write(record *data.Record) error {
|
||
|
|
return r.Client.Set(record.Key, record.Value, record.Expiration).Err()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *rkv) Dump() ([]*data.Record, error) {
|
||
|
|
keys, err := r.Client.Keys("*").Result()
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
var vals []*data.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 NewData(opts ...data.Option) data.Data {
|
||
|
|
var options data.Options
|
||
|
|
for _, o := range opts {
|
||
|
|
o(&options)
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(options.Nodes) == 0 {
|
||
|
|
options.Nodes = []string{"127.0.0.1:6379"}
|
||
|
|
}
|
||
|
|
|
||
|
|
return &rkv{
|
||
|
|
Client: redis.NewClient(&redis.Options{
|
||
|
|
Addr: options.Nodes[0],
|
||
|
|
Password: "", // no password set
|
||
|
|
DB: 0, // use default DB
|
||
|
|
}),
|
||
|
|
}
|
||
|
|
}
|