add cmd get, delete, keys

cmd/data.go

@@ -0,0 +1,83 @@
+package cmd
+
+import (
+	"fmt"
+	"github.com/desertbit/grumble"
+)
+
+func putData(c *grumble.Context) error {
+	key := c.Args.String("key")
+	value := c.Args.String("value")
+	if key == "" || value == "" {
+		fmt.Println("key or value is empty")
+		return nil
+	}
+	if db == nil {
+		fmt.Println("start server first")
+		return nil
+	}
+	err := db.Put([]byte(key), []byte(value))
+	if err != nil {
+		fmt.Println("put data error: ", err)
+		return err
+	}
+	fmt.Println("put data success")
+	return nil
+}
+
+func getData(c *grumble.Context) error {
+	key := c.Args.String("key")
+	if key == "" {
+		fmt.Println("key is empty")
+		return nil
+	}
+	if db == nil {
+		fmt.Println("start server first")
+		return nil
+	}
+	value, err := db.Get([]byte(key))
+	if err != nil {
+		fmt.Println("get data error: ", err)
+		return err
+	}
+	fmt.Println(string(value))
+	return nil
+}
+
+func deleteKey(c *grumble.Context) error {
+	key := c.Args.String("key")
+	if key == "" {
+		fmt.Println("key is empty")
+		return nil
+	}
+	if db == nil {
+		fmt.Println("start server first")
+		return nil
+	}
+	err := db.Delete([]byte(key))
+	if err != nil {
+		fmt.Println("delete key error: ", err)
+		return err
+	}
+	fmt.Println("delete key success")
+	return nil
+}
+
+func getKeys(c *grumble.Context) error {
+	if db == nil {
+		fmt.Println("start server first")
+		return nil
+	}
+	list := db.GetListKeys()
+	fmt.Println("Total keys: ", len(list))
+	for i, bytes := range list {
+		fmt.Printf(string(bytes[:]) + "\t")
+		if i%8 == 7 {
+			fmt.Println()
+		}
+	}
+	if len(list)%8 != 0 {
+		fmt.Println()
+	}
+	return nil
+}

cmd/root.go

@@ -22,6 +22,7 @@ func register(app *grumble.App) {
 		Help: "clean server",
 		Run:  cleanServer,
 	})
+
 	app.AddCommand(&grumble.Command{
 		Name: "put",
 		Help: "put data",
@@ -31,4 +32,29 @@ func register(app *grumble.App) {
 			a.String("value", "value", grumble.Default(""))
 		},
 	})
+
+	app.AddCommand(&grumble.Command{
+		Name: "get",
+		Help: "get data",
+		Run:  getData,
+		Args: func(a *grumble.Args) {
+			a.String("key", "key", grumble.Default(""))
+		},
+	})
+
+	app.AddCommand(&grumble.Command{
+		Name: "delete",
+		Help: "delete key",
+		Run:  deleteKey,
+		Args: func(a *grumble.Args) {
+			a.String("key", "key", grumble.Default(""))
+		},
+	})
+
+	app.AddCommand(&grumble.Command{
+		Name: "keys",
+		Help: "list keys",
+		Run:  getKeys,
+	})
+
 }