Merge pull request #1 from redis-go/list

List

Author: robinbraemer

Gopkg.lock

@@ -39,7 +39,7 @@ go get -u github.com/redis-go/redis
 - [x] active key expirer
 - [ ] Implementing data structures
   - [x] String
-  - [ ] List
+  - [x] List
   - [ ] Set
   - [ ] Sorted Set
   - [ ] Hash

README.md

@@ -34,6 +34,15 @@ func (r *Redis) NewClient(conn redcon.Conn) *Client {
 	return c
 }
 
+// NextClientId atomically gets and increments a counter to return the next client id.
+func (r *Redis) NextClientId() ClientId {
+	r.Mu().Lock()
+	defer r.Mu().Unlock()
+	id := r.nextClientId
+	r.nextClientId++
+	return id
+}
+
 // Clients gets the current connected clients.
 func (r *Redis) Clients() Clients {
 	r.Mu().RLock()

client.go

@@ -14,8 +14,8 @@ func GetCommand(c *Client, cmd redcon.Command) {
 		return
 	}
 
-	if i.ValueType() != StringType {
-		c.Conn().WriteError(fmt.Sprintf("%s: key is a %s not a %s", WrongTypeErr, i.ValueTypeFancy(), StringTypeFancy))
+	if i.Type() != StringType {
+		c.Conn().WriteError(fmt.Sprintf("%s: key is a %s not a %s", WrongTypeErr, i.TypeFancy(), StringTypeFancy))
 		return
 	}
 

del.go renamed to cmd_del.go

@@ -0,0 +1,34 @@
+package redis
+
+import (
+	"fmt"
+	"github.com/redis-go/redcon"
+)
+
+func LPopCommand(c *Client, cmd redcon.Command) {
+	if len(cmd.Args) < 2 {
+		c.Conn().WriteError(fmt.Sprintf(WrongNumOfArgsErr, "lpop"))
+		return
+	}
+	key := string(cmd.Args[1])
+
+	db := c.Db()
+	i := db.GetOrExpire(&key, true)
+	if i == nil {
+		c.Conn().WriteNull()
+		return
+	} else if i.Type() != ListType {
+		c.Conn().WriteError(fmt.Sprintf("%s: key is a %s not a %s", WrongTypeErr, i.TypeFancy(), ListTypeFancy))
+		return
+	}
+
+	l := i.(*List)
+	c.Redis().Mu().Lock()
+	v, b := l.LPop()
+	if b {
+		db.Delete(&key)
+	}
+	c.Redis().Mu().Unlock()
+
+	c.Conn().WriteBulkString(*v)
+}

get.go renamed to cmd_get.go

@@ -0,0 +1,37 @@
+package redis
+
+import (
+	"fmt"
+	"github.com/redis-go/redcon"
+	"time"
+)
+
+func LPushCommand(c *Client, cmd redcon.Command) {
+	if len(cmd.Args) < 3 {
+		c.Conn().WriteError(fmt.Sprintf(WrongNumOfArgsErr, "lpush"))
+		return
+	}
+	key := string(cmd.Args[1])
+	fmt.Println("KEY:", key)
+	db := c.Db()
+	i := db.GetOrExpire(&key, true)
+	if i == nil {
+		i = NewList()
+		db.Set(&key, i, false, time.Time{})
+		fmt.Println("CREATED NEW LIST")
+	} else if i.Type() != ListType {
+		c.Conn().WriteError(fmt.Sprintf("%s: key is a %s not a %s", WrongTypeErr, i.TypeFancy(), ListTypeFancy))
+		return
+	}
+
+	l := i.(*List)
+	var length int
+	c.Redis().Mu().Lock()
+	for j := 2; j < len(cmd.Args); j++ {
+		v := string(cmd.Args[j])
+		length = l.LPush(&v)
+	}
+	c.Redis().Mu().Unlock()
+
+	c.Conn().WriteInt(length)
+}

cmd_lpop.go

@@ -0,0 +1,45 @@
+package redis
+
+import (
+	"fmt"
+	"github.com/redis-go/redcon"
+	"strconv"
+)
+
+func LRangeCommand(c *Client, cmd redcon.Command) {
+	if len(cmd.Args) < 3 {
+		c.Conn().WriteError(fmt.Sprintf(WrongNumOfArgsErr, "lrange"))
+		return
+	}
+	key := string(cmd.Args[1])
+	start, err := strconv.Atoi(string(cmd.Args[2]))
+	if err != nil {
+		c.Conn().WriteError(fmt.Sprintf("%s: %s", InvalidIntErr, err.Error()))
+		return
+	}
+	end, err := strconv.Atoi(string(cmd.Args[3]))
+	if err != nil {
+		c.Conn().WriteError(fmt.Sprintf("%s: %s", InvalidIntErr, err.Error()))
+		return
+	}
+
+	db := c.Db()
+	i := db.GetOrExpire(&key, true)
+	if i == nil {
+		c.Conn().WriteNull()
+		return
+	} else if i.Type() != ListType {
+		c.Conn().WriteError(fmt.Sprintf("%s: key is a %s not a %s", WrongTypeErr, i.TypeFancy(), ListTypeFancy))
+		return
+	}
+
+	l := i.(*List)
+	c.Redis().Mu().RLock()
+	values := l.LRange(start, end)
+	c.Redis().Mu().RUnlock()
+
+	c.Conn().WriteArray(len(values))
+	for _, v := range values {
+		c.Conn().WriteBulkString(v)
+	}
+}

cmd_lpush.go

@@ -0,0 +1,34 @@
+package redis
+
+import (
+	"fmt"
+	"github.com/redis-go/redcon"
+)
+
+func RPopCommand(c *Client, cmd redcon.Command) {
+	if len(cmd.Args) < 2 {
+		c.Conn().WriteError(fmt.Sprintf(WrongNumOfArgsErr, "rpop"))
+		return
+	}
+	key := string(cmd.Args[1])
+
+	db := c.Db()
+	i := db.GetOrExpire(&key, true)
+	if i == nil {
+		c.Conn().WriteNull()
+		return
+	} else if i.Type() != ListType {
+		c.Conn().WriteError(fmt.Sprintf("%s: key is a %s not a %s", WrongTypeErr, i.TypeFancy(), ListTypeFancy))
+		return
+	}
+
+	l := i.(*List)
+	c.Redis().Mu().Lock()
+	v, b := l.RPop()
+	if b {
+		db.Delete(&key)
+	}
+	c.Redis().Mu().Unlock()
+
+	c.Conn().WriteBulkString(*v)
+}