2014-05-16 11:03:23 +04:00
|
|
|
package server
|
2014-05-03 10:55:12 +04:00
|
|
|
|
2014-05-16 11:03:23 +04:00
|
|
|
import (
|
|
|
|
"github.com/siddontang/ledisdb/ledis"
|
2014-05-23 07:44:50 +04:00
|
|
|
"strings"
|
2014-05-16 11:03:23 +04:00
|
|
|
)
|
2014-05-04 15:02:55 +04:00
|
|
|
|
2014-05-03 10:55:12 +04:00
|
|
|
func getCommand(c *client) error {
|
2014-05-04 15:02:55 +04:00
|
|
|
args := c.args
|
|
|
|
if len(args) != 1 {
|
|
|
|
return ErrCmdParams
|
|
|
|
}
|
|
|
|
|
2014-05-15 10:19:48 +04:00
|
|
|
if v, err := c.db.Get(args[0]); err != nil {
|
2014-05-04 15:02:55 +04:00
|
|
|
return err
|
|
|
|
} else {
|
|
|
|
c.writeBulk(v)
|
|
|
|
}
|
2014-05-03 10:55:12 +04:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func setCommand(c *client) error {
|
2014-05-04 15:02:55 +04:00
|
|
|
args := c.args
|
|
|
|
if len(args) < 2 {
|
|
|
|
return ErrCmdParams
|
|
|
|
}
|
|
|
|
|
2014-05-15 10:19:48 +04:00
|
|
|
if err := c.db.Set(args[0], args[1]); err != nil {
|
2014-05-04 15:02:55 +04:00
|
|
|
return err
|
|
|
|
} else {
|
|
|
|
c.writeStatus(OK)
|
|
|
|
}
|
|
|
|
|
2014-05-03 10:55:12 +04:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func getsetCommand(c *client) error {
|
2014-05-04 15:02:55 +04:00
|
|
|
args := c.args
|
|
|
|
if len(args) != 2 {
|
|
|
|
return ErrCmdParams
|
|
|
|
}
|
|
|
|
|
2014-05-15 10:19:48 +04:00
|
|
|
if v, err := c.db.GetSet(args[0], args[1]); err != nil {
|
2014-05-04 15:02:55 +04:00
|
|
|
return err
|
|
|
|
} else {
|
|
|
|
c.writeBulk(v)
|
|
|
|
}
|
|
|
|
|
2014-05-03 10:55:12 +04:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func setnxCommand(c *client) error {
|
2014-05-04 15:02:55 +04:00
|
|
|
args := c.args
|
|
|
|
if len(args) != 2 {
|
|
|
|
return ErrCmdParams
|
|
|
|
}
|
|
|
|
|
2014-05-15 10:19:48 +04:00
|
|
|
if n, err := c.db.SetNX(args[0], args[1]); err != nil {
|
2014-05-04 15:02:55 +04:00
|
|
|
return err
|
|
|
|
} else {
|
|
|
|
c.writeInteger(n)
|
|
|
|
}
|
|
|
|
|
2014-05-03 10:55:12 +04:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func existsCommand(c *client) error {
|
2014-05-04 15:02:55 +04:00
|
|
|
args := c.args
|
|
|
|
if len(args) != 1 {
|
|
|
|
return ErrCmdParams
|
|
|
|
}
|
|
|
|
|
2014-05-15 10:19:48 +04:00
|
|
|
if n, err := c.db.Exists(args[0]); err != nil {
|
2014-05-04 15:02:55 +04:00
|
|
|
return err
|
|
|
|
} else {
|
|
|
|
c.writeInteger(n)
|
|
|
|
}
|
|
|
|
|
2014-05-03 10:55:12 +04:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func incrCommand(c *client) error {
|
2014-05-04 15:02:55 +04:00
|
|
|
args := c.args
|
|
|
|
if len(args) != 1 {
|
|
|
|
return ErrCmdParams
|
|
|
|
}
|
|
|
|
|
2014-05-15 10:19:48 +04:00
|
|
|
if n, err := c.db.Incr(c.args[0]); err != nil {
|
2014-05-04 15:02:55 +04:00
|
|
|
return err
|
|
|
|
} else {
|
|
|
|
c.writeInteger(n)
|
|
|
|
}
|
|
|
|
|
2014-05-03 10:55:12 +04:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func decrCommand(c *client) error {
|
2014-05-04 15:02:55 +04:00
|
|
|
args := c.args
|
|
|
|
if len(args) != 1 {
|
|
|
|
return ErrCmdParams
|
|
|
|
}
|
|
|
|
|
2014-05-15 10:19:48 +04:00
|
|
|
if n, err := c.db.Decr(c.args[0]); err != nil {
|
2014-05-04 15:02:55 +04:00
|
|
|
return err
|
|
|
|
} else {
|
|
|
|
c.writeInteger(n)
|
|
|
|
}
|
|
|
|
|
2014-05-03 10:55:12 +04:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func incrbyCommand(c *client) error {
|
2014-05-04 15:02:55 +04:00
|
|
|
args := c.args
|
|
|
|
if len(args) != 2 {
|
|
|
|
return ErrCmdParams
|
|
|
|
}
|
|
|
|
|
2014-05-16 11:03:23 +04:00
|
|
|
delta, err := ledis.StrInt64(args[1], nil)
|
2014-05-04 15:02:55 +04:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2014-05-15 10:19:48 +04:00
|
|
|
if n, err := c.db.IncryBy(c.args[0], delta); err != nil {
|
2014-05-04 15:02:55 +04:00
|
|
|
return err
|
|
|
|
} else {
|
|
|
|
c.writeInteger(n)
|
|
|
|
}
|
|
|
|
|
2014-05-03 10:55:12 +04:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func decrbyCommand(c *client) error {
|
2014-05-04 15:02:55 +04:00
|
|
|
args := c.args
|
|
|
|
if len(args) != 2 {
|
|
|
|
return ErrCmdParams
|
|
|
|
}
|
|
|
|
|
2014-05-16 11:03:23 +04:00
|
|
|
delta, err := ledis.StrInt64(args[1], nil)
|
2014-05-04 15:02:55 +04:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2014-05-15 16:48:38 +04:00
|
|
|
if n, err := c.db.DecrBy(c.args[0], delta); err != nil {
|
2014-05-04 15:02:55 +04:00
|
|
|
return err
|
|
|
|
} else {
|
|
|
|
c.writeInteger(n)
|
|
|
|
}
|
|
|
|
|
2014-05-03 10:55:12 +04:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func delCommand(c *client) error {
|
2014-05-04 15:02:55 +04:00
|
|
|
args := c.args
|
|
|
|
if len(args) == 0 {
|
|
|
|
return ErrCmdParams
|
|
|
|
}
|
|
|
|
|
2014-05-16 04:56:32 +04:00
|
|
|
if n, err := c.db.Del(args...); err != nil {
|
2014-05-04 15:02:55 +04:00
|
|
|
return err
|
|
|
|
} else {
|
|
|
|
c.writeInteger(n)
|
|
|
|
}
|
|
|
|
|
2014-05-03 10:55:12 +04:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func msetCommand(c *client) error {
|
2014-05-04 15:02:55 +04:00
|
|
|
args := c.args
|
|
|
|
if len(args) == 0 || len(args)%2 != 0 {
|
|
|
|
return ErrCmdParams
|
|
|
|
}
|
|
|
|
|
2014-05-16 11:03:23 +04:00
|
|
|
kvs := make([]ledis.KVPair, len(args)/2)
|
2014-05-16 04:56:32 +04:00
|
|
|
for i := 0; i < len(kvs); i++ {
|
|
|
|
kvs[i].Key = args[2*i]
|
|
|
|
kvs[i].Value = args[2*i+1]
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := c.db.MSet(kvs...); err != nil {
|
2014-05-04 15:02:55 +04:00
|
|
|
return err
|
|
|
|
} else {
|
|
|
|
c.writeStatus(OK)
|
|
|
|
}
|
2014-05-03 10:55:12 +04:00
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2014-05-04 15:02:55 +04:00
|
|
|
// func setexCommand(c *client) error {
|
|
|
|
// return nil
|
|
|
|
// }
|
|
|
|
|
2014-05-03 10:55:12 +04:00
|
|
|
func mgetCommand(c *client) error {
|
2014-05-04 15:02:55 +04:00
|
|
|
args := c.args
|
|
|
|
if len(args) == 0 {
|
|
|
|
return ErrCmdParams
|
|
|
|
}
|
|
|
|
|
2014-05-16 04:56:32 +04:00
|
|
|
if v, err := c.db.MGet(args...); err != nil {
|
2014-05-04 15:02:55 +04:00
|
|
|
return err
|
|
|
|
} else {
|
|
|
|
c.writeArray(v)
|
|
|
|
}
|
|
|
|
|
2014-05-03 10:55:12 +04:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2014-05-23 07:44:50 +04:00
|
|
|
func scanCommand(c *client) error {
|
|
|
|
args := c.args
|
|
|
|
|
|
|
|
if len(args) != 1 && len(args) != 3 {
|
|
|
|
return ErrCmdParams
|
|
|
|
}
|
|
|
|
var offset int64
|
|
|
|
var count int64
|
|
|
|
var err error
|
|
|
|
|
|
|
|
if offset, err = ledis.StrInt64(args[0], nil); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
//now we only support count
|
|
|
|
if len(args) == 3 {
|
|
|
|
if strings.ToLower(ledis.String(args[1])) != "count" {
|
|
|
|
return ErrCmdParams
|
|
|
|
}
|
|
|
|
|
|
|
|
if count, err = ledis.StrInt64(args[2], nil); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if v, err := c.db.Scan(int(offset), int(count)); err != nil {
|
|
|
|
return err
|
|
|
|
} else {
|
|
|
|
c.writeArray(v)
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2014-05-03 10:55:12 +04:00
|
|
|
func init() {
|
|
|
|
register("decr", decrCommand)
|
|
|
|
register("decrby", decrbyCommand)
|
|
|
|
register("del", delCommand)
|
|
|
|
register("exists", existsCommand)
|
|
|
|
register("get", getCommand)
|
|
|
|
register("getset", getsetCommand)
|
|
|
|
register("incr", incrCommand)
|
|
|
|
register("incrby", incrbyCommand)
|
|
|
|
register("mget", mgetCommand)
|
|
|
|
register("mset", msetCommand)
|
|
|
|
register("set", setCommand)
|
|
|
|
register("setnx", setnxCommand)
|
2014-05-23 07:44:50 +04:00
|
|
|
register("scan", scanCommand)
|
2014-05-03 10:55:12 +04:00
|
|
|
}
|