1 Star 0 Fork 0

WuYijin/go-redis

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
commands.go 39.31 KB
一键复制 编辑 原始数据 按行查看 历史
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749
package redis
import (
"io"
"log"
"strconv"
"time"
)
func formatInt(i int64) string {
return strconv.FormatInt(i, 10)
}
func formatUint(i uint64) string {
return strconv.FormatUint(i, 10)
}
func formatFloat(f float64) string {
return strconv.FormatFloat(f, 'f', -1, 64)
}
func readTimeout(timeout time.Duration) time.Duration {
if timeout == 0 {
return 0
}
return timeout + time.Second
}
func usePrecise(dur time.Duration) bool {
return dur < time.Second || dur%time.Second != 0
}
func formatMs(dur time.Duration) string {
if dur > 0 && dur < time.Millisecond {
log.Printf(
"redis: specified duration is %s, but minimal supported value is %s",
dur, time.Millisecond,
)
}
return formatInt(int64(dur / time.Millisecond))
}
func formatSec(dur time.Duration) string {
if dur > 0 && dur < time.Second {
log.Printf(
"redis: specified duration is %s, but minimal supported value is %s",
dur, time.Second,
)
}
return formatInt(int64(dur / time.Second))
}
type commandable struct {
process func(cmd Cmder)
}
func (c *commandable) Process(cmd Cmder) {
c.process(cmd)
}
//------------------------------------------------------------------------------
func (c *commandable) Auth(password string) *StatusCmd {
cmd := newKeylessStatusCmd("AUTH", password)
c.Process(cmd)
return cmd
}
func (c *commandable) Echo(message string) *StringCmd {
cmd := NewStringCmd("ECHO", message)
cmd._clusterKeyPos = 0
c.Process(cmd)
return cmd
}
func (c *commandable) Ping() *StatusCmd {
cmd := newKeylessStatusCmd("PING")
c.Process(cmd)
return cmd
}
func (c *commandable) Quit() *StatusCmd {
panic("not implemented")
}
func (c *commandable) Select(index int64) *StatusCmd {
cmd := newKeylessStatusCmd("SELECT", index)
c.Process(cmd)
return cmd
}
//------------------------------------------------------------------------------
func (c *commandable) Del(keys ...string) *IntCmd {
args := make([]interface{}, 1+len(keys))
args[0] = "DEL"
for i, key := range keys {
args[1+i] = key
}
cmd := NewIntCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) Dump(key string) *StringCmd {
cmd := NewStringCmd("DUMP", key)
c.Process(cmd)
return cmd
}
func (c *commandable) Exists(key string) *BoolCmd {
cmd := NewBoolCmd("EXISTS", key)
c.Process(cmd)
return cmd
}
func (c *commandable) Expire(key string, expiration time.Duration) *BoolCmd {
cmd := NewBoolCmd("EXPIRE", key, formatSec(expiration))
c.Process(cmd)
return cmd
}
func (c *commandable) ExpireAt(key string, tm time.Time) *BoolCmd {
cmd := NewBoolCmd("EXPIREAT", key, tm.Unix())
c.Process(cmd)
return cmd
}
func (c *commandable) Keys(pattern string) *StringSliceCmd {
cmd := NewStringSliceCmd("KEYS", pattern)
c.Process(cmd)
return cmd
}
func (c *commandable) Migrate(host, port, key string, db int64, timeout time.Duration) *StatusCmd {
cmd := NewStatusCmd(
"MIGRATE",
host,
port,
key,
db,
formatMs(timeout),
)
cmd._clusterKeyPos = 3
cmd.setReadTimeout(readTimeout(timeout))
c.Process(cmd)
return cmd
}
func (c *commandable) Move(key string, db int64) *BoolCmd {
cmd := NewBoolCmd("MOVE", key, db)
c.Process(cmd)
return cmd
}
func (c *commandable) ObjectRefCount(keys ...string) *IntCmd {
args := make([]interface{}, 2+len(keys))
args[0] = "OBJECT"
args[1] = "REFCOUNT"
for i, key := range keys {
args[2+i] = key
}
cmd := NewIntCmd(args...)
cmd._clusterKeyPos = 2
c.Process(cmd)
return cmd
}
func (c *commandable) ObjectEncoding(keys ...string) *StringCmd {
args := make([]interface{}, 2+len(keys))
args[0] = "OBJECT"
args[1] = "ENCODING"
for i, key := range keys {
args[2+i] = key
}
cmd := NewStringCmd(args...)
cmd._clusterKeyPos = 2
c.Process(cmd)
return cmd
}
func (c *commandable) ObjectIdleTime(keys ...string) *DurationCmd {
args := make([]interface{}, 2+len(keys))
args[0] = "OBJECT"
args[1] = "IDLETIME"
for i, key := range keys {
args[2+i] = key
}
cmd := NewDurationCmd(time.Second, args...)
cmd._clusterKeyPos = 2
c.Process(cmd)
return cmd
}
func (c *commandable) Persist(key string) *BoolCmd {
cmd := NewBoolCmd("PERSIST", key)
c.Process(cmd)
return cmd
}
func (c *commandable) PExpire(key string, expiration time.Duration) *BoolCmd {
cmd := NewBoolCmd("PEXPIRE", key, formatMs(expiration))
c.Process(cmd)
return cmd
}
func (c *commandable) PExpireAt(key string, tm time.Time) *BoolCmd {
cmd := NewBoolCmd(
"PEXPIREAT",
key,
tm.UnixNano()/int64(time.Millisecond),
)
c.Process(cmd)
return cmd
}
func (c *commandable) PTTL(key string) *DurationCmd {
cmd := NewDurationCmd(time.Millisecond, "PTTL", key)
c.Process(cmd)
return cmd
}
func (c *commandable) RandomKey() *StringCmd {
cmd := NewStringCmd("RANDOMKEY")
c.Process(cmd)
return cmd
}
func (c *commandable) Rename(key, newkey string) *StatusCmd {
cmd := NewStatusCmd("RENAME", key, newkey)
c.Process(cmd)
return cmd
}
func (c *commandable) RenameNX(key, newkey string) *BoolCmd {
cmd := NewBoolCmd("RENAMENX", key, newkey)
c.Process(cmd)
return cmd
}
func (c *commandable) Restore(key string, ttl time.Duration, value string) *StatusCmd {
cmd := NewStatusCmd(
"RESTORE",
key,
formatMs(ttl),
value,
)
c.Process(cmd)
return cmd
}
func (c *commandable) RestoreReplace(key string, ttl time.Duration, value string) *StatusCmd {
cmd := NewStatusCmd(
"RESTORE",
key,
formatMs(ttl),
value,
"REPLACE",
)
c.Process(cmd)
return cmd
}
type Sort struct {
By string
Offset, Count float64
Get []string
Order string
IsAlpha bool
Store string
}
func (c *commandable) Sort(key string, sort Sort) *StringSliceCmd {
args := []interface{}{"SORT", key}
if sort.By != "" {
args = append(args, "BY", sort.By)
}
if sort.Offset != 0 || sort.Count != 0 {
args = append(args, "LIMIT", sort.Offset, sort.Count)
}
for _, get := range sort.Get {
args = append(args, "GET", get)
}
if sort.Order != "" {
args = append(args, sort.Order)
}
if sort.IsAlpha {
args = append(args, "ALPHA")
}
if sort.Store != "" {
args = append(args, "STORE", sort.Store)
}
cmd := NewStringSliceCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) TTL(key string) *DurationCmd {
cmd := NewDurationCmd(time.Second, "TTL", key)
c.Process(cmd)
return cmd
}
func (c *commandable) Type(key string) *StatusCmd {
cmd := NewStatusCmd("TYPE", key)
c.Process(cmd)
return cmd
}
func (c *commandable) Scan(cursor int64, match string, count int64) *ScanCmd {
args := []interface{}{"SCAN", cursor}
if match != "" {
args = append(args, "MATCH", match)
}
if count > 0 {
args = append(args, "COUNT", count)
}
cmd := NewScanCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) SScan(key string, cursor int64, match string, count int64) *ScanCmd {
args := []interface{}{"SSCAN", key, cursor}
if match != "" {
args = append(args, "MATCH", match)
}
if count > 0 {
args = append(args, "COUNT", count)
}
cmd := NewScanCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) HScan(key string, cursor int64, match string, count int64) *ScanCmd {
args := []interface{}{"HSCAN", key, cursor}
if match != "" {
args = append(args, "MATCH", match)
}
if count > 0 {
args = append(args, "COUNT", count)
}
cmd := NewScanCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) ZScan(key string, cursor int64, match string, count int64) *ScanCmd {
args := []interface{}{"ZSCAN", key, cursor}
if match != "" {
args = append(args, "MATCH", match)
}
if count > 0 {
args = append(args, "COUNT", count)
}
cmd := NewScanCmd(args...)
c.Process(cmd)
return cmd
}
//------------------------------------------------------------------------------
func (c *commandable) Append(key, value string) *IntCmd {
cmd := NewIntCmd("APPEND", key, value)
c.Process(cmd)
return cmd
}
type BitCount struct {
Start, End int64
}
func (c *commandable) BitCount(key string, bitCount *BitCount) *IntCmd {
args := []interface{}{"BITCOUNT", key}
if bitCount != nil {
args = append(
args,
bitCount.Start,
bitCount.End,
)
}
cmd := NewIntCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) bitOp(op, destKey string, keys ...string) *IntCmd {
args := make([]interface{}, 3+len(keys))
args[0] = "BITOP"
args[1] = op
args[2] = destKey
for i, key := range keys {
args[3+i] = key
}
cmd := NewIntCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) BitOpAnd(destKey string, keys ...string) *IntCmd {
return c.bitOp("AND", destKey, keys...)
}
func (c *commandable) BitOpOr(destKey string, keys ...string) *IntCmd {
return c.bitOp("OR", destKey, keys...)
}
func (c *commandable) BitOpXor(destKey string, keys ...string) *IntCmd {
return c.bitOp("XOR", destKey, keys...)
}
func (c *commandable) BitOpNot(destKey string, key string) *IntCmd {
return c.bitOp("NOT", destKey, key)
}
func (c *commandable) BitPos(key string, bit int64, pos ...int64) *IntCmd {
args := make([]interface{}, 3+len(pos))
args[0] = "BITPOS"
args[1] = key
args[2] = bit
switch len(pos) {
case 0:
case 1:
args[3] = pos[0]
case 2:
args[3] = pos[0]
args[4] = pos[1]
default:
panic("too many arguments")
}
cmd := NewIntCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) Decr(key string) *IntCmd {
cmd := NewIntCmd("DECR", key)
c.Process(cmd)
return cmd
}
func (c *commandable) DecrBy(key string, decrement int64) *IntCmd {
cmd := NewIntCmd("DECRBY", key, decrement)
c.Process(cmd)
return cmd
}
func (c *commandable) Get(key string) *StringCmd {
cmd := NewStringCmd("GET", key)
c.Process(cmd)
return cmd
}
func (c *commandable) GetBit(key string, offset int64) *IntCmd {
cmd := NewIntCmd("GETBIT", key, offset)
c.Process(cmd)
return cmd
}
func (c *commandable) GetRange(key string, start, end int64) *StringCmd {
cmd := NewStringCmd("GETRANGE", key, start, end)
c.Process(cmd)
return cmd
}
func (c *commandable) GetSet(key string, value interface{}) *StringCmd {
cmd := NewStringCmd("GETSET", key, value)
c.Process(cmd)
return cmd
}
func (c *commandable) Incr(key string) *IntCmd {
cmd := NewIntCmd("INCR", key)
c.Process(cmd)
return cmd
}
func (c *commandable) IncrBy(key string, value int64) *IntCmd {
cmd := NewIntCmd("INCRBY", key, value)
c.Process(cmd)
return cmd
}
func (c *commandable) IncrByFloat(key string, value float64) *FloatCmd {
cmd := NewFloatCmd("INCRBYFLOAT", key, value)
c.Process(cmd)
return cmd
}
func (c *commandable) MGet(keys ...string) *SliceCmd {
args := make([]interface{}, 1+len(keys))
args[0] = "MGET"
for i, key := range keys {
args[1+i] = key
}
cmd := NewSliceCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) MSet(pairs ...string) *StatusCmd {
args := make([]interface{}, 1+len(pairs))
args[0] = "MSET"
for i, pair := range pairs {
args[1+i] = pair
}
cmd := NewStatusCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) MSetNX(pairs ...string) *BoolCmd {
args := make([]interface{}, 1+len(pairs))
args[0] = "MSETNX"
for i, pair := range pairs {
args[1+i] = pair
}
cmd := NewBoolCmd(args...)
c.Process(cmd)
return cmd
}
// Redis `SET key value [expiration]` command.
//
// Zero expiration means the key has no expiration time.
func (c *commandable) Set(key string, value interface{}, expiration time.Duration) *StatusCmd {
args := make([]interface{}, 3, 5)
args[0] = "SET"
args[1] = key
args[2] = value
if expiration > 0 {
if usePrecise(expiration) {
args = append(args, "PX", formatMs(expiration))
} else {
args = append(args, "EX", formatSec(expiration))
}
}
cmd := NewStatusCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) SetBit(key string, offset int64, value int) *IntCmd {
cmd := NewIntCmd(
"SETBIT",
key,
offset,
value,
)
c.Process(cmd)
return cmd
}
// Redis `SET key value [expiration] NX` command.
//
// Zero expiration means the key has no expiration time.
func (c *commandable) SetNX(key string, value interface{}, expiration time.Duration) *BoolCmd {
var cmd *BoolCmd
if expiration == 0 {
// Use old `SETNX` to support old Redis versions.
cmd = NewBoolCmd("SETNX", key, value)
} else {
if usePrecise(expiration) {
cmd = NewBoolCmd("SET", key, value, "PX", formatMs(expiration), "NX")
} else {
cmd = NewBoolCmd("SET", key, value, "EX", formatSec(expiration), "NX")
}
}
c.Process(cmd)
return cmd
}
// Redis `SET key value [expiration] XX` command.
//
// Zero expiration means the key has no expiration time.
func (c *Client) SetXX(key string, value interface{}, expiration time.Duration) *BoolCmd {
var cmd *BoolCmd
if usePrecise(expiration) {
cmd = NewBoolCmd("SET", key, value, "PX", formatMs(expiration), "XX")
} else {
cmd = NewBoolCmd("SET", key, value, "EX", formatSec(expiration), "XX")
}
c.Process(cmd)
return cmd
}
func (c *commandable) SetRange(key string, offset int64, value string) *IntCmd {
cmd := NewIntCmd("SETRANGE", key, offset, value)
c.Process(cmd)
return cmd
}
func (c *commandable) StrLen(key string) *IntCmd {
cmd := NewIntCmd("STRLEN", key)
c.Process(cmd)
return cmd
}
//------------------------------------------------------------------------------
func (c *commandable) HDel(key string, fields ...string) *IntCmd {
args := make([]interface{}, 2+len(fields))
args[0] = "HDEL"
args[1] = key
for i, field := range fields {
args[2+i] = field
}
cmd := NewIntCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) HExists(key, field string) *BoolCmd {
cmd := NewBoolCmd("HEXISTS", key, field)
c.Process(cmd)
return cmd
}
func (c *commandable) HGet(key, field string) *StringCmd {
cmd := NewStringCmd("HGET", key, field)
c.Process(cmd)
return cmd
}
func (c *commandable) HGetAll(key string) *StringSliceCmd {
cmd := NewStringSliceCmd("HGETALL", key)
c.Process(cmd)
return cmd
}
func (c *commandable) HGetAllMap(key string) *StringStringMapCmd {
cmd := NewStringStringMapCmd("HGETALL", key)
c.Process(cmd)
return cmd
}
func (c *commandable) HIncrBy(key, field string, incr int64) *IntCmd {
cmd := NewIntCmd("HINCRBY", key, field, incr)
c.Process(cmd)
return cmd
}
func (c *commandable) HIncrByFloat(key, field string, incr float64) *FloatCmd {
cmd := NewFloatCmd("HINCRBYFLOAT", key, field, incr)
c.Process(cmd)
return cmd
}
func (c *commandable) HKeys(key string) *StringSliceCmd {
cmd := NewStringSliceCmd("HKEYS", key)
c.Process(cmd)
return cmd
}
func (c *commandable) HLen(key string) *IntCmd {
cmd := NewIntCmd("HLEN", key)
c.Process(cmd)
return cmd
}
func (c *commandable) HMGet(key string, fields ...string) *SliceCmd {
args := make([]interface{}, 2+len(fields))
args[0] = "HMGET"
args[1] = key
for i, field := range fields {
args[2+i] = field
}
cmd := NewSliceCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) HMSet(key, field, value string, pairs ...string) *StatusCmd {
args := make([]interface{}, 4+len(pairs))
args[0] = "HMSET"
args[1] = key
args[2] = field
args[3] = value
for i, pair := range pairs {
args[4+i] = pair
}
cmd := NewStatusCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) HSet(key, field, value string) *BoolCmd {
cmd := NewBoolCmd("HSET", key, field, value)
c.Process(cmd)
return cmd
}
func (c *commandable) HSetNX(key, field, value string) *BoolCmd {
cmd := NewBoolCmd("HSETNX", key, field, value)
c.Process(cmd)
return cmd
}
func (c *commandable) HVals(key string) *StringSliceCmd {
cmd := NewStringSliceCmd("HVALS", key)
c.Process(cmd)
return cmd
}
//------------------------------------------------------------------------------
func (c *commandable) BLPop(timeout time.Duration, keys ...string) *StringSliceCmd {
args := make([]interface{}, 2+len(keys))
args[0] = "BLPOP"
for i, key := range keys {
args[1+i] = key
}
args[len(args)-1] = formatSec(timeout)
cmd := NewStringSliceCmd(args...)
cmd.setReadTimeout(readTimeout(timeout))
c.Process(cmd)
return cmd
}
func (c *commandable) BRPop(timeout time.Duration, keys ...string) *StringSliceCmd {
args := make([]interface{}, 2+len(keys))
args[0] = "BRPOP"
for i, key := range keys {
args[1+i] = key
}
args[len(args)-1] = formatSec(timeout)
cmd := NewStringSliceCmd(args...)
cmd.setReadTimeout(readTimeout(timeout))
c.Process(cmd)
return cmd
}
func (c *commandable) BRPopLPush(source, destination string, timeout time.Duration) *StringCmd {
cmd := NewStringCmd(
"BRPOPLPUSH",
source,
destination,
formatSec(timeout),
)
cmd.setReadTimeout(readTimeout(timeout))
c.Process(cmd)
return cmd
}
func (c *commandable) LIndex(key string, index int64) *StringCmd {
cmd := NewStringCmd("LINDEX", key, index)
c.Process(cmd)
return cmd
}
func (c *commandable) LInsert(key, op, pivot, value string) *IntCmd {
cmd := NewIntCmd("LINSERT", key, op, pivot, value)
c.Process(cmd)
return cmd
}
func (c *commandable) LLen(key string) *IntCmd {
cmd := NewIntCmd("LLEN", key)
c.Process(cmd)
return cmd
}
func (c *commandable) LPop(key string) *StringCmd {
cmd := NewStringCmd("LPOP", key)
c.Process(cmd)
return cmd
}
func (c *commandable) LPush(key string, values ...string) *IntCmd {
args := make([]interface{}, 2+len(values))
args[0] = "LPUSH"
args[1] = key
for i, value := range values {
args[2+i] = value
}
cmd := NewIntCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) LPushX(key, value string) *IntCmd {
cmd := NewIntCmd("LPUSHX", key, value)
c.Process(cmd)
return cmd
}
func (c *commandable) LRange(key string, start, stop int64) *StringSliceCmd {
cmd := NewStringSliceCmd(
"LRANGE",
key,
start,
stop,
)
c.Process(cmd)
return cmd
}
func (c *commandable) LRem(key string, count int64, value string) *IntCmd {
cmd := NewIntCmd("LREM", key, count, value)
c.Process(cmd)
return cmd
}
func (c *commandable) LSet(key string, index int64, value string) *StatusCmd {
cmd := NewStatusCmd("LSET", key, index, value)
c.Process(cmd)
return cmd
}
func (c *commandable) LTrim(key string, start, stop int64) *StatusCmd {
cmd := NewStatusCmd(
"LTRIM",
key,
start,
stop,
)
c.Process(cmd)
return cmd
}
func (c *commandable) RPop(key string) *StringCmd {
cmd := NewStringCmd("RPOP", key)
c.Process(cmd)
return cmd
}
func (c *commandable) RPopLPush(source, destination string) *StringCmd {
cmd := NewStringCmd("RPOPLPUSH", source, destination)
c.Process(cmd)
return cmd
}
func (c *commandable) RPush(key string, values ...string) *IntCmd {
args := make([]interface{}, 2+len(values))
args[0] = "RPUSH"
args[1] = key
for i, value := range values {
args[2+i] = value
}
cmd := NewIntCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) RPushX(key string, value string) *IntCmd {
cmd := NewIntCmd("RPUSHX", key, value)
c.Process(cmd)
return cmd
}
//------------------------------------------------------------------------------
func (c *commandable) SAdd(key string, members ...string) *IntCmd {
args := make([]interface{}, 2+len(members))
args[0] = "SADD"
args[1] = key
for i, member := range members {
args[2+i] = member
}
cmd := NewIntCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) SCard(key string) *IntCmd {
cmd := NewIntCmd("SCARD", key)
c.Process(cmd)
return cmd
}
func (c *commandable) SDiff(keys ...string) *StringSliceCmd {
args := make([]interface{}, 1+len(keys))
args[0] = "SDIFF"
for i, key := range keys {
args[1+i] = key
}
cmd := NewStringSliceCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) SDiffStore(destination string, keys ...string) *IntCmd {
args := make([]interface{}, 2+len(keys))
args[0] = "SDIFFSTORE"
args[1] = destination
for i, key := range keys {
args[2+i] = key
}
cmd := NewIntCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) SInter(keys ...string) *StringSliceCmd {
args := make([]interface{}, 1+len(keys))
args[0] = "SINTER"
for i, key := range keys {
args[1+i] = key
}
cmd := NewStringSliceCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) SInterStore(destination string, keys ...string) *IntCmd {
args := make([]interface{}, 2+len(keys))
args[0] = "SINTERSTORE"
args[1] = destination
for i, key := range keys {
args[2+i] = key
}
cmd := NewIntCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) SIsMember(key, member string) *BoolCmd {
cmd := NewBoolCmd("SISMEMBER", key, member)
c.Process(cmd)
return cmd
}
func (c *commandable) SMembers(key string) *StringSliceCmd {
cmd := NewStringSliceCmd("SMEMBERS", key)
c.Process(cmd)
return cmd
}
func (c *commandable) SMove(source, destination, member string) *BoolCmd {
cmd := NewBoolCmd("SMOVE", source, destination, member)
c.Process(cmd)
return cmd
}
func (c *commandable) SPop(key string) *StringCmd {
cmd := NewStringCmd("SPOP", key)
c.Process(cmd)
return cmd
}
// Redis `SRANDMEMBER key` command.
func (c *commandable) SRandMember(key string) *StringCmd {
cmd := NewStringCmd("SRANDMEMBER", key)
c.Process(cmd)
return cmd
}
// Redis `SRANDMEMBER key count` command.
func (c *commandable) SRandMemberN(key string, count int64) *StringSliceCmd {
cmd := NewStringSliceCmd("SRANDMEMBER", key, count)
c.Process(cmd)
return cmd
}
func (c *commandable) SRem(key string, members ...string) *IntCmd {
args := make([]interface{}, 2+len(members))
args[0] = "SREM"
args[1] = key
for i, member := range members {
args[2+i] = member
}
cmd := NewIntCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) SUnion(keys ...string) *StringSliceCmd {
args := make([]interface{}, 1+len(keys))
args[0] = "SUNION"
for i, key := range keys {
args[1+i] = key
}
cmd := NewStringSliceCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) SUnionStore(destination string, keys ...string) *IntCmd {
args := make([]interface{}, 2+len(keys))
args[0] = "SUNIONSTORE"
args[1] = destination
for i, key := range keys {
args[2+i] = key
}
cmd := NewIntCmd(args...)
c.Process(cmd)
return cmd
}
//------------------------------------------------------------------------------
// Sorted set member.
type Z struct {
Score float64
Member interface{}
}
// Sorted set store operation.
type ZStore struct {
Weights []int64
// Can be SUM, MIN or MAX.
Aggregate string
}
func (c *commandable) zAdd(a []interface{}, n int, members ...Z) *IntCmd {
for i, m := range members {
a[n+2*i] = m.Score
a[n+2*i+1] = m.Member
}
cmd := NewIntCmd(a...)
c.Process(cmd)
return cmd
}
// Redis `ZADD key score member [score member ...]` command.
func (c *commandable) ZAdd(key string, members ...Z) *IntCmd {
const n = 2
a := make([]interface{}, n+2*len(members))
a[0], a[1] = "ZADD", key
return c.zAdd(a, n, members...)
}
// Redis `ZADD key NX score member [score member ...]` command.
func (c *commandable) ZAddNX(key string, members ...Z) *IntCmd {
const n = 3
a := make([]interface{}, n+2*len(members))
a[0], a[1], a[2] = "ZADD", key, "NX"
return c.zAdd(a, n, members...)
}
// Redis `ZADD key XX score member [score member ...]` command.
func (c *commandable) ZAddXX(key string, members ...Z) *IntCmd {
const n = 3
a := make([]interface{}, n+2*len(members))
a[0], a[1], a[2] = "ZADD", key, "XX"
return c.zAdd(a, n, members...)
}
// Redis `ZADD key CH score member [score member ...]` command.
func (c *commandable) ZAddCh(key string, members ...Z) *IntCmd {
const n = 3
a := make([]interface{}, n+2*len(members))
a[0], a[1], a[2] = "ZADD", key, "CH"
return c.zAdd(a, n, members...)
}
// Redis `ZADD key NX CH score member [score member ...]` command.
func (c *commandable) ZAddNXCh(key string, members ...Z) *IntCmd {
const n = 4
a := make([]interface{}, n+2*len(members))
a[0], a[1], a[2], a[3] = "ZADD", key, "NX", "CH"
return c.zAdd(a, n, members...)
}
// Redis `ZADD key XX CH score member [score member ...]` command.
func (c *commandable) ZAddXXCh(key string, members ...Z) *IntCmd {
const n = 4
a := make([]interface{}, n+2*len(members))
a[0], a[1], a[2], a[3] = "ZADD", key, "XX", "CH"
return c.zAdd(a, n, members...)
}
func (c *commandable) zIncr(a []interface{}, n int, members ...Z) *FloatCmd {
for i, m := range members {
a[n+2*i] = m.Score
a[n+2*i+1] = m.Member
}
cmd := NewFloatCmd(a...)
c.Process(cmd)
return cmd
}
// Redis `ZADD key INCR score member` command.
func (c *commandable) ZIncr(key string, member Z) *FloatCmd {
const n = 3
a := make([]interface{}, n+2)
a[0], a[1], a[2] = "ZADD", key, "INCR"
return c.zIncr(a, n, member)
}
// Redis `ZADD key NX INCR score member` command.
func (c *commandable) ZIncrNX(key string, member Z) *FloatCmd {
const n = 4
a := make([]interface{}, n+2)
a[0], a[1], a[2], a[3] = "ZADD", key, "INCR", "NX"
return c.zIncr(a, n, member)
}
// Redis `ZADD key XX INCR score member` command.
func (c *commandable) ZIncrXX(key string, member Z) *FloatCmd {
const n = 4
a := make([]interface{}, n+2)
a[0], a[1], a[2], a[3] = "ZADD", key, "INCR", "XX"
return c.zIncr(a, n, member)
}
func (c *commandable) ZCard(key string) *IntCmd {
cmd := NewIntCmd("ZCARD", key)
c.Process(cmd)
return cmd
}
func (c *commandable) ZCount(key, min, max string) *IntCmd {
cmd := NewIntCmd("ZCOUNT", key, min, max)
c.Process(cmd)
return cmd
}
func (c *commandable) ZIncrBy(key string, increment float64, member string) *FloatCmd {
cmd := NewFloatCmd("ZINCRBY", key, increment, member)
c.Process(cmd)
return cmd
}
func (c *commandable) ZInterStore(
destination string,
store ZStore,
keys ...string,
) *IntCmd {
args := make([]interface{}, 3+len(keys))
args[0] = "ZINTERSTORE"
args[1] = destination
args[2] = strconv.Itoa(len(keys))
for i, key := range keys {
args[3+i] = key
}
if len(store.Weights) > 0 {
args = append(args, "WEIGHTS")
for _, weight := range store.Weights {
args = append(args, weight)
}
}
if store.Aggregate != "" {
args = append(args, "AGGREGATE", store.Aggregate)
}
cmd := NewIntCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) zRange(key string, start, stop int64, withScores bool) *StringSliceCmd {
args := []interface{}{
"ZRANGE",
key,
start,
stop,
}
if withScores {
args = append(args, "WITHSCORES")
}
cmd := NewStringSliceCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) ZRange(key string, start, stop int64) *StringSliceCmd {
return c.zRange(key, start, stop, false)
}
func (c *commandable) ZRangeWithScores(key string, start, stop int64) *ZSliceCmd {
cmd := NewZSliceCmd("ZRANGE", key, start, stop, "WITHSCORES")
c.Process(cmd)
return cmd
}
// TODO: Rename to something more generic in v4
type ZRangeByScore struct {
Min, Max string
Offset, Count int64
}
func (c *commandable) zRangeBy(zcmd, key string, opt ZRangeByScore, withScores bool) *StringSliceCmd {
args := []interface{}{zcmd, key, opt.Min, opt.Max}
if withScores {
args = append(args, "WITHSCORES")
}
if opt.Offset != 0 || opt.Count != 0 {
args = append(
args,
"LIMIT",
opt.Offset,
opt.Count,
)
}
cmd := NewStringSliceCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) ZRangeByScore(key string, opt ZRangeByScore) *StringSliceCmd {
return c.zRangeBy("ZRANGEBYSCORE", key, opt, false)
}
func (c *commandable) ZRangeByLex(key string, opt ZRangeByScore) *StringSliceCmd {
return c.zRangeBy("ZRANGEBYLEX", key, opt, false)
}
func (c *commandable) ZRangeByScoreWithScores(key string, opt ZRangeByScore) *ZSliceCmd {
args := []interface{}{"ZRANGEBYSCORE", key, opt.Min, opt.Max, "WITHSCORES"}
if opt.Offset != 0 || opt.Count != 0 {
args = append(
args,
"LIMIT",
opt.Offset,
opt.Count,
)
}
cmd := NewZSliceCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) ZRank(key, member string) *IntCmd {
cmd := NewIntCmd("ZRANK", key, member)
c.Process(cmd)
return cmd
}
func (c *commandable) ZRem(key string, members ...string) *IntCmd {
args := make([]interface{}, 2+len(members))
args[0] = "ZREM"
args[1] = key
for i, member := range members {
args[2+i] = member
}
cmd := NewIntCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) ZRemRangeByRank(key string, start, stop int64) *IntCmd {
cmd := NewIntCmd(
"ZREMRANGEBYRANK",
key,
start,
stop,
)
c.Process(cmd)
return cmd
}
func (c *commandable) ZRemRangeByScore(key, min, max string) *IntCmd {
cmd := NewIntCmd("ZREMRANGEBYSCORE", key, min, max)
c.Process(cmd)
return cmd
}
func (c *commandable) ZRevRange(key string, start, stop int64) *StringSliceCmd {
cmd := NewStringSliceCmd("ZREVRANGE", key, start, stop)
c.Process(cmd)
return cmd
}
func (c *commandable) ZRevRangeWithScores(key string, start, stop int64) *ZSliceCmd {
cmd := NewZSliceCmd("ZREVRANGE", key, start, stop, "WITHSCORES")
c.Process(cmd)
return cmd
}
func (c *commandable) zRevRangeBy(zcmd, key string, opt ZRangeByScore) *StringSliceCmd {
args := []interface{}{zcmd, key, opt.Max, opt.Min}
if opt.Offset != 0 || opt.Count != 0 {
args = append(
args,
"LIMIT",
opt.Offset,
opt.Count,
)
}
cmd := NewStringSliceCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) ZRevRangeByScore(key string, opt ZRangeByScore) *StringSliceCmd {
return c.zRevRangeBy("ZREVRANGEBYSCORE", key, opt)
}
func (c commandable) ZRevRangeByLex(key string, opt ZRangeByScore) *StringSliceCmd {
return c.zRevRangeBy("ZREVRANGEBYLEX", key, opt)
}
func (c *commandable) ZRevRangeByScoreWithScores(key string, opt ZRangeByScore) *ZSliceCmd {
args := []interface{}{"ZREVRANGEBYSCORE", key, opt.Max, opt.Min, "WITHSCORES"}
if opt.Offset != 0 || opt.Count != 0 {
args = append(
args,
"LIMIT",
opt.Offset,
opt.Count,
)
}
cmd := NewZSliceCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) ZRevRank(key, member string) *IntCmd {
cmd := NewIntCmd("ZREVRANK", key, member)
c.Process(cmd)
return cmd
}
func (c *commandable) ZScore(key, member string) *FloatCmd {
cmd := NewFloatCmd("ZSCORE", key, member)
c.Process(cmd)
return cmd
}
func (c *commandable) ZUnionStore(dest string, store ZStore, keys ...string) *IntCmd {
args := make([]interface{}, 3+len(keys))
args[0] = "ZUNIONSTORE"
args[1] = dest
args[2] = strconv.Itoa(len(keys))
for i, key := range keys {
args[3+i] = key
}
if len(store.Weights) > 0 {
args = append(args, "WEIGHTS")
for _, weight := range store.Weights {
args = append(args, weight)
}
}
if store.Aggregate != "" {
args = append(args, "AGGREGATE", store.Aggregate)
}
cmd := NewIntCmd(args...)
c.Process(cmd)
return cmd
}
//------------------------------------------------------------------------------
func (c *commandable) BgRewriteAOF() *StatusCmd {
cmd := NewStatusCmd("BGREWRITEAOF")
cmd._clusterKeyPos = 0
c.Process(cmd)
return cmd
}
func (c *commandable) BgSave() *StatusCmd {
cmd := NewStatusCmd("BGSAVE")
cmd._clusterKeyPos = 0
c.Process(cmd)
return cmd
}
func (c *commandable) ClientKill(ipPort string) *StatusCmd {
cmd := NewStatusCmd("CLIENT", "KILL", ipPort)
cmd._clusterKeyPos = 0
c.Process(cmd)
return cmd
}
func (c *commandable) ClientList() *StringCmd {
cmd := NewStringCmd("CLIENT", "LIST")
cmd._clusterKeyPos = 0
c.Process(cmd)
return cmd
}
func (c *commandable) ClientPause(dur time.Duration) *BoolCmd {
cmd := NewBoolCmd("CLIENT", "PAUSE", formatMs(dur))
cmd._clusterKeyPos = 0
c.Process(cmd)
return cmd
}
// ClientSetName assigns a name to the one of many connections in the pool.
func (c *commandable) ClientSetName(name string) *BoolCmd {
cmd := NewBoolCmd("CLIENT", "SETNAME", name)
c.Process(cmd)
return cmd
}
// ClientGetName returns the name of the one of many connections in the pool.
func (c *Client) ClientGetName() *StringCmd {
cmd := NewStringCmd("CLIENT", "GETNAME")
c.Process(cmd)
return cmd
}
func (c *commandable) ConfigGet(parameter string) *SliceCmd {
cmd := NewSliceCmd("CONFIG", "GET", parameter)
cmd._clusterKeyPos = 0
c.Process(cmd)
return cmd
}
func (c *commandable) ConfigResetStat() *StatusCmd {
cmd := NewStatusCmd("CONFIG", "RESETSTAT")
cmd._clusterKeyPos = 0
c.Process(cmd)
return cmd
}
func (c *commandable) ConfigSet(parameter, value string) *StatusCmd {
cmd := NewStatusCmd("CONFIG", "SET", parameter, value)
cmd._clusterKeyPos = 0
c.Process(cmd)
return cmd
}
func (c *commandable) DbSize() *IntCmd {
cmd := NewIntCmd("DBSIZE")
cmd._clusterKeyPos = 0
c.Process(cmd)
return cmd
}
func (c *commandable) FlushAll() *StatusCmd {
cmd := newKeylessStatusCmd("FLUSHALL")
c.Process(cmd)
return cmd
}
func (c *commandable) FlushDb() *StatusCmd {
cmd := newKeylessStatusCmd("FLUSHDB")
c.Process(cmd)
return cmd
}
func (c *commandable) Info() *StringCmd {
cmd := NewStringCmd("INFO")
cmd._clusterKeyPos = 0
c.Process(cmd)
return cmd
}
func (c *commandable) LastSave() *IntCmd {
cmd := NewIntCmd("LASTSAVE")
cmd._clusterKeyPos = 0
c.Process(cmd)
return cmd
}
func (c *commandable) Save() *StatusCmd {
cmd := newKeylessStatusCmd("SAVE")
c.Process(cmd)
return cmd
}
func (c *commandable) shutdown(modifier string) *StatusCmd {
var args []interface{}
if modifier == "" {
args = []interface{}{"SHUTDOWN"}
} else {
args = []interface{}{"SHUTDOWN", modifier}
}
cmd := newKeylessStatusCmd(args...)
c.Process(cmd)
if err := cmd.Err(); err != nil {
if err == io.EOF {
// Server quit as expected.
cmd.err = nil
}
} else {
// Server did not quit. String reply contains the reason.
cmd.err = errorf(cmd.val)
cmd.val = ""
}
return cmd
}
func (c *commandable) Shutdown() *StatusCmd {
return c.shutdown("")
}
func (c *commandable) ShutdownSave() *StatusCmd {
return c.shutdown("SAVE")
}
func (c *commandable) ShutdownNoSave() *StatusCmd {
return c.shutdown("NOSAVE")
}
func (c *commandable) SlaveOf(host, port string) *StatusCmd {
cmd := newKeylessStatusCmd("SLAVEOF", host, port)
c.Process(cmd)
return cmd
}
func (c *commandable) SlowLog() {
panic("not implemented")
}
func (c *commandable) Sync() {
panic("not implemented")
}
func (c *commandable) Time() *StringSliceCmd {
cmd := NewStringSliceCmd("TIME")
cmd._clusterKeyPos = 0
c.Process(cmd)
return cmd
}
//------------------------------------------------------------------------------
func (c *commandable) Eval(script string, keys []string, args []string) *Cmd {
cmdArgs := make([]interface{}, 3+len(keys)+len(args))
cmdArgs[0] = "EVAL"
cmdArgs[1] = script
cmdArgs[2] = strconv.Itoa(len(keys))
for i, key := range keys {
cmdArgs[3+i] = key
}
pos := 3 + len(keys)
for i, arg := range args {
cmdArgs[pos+i] = arg
}
cmd := NewCmd(cmdArgs...)
if len(keys) > 0 {
cmd._clusterKeyPos = 3
}
c.Process(cmd)
return cmd
}
func (c *commandable) EvalSha(sha1 string, keys []string, args []string) *Cmd {
cmdArgs := make([]interface{}, 3+len(keys)+len(args))
cmdArgs[0] = "EVALSHA"
cmdArgs[1] = sha1
cmdArgs[2] = strconv.Itoa(len(keys))
for i, key := range keys {
cmdArgs[3+i] = key
}
pos := 3 + len(keys)
for i, arg := range args {
cmdArgs[pos+i] = arg
}
cmd := NewCmd(cmdArgs...)
if len(keys) > 0 {
cmd._clusterKeyPos = 3
}
c.Process(cmd)
return cmd
}
func (c *commandable) ScriptExists(scripts ...string) *BoolSliceCmd {
args := make([]interface{}, 2+len(scripts))
args[0] = "SCRIPT"
args[1] = "EXISTS"
for i, script := range scripts {
args[2+i] = script
}
cmd := NewBoolSliceCmd(args...)
cmd._clusterKeyPos = 0
c.Process(cmd)
return cmd
}
func (c *commandable) ScriptFlush() *StatusCmd {
cmd := newKeylessStatusCmd("SCRIPT", "FLUSH")
c.Process(cmd)
return cmd
}
func (c *commandable) ScriptKill() *StatusCmd {
cmd := newKeylessStatusCmd("SCRIPT", "KILL")
c.Process(cmd)
return cmd
}
func (c *commandable) ScriptLoad(script string) *StringCmd {
cmd := NewStringCmd("SCRIPT", "LOAD", script)
cmd._clusterKeyPos = 0
c.Process(cmd)
return cmd
}
//------------------------------------------------------------------------------
func (c *commandable) DebugObject(key string) *StringCmd {
cmd := NewStringCmd("DEBUG", "OBJECT", key)
cmd._clusterKeyPos = 2
c.Process(cmd)
return cmd
}
//------------------------------------------------------------------------------
func (c *commandable) PubSubChannels(pattern string) *StringSliceCmd {
args := []interface{}{"PUBSUB", "CHANNELS"}
if pattern != "*" {
args = append(args, pattern)
}
cmd := NewStringSliceCmd(args...)
cmd._clusterKeyPos = 0
c.Process(cmd)
return cmd
}
func (c *commandable) PubSubNumSub(channels ...string) *StringIntMapCmd {
args := make([]interface{}, 2+len(channels))
args[0] = "PUBSUB"
args[1] = "NUMSUB"
for i, channel := range channels {
args[2+i] = channel
}
cmd := NewStringIntMapCmd(args...)
cmd._clusterKeyPos = 0
c.Process(cmd)
return cmd
}
func (c *commandable) PubSubNumPat() *IntCmd {
cmd := NewIntCmd("PUBSUB", "NUMPAT")
cmd._clusterKeyPos = 0
c.Process(cmd)
return cmd
}
//------------------------------------------------------------------------------
func (c *commandable) ClusterSlots() *ClusterSlotCmd {
cmd := NewClusterSlotCmd("CLUSTER", "slots")
cmd._clusterKeyPos = 0
c.Process(cmd)
return cmd
}
func (c *commandable) ClusterNodes() *StringCmd {
cmd := NewStringCmd("CLUSTER", "nodes")
cmd._clusterKeyPos = 0
c.Process(cmd)
return cmd
}
func (c *commandable) ClusterMeet(host, port string) *StatusCmd {
cmd := newKeylessStatusCmd("CLUSTER", "meet", host, port)
c.Process(cmd)
return cmd
}
func (c *commandable) ClusterReplicate(nodeID string) *StatusCmd {
cmd := newKeylessStatusCmd("CLUSTER", "replicate", nodeID)
c.Process(cmd)
return cmd
}
func (c *commandable) ClusterInfo() *StringCmd {
cmd := NewStringCmd("CLUSTER", "info")
cmd._clusterKeyPos = 0
c.Process(cmd)
return cmd
}
func (c *commandable) ClusterFailover() *StatusCmd {
cmd := newKeylessStatusCmd("CLUSTER", "failover")
c.Process(cmd)
return cmd
}
func (c *commandable) ClusterAddSlots(slots ...int) *StatusCmd {
args := make([]interface{}, 2+len(slots))
args[0] = "CLUSTER"
args[1] = "ADDSLOTS"
for i, num := range slots {
args[2+i] = strconv.Itoa(num)
}
cmd := newKeylessStatusCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) ClusterAddSlotsRange(min, max int) *StatusCmd {
size := max - min + 1
slots := make([]int, size)
for i := 0; i < size; i++ {
slots[i] = min + i
}
return c.ClusterAddSlots(slots...)
}
//------------------------------------------------------------------------------
func (c *commandable) GeoAdd(key string, geoLocation ...*GeoLocation) *IntCmd {
args := make([]interface{}, 2+3*len(geoLocation))
args[0] = "GEOADD"
args[1] = key
for i, eachLoc := range geoLocation {
args[2+3*i] = eachLoc.Longitude
args[2+3*i+1] = eachLoc.Latitude
args[2+3*i+2] = eachLoc.Name
}
cmd := NewIntCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) geoRadius(args []interface{}, query *GeoRadiusQuery) *GeoLocationCmd {
args = append(args, query.Radius)
if query.Unit != "" {
args = append(args, query.Unit)
} else {
args = append(args, "km")
}
if query.WithCoord {
args = append(args, "WITHCOORD")
}
if query.WithDist {
args = append(args, "WITHDIST")
}
if query.WithGeoHash {
args = append(args, "WITHHASH")
}
if query.Count > 0 {
args = append(args, "COUNT", query.Count)
}
if query.Sort != "" {
args = append(args, query.Sort)
}
cmd := NewGeoLocationCmd(args...)
c.Process(cmd)
return cmd
}
func (c *commandable) GeoRadius(key string, longitude, latitude float64, query *GeoRadiusQuery) *GeoLocationCmd {
args := []interface{}{"GEORADIUS", key, longitude, latitude}
return c.geoRadius(args, query)
}
func (c *commandable) GeoRadiusByMember(key, member string, query *GeoRadiusQuery) *GeoLocationCmd {
args := []interface{}{"GEORADIUSBYMEMBER", key, member}
return c.geoRadius(args, query)
}
func (c *commandable) GeoDist(key string, member1, member2, unit string) *FloatCmd {
if unit == "" {
unit = "km"
}
cmd := NewFloatCmd("GEODIST", key, member1, member2, unit)
c.Process(cmd)
return cmd
}
func (c *commandable) GeoHash(key string, members ...string) *StringSliceCmd {
args := make([]interface{}, 2+len(members))
args[0] = "GEOHASH"
args[1] = key
for i, member := range members {
args[2+i] = member
}
cmd := NewStringSliceCmd(args...)
c.Process(cmd)
return cmd
}
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/yijin_wuwu/go-redis.git
git@gitee.com:yijin_wuwu/go-redis.git
yijin_wuwu
go-redis
go-redis
v3.2.13

搜索帮助