Added Lset

Author: hoisie

redis-dump.go

@@ -28,7 +28,6 @@ func dump_db(port int, db int, output io.Writer) {
         return
     }
 
-
     for _, key := range (keys) {
         typ, _ := client.Type(key)
 

redis.go

@@ -73,16 +73,16 @@ func readResponse(reader *bufio.Reader) (interface{}, os.Error) {
 
     //read until the first non-whitespace line
     for {
-    	line, err = reader.ReadString('\n')
-    	if len(line) == 0 || err != nil {
+        line, err = reader.ReadString('\n')
+        if len(line) == 0 || err != nil {
             return nil, err
-    	}
-    	line = strings.TrimSpace(line)
+        }
+        line = strings.TrimSpace(line)
         if len(line) > 0 {
             break
         }
     }
-     
+
     if line[0] == '+' {
         return strings.TrimSpace(line[1:]), nil
     }
@@ -207,7 +207,7 @@ func (client *Client) Get(name string) ([]byte, os.Error) {
     res, _ := client.sendCommand(cmd)
 
     if res == nil {
-        return nil, RedisError("Key `"+name+"` does not exist")
+        return nil, RedisError("Key `" + name + "` does not exist")
     }
 
     data := res.([]byte)
@@ -303,6 +303,16 @@ func (client *Client) Lindex(name string, index int) ([]byte, os.Error) {
     return res.([]byte), nil
 }
 
+func (client *Client) Lset(name string, index int, value []byte) os.Error {
+    cmd := fmt.Sprintf("LSET %s %d %d\r\n%s\r\n", name, index, len(value), value)
+    _, err := client.sendCommand(cmd)
+    if err != nil {
+        return err
+    }
+
+    return nil
+}
+
 func (client *Client) Rpush(name string, value []byte) os.Error {
     cmd := fmt.Sprintf("RPUSH %s %d\r\n%s\r\n", name, len(value), value)
     _, err := client.sendCommand(cmd)

redis_test.go

@@ -21,6 +21,7 @@ var client Client
 func init() {
     runtime.GOMAXPROCS(2)
     client.Addr = "127.0.0.1:7379"
+    client.Db = 13
 }
 
 func TestBasic(t *testing.T) {
@@ -134,8 +135,6 @@ func TestList(t *testing.T) {
         client.Rpush("l", strings.Bytes(v))
     }
 
-    //var members [][]byte
-
     if l, err := client.Llen("l"); err != nil || l != 5 {
         t.Fatal("Llen failed", err.String())
     }
@@ -146,6 +145,18 @@ func TestList(t *testing.T) {
         }
     }
 
+    for i := 0; i < len(vals); i++ {
+        if err := client.Lset("l", i, strings.Bytes("a")); err != nil {
+            t.Fatal("Lset failed", err.String())
+        }
+    }
+
+    for i := 0; i < len(vals); i++ {
+        if val, err := client.Lindex("l", i); err != nil || string(val) != "a" {
+            t.Fatal("Lindex failed", err.String())
+        }
+    }
+
     client.Del("l")
 
 }