redis/v2/command.go

package redis

import (
	"fmt"
	"strconv"
	"strings"
	"time"
)

type Cmder interface {
	args() []string
	parseReply(reader) error
	setErr(error)
	setVal(interface{})

	writeTimeout() *time.Duration
	readTimeout() *time.Duration

	Err() error
}

func setCmdsErr(cmds []Cmder, e error) {
	for _, cmd := range cmds {
		cmd.setErr(e)
	}
}

//------------------------------------------------------------------------------

type baseCmd struct {
	_args []string

	val interface{}
	err error

	_writeTimeout, _readTimeout *time.Duration
}

func newBaseCmd(args ...string) *baseCmd {
	return &baseCmd{
		_args: args,
	}
}

func (cmd *baseCmd) String() string {
	args := strings.Join(cmd._args, " ")
	if cmd.err != nil {
		return args + ": " + cmd.err.Error()
	} else if cmd.val != nil {
		return args + ": " + fmt.Sprint(cmd.val)
	}
	return args
}

func (cmd *baseCmd) Err() error {
	if cmd.err != nil {
		return cmd.err
	}
	if cmd.val == nil {
		return errValNotSet
	}
	return nil
}

func (cmd *baseCmd) args() []string {
	return cmd._args
}

func (cmd *baseCmd) setErr(err error) {
	if err == nil {
		panic("non-nil value expected")
	}
	cmd.err = err
}

func (cmd *baseCmd) setVal(val interface{}) {
	if val == nil {
		panic("non-nil value expected")
	}
	cmd.val = val
}

func (cmd *baseCmd) parseReply(rd reader) error {
	cmd.val, cmd.err = parseReply(rd, parseSlice)
	return cmd.err
}

func (cmd *baseCmd) readTimeout() *time.Duration {
	return cmd._readTimeout
}

func (cmd *baseCmd) setReadTimeout(d time.Duration) {
	cmd._readTimeout = &d
}

func (cmd *baseCmd) writeTimeout() *time.Duration {
	return cmd._writeTimeout
}

func (cmd *baseCmd) setWriteTimeout(d time.Duration) {
	cmd._writeTimeout = &d
}

//------------------------------------------------------------------------------

type Cmd struct {
	*baseCmd
}

func NewCmd(args ...string) *Cmd {
	return &Cmd{
		baseCmd: newBaseCmd(args...),
	}
}

func (cmd *Cmd) Val() interface{} {
	return cmd.val
}

//------------------------------------------------------------------------------

type StatusCmd struct {
	*baseCmd
}

func NewStatusCmd(args ...string) *StatusCmd {
	return &StatusCmd{
		baseCmd: newBaseCmd(args...),
	}
}

func (cmd *StatusCmd) Val() string {
	if cmd.val == nil {
		return ""
	}
	return cmd.val.(string)
}

//------------------------------------------------------------------------------

type IntCmd struct {
	*baseCmd
}

func NewIntCmd(args ...string) *IntCmd {
	return &IntCmd{
		baseCmd: newBaseCmd(args...),
	}
}

func (cmd *IntCmd) Val() int64 {
	if cmd.val == nil {
		return 0
	}
	return cmd.val.(int64)
}

//------------------------------------------------------------------------------

type DurationCmd struct {
	*baseCmd
	precision time.Duration
}

func NewDurationCmd(precision time.Duration, args ...string) *DurationCmd {
	return &DurationCmd{
		baseCmd:   newBaseCmd(args...),
		precision: precision,
	}
}

func (cmd *DurationCmd) parseReply(rd reader) error {
	cmd.val, cmd.err = parseReply(rd, nil)
	cmd.val = time.Duration(cmd.val.(int64)) * cmd.precision
	return cmd.err
}

func (cmd *DurationCmd) Val() time.Duration {
	if cmd.val == nil {
		return 0
	}
	return cmd.val.(time.Duration)
}

//------------------------------------------------------------------------------

type BoolCmd struct {
	*baseCmd
}

func NewBoolCmd(args ...string) *BoolCmd {
	return &BoolCmd{
		baseCmd: newBaseCmd(args...),
	}
}

