cli.go

package rdl

import (
	"time"

	"github.com/garyburd/redigo/redis"
	goredis "github.com/go-redis/redis"
)

// cmdScript is the script to implement the `SetIfValIs` interface
var cmdScript string = `
local v=redis.call("get", KEYS[1]);
if (not v) or (v == ARGV[3]) then
    return redis.call("setex", KEYS[1], ARGV[1], ARGV[2])
end
`

// RedigoClient implemented `SetIfValIs` interface
type RedigoClient struct {
	pool *redis.Pool
}

func NewRedigo(pool *redis.Pool) *RedigoClient {
	return &RedigoClient{
		pool: pool,
	}
}

func (cli *RedigoClient) SetIfValIs(
	k string,
	newVal string,
	ex time.Duration,
	origin string,
) (ok bool) {
	c := cli.pool.Get()
	defer c.Close()

	s := redis.NewScript(1, cmdScript)
	reply, err := s.Do(c, k, int64(ex.Seconds()), newVal, origin)
	return err == nil && reply == "OK"
}

// GoRedisClient implemented `SetIfValIs` interface
type GoRedisClient struct {
	conn *goredis.Client
}

func NewGoRedis(conn *goredis.Client) *GoRedisClient {
	return &GoRedisClient{
		conn: conn,
	}
}

func (cli *GoRedisClient) SetIfValIs(
	k string,
	newVal string,
	ex time.Duration,
	origin string,
) (ok bool) {
	s := goredis.NewScript(cmdScript)
	reply, err := s.Run(
		cli.conn,
		[]string{k},
		int64(ex.Seconds()),
		newVal,
		origin,
	).Result()
	return err == nil && reply == "OK"
}