func (cmd *BoolCmd) parseReply(rd reader) error {
	cmd.val, cmd.err = parseReply(rd, nil)
	cmd.val = cmd.val.(int64) == 1
	return cmd.err
}

func (cmd *BoolCmd) Val() bool {
	if cmd.val == nil {
		return false
	}
	return cmd.val.(bool)
}

//------------------------------------------------------------------------------

type StringCmd struct {
	*baseCmd
}

func NewStringCmd(args ...string) *StringCmd {
	return &StringCmd{
		baseCmd: newBaseCmd(args...),
	}
}

func (cmd *StringCmd) Val() string {
	if cmd.val == nil {
		return ""
	}
	return cmd.val.(string)
}

//------------------------------------------------------------------------------

type FloatCmd struct {
	*baseCmd
}

func NewFloatCmd(args ...string) *FloatCmd {
	return &FloatCmd{
		baseCmd: newBaseCmd(args...),
	}
}

func (cmd *FloatCmd) parseReply(rd reader) error {
	cmd.val, cmd.err = parseReply(rd, nil)
	if cmd.err != nil {
		return cmd.err
	}
	cmd.val, cmd.err = strconv.ParseFloat(cmd.val.(string), 64)
	return cmd.err
}

func (cmd *FloatCmd) Val() float64 {
	if cmd.val == nil {
		return 0
	}
	return cmd.val.(float64)
}

//------------------------------------------------------------------------------

type SliceCmd struct {
	*baseCmd
}

func NewSliceCmd(args ...string) *SliceCmd {
	return &SliceCmd{
		baseCmd: newBaseCmd(args...),
	}
}

func (cmd *SliceCmd) Val() []interface{} {
	if cmd.val == nil {
		return nil
	}
	return cmd.val.([]interface{})
}

//------------------------------------------------------------------------------

type StringSliceCmd struct {
	*baseCmd
}

func NewStringSliceCmd(args ...string) *StringSliceCmd {
	return &StringSliceCmd{
		baseCmd: newBaseCmd(args...),
	}
}

func (cmd *StringSliceCmd) parseReply(rd reader) error {
	cmd.val, cmd.err = parseReply(rd, parseStringSlice)
	return cmd.err
}

func (cmd *StringSliceCmd) Val() []string {
	if cmd.val == nil {
		return nil
	}
	return cmd.val.([]string)
}

//------------------------------------------------------------------------------

type BoolSliceCmd struct {
	*baseCmd
}

func NewBoolSliceCmd(args ...string) *BoolSliceCmd {
	return &BoolSliceCmd{
		baseCmd: newBaseCmd(args...),
	}
}

func (cmd *BoolSliceCmd) parseReply(rd reader) error {
	cmd.val, cmd.err = parseReply(rd, parseBoolSlice)
	return cmd.err
}

func (cmd *BoolSliceCmd) Val() []bool {
	if cmd.val == nil {
		return nil
	}
	return cmd.val.([]bool)
}

//------------------------------------------------------------------------------

type StringStringMapCmd struct {
	*baseCmd
}

func NewStringStringMapCmd(args ...string) *StringStringMapCmd {
	return &StringStringMapCmd{
		baseCmd: newBaseCmd(args...),
	}
}

func (cmd *StringStringMapCmd) parseReply(rd reader) error {
	cmd.val, cmd.err = parseReply(rd, parseStringStringMap)
	return cmd.err
}

func (cmd *StringStringMapCmd) Val() map[string]string {
	if cmd.val == nil {
		return nil
	}
	return cmd.val.(map[string]string)
}

//------------------------------------------------------------------------------

type StringFloatMapCmd struct {
	*baseCmd
}

func NewStringFloatMapCmd(args ...string) *StringFloatMapCmd {
	return &StringFloatMapCmd{
		baseCmd: newBaseCmd(args...),
	}
}

func (cmd *StringFloatMapCmd) parseReply(rd reader) error {
	cmd.val, cmd.err = parseReply(rd, parseStringFloatMap)
	return cmd.err
}

func (cmd *StringFloatMapCmd) Val() map[string]float64 {
	if cmd.val == nil {
		return nil
	}
	return cmd.val.(map[string]float64)
}