add transaction command

Author: Sora233

.gitignore

@@ -4,4 +4,5 @@
 /vendor
 /.coverage
 /coverage.html
-.test*
+.test*
+/play.go

README.md

@@ -2,7 +2,8 @@
 
 an interactive [buntdb](https://github.com/tidwall/buntdb) shell client
 
-![build](https://github.com/Sora233/buntdb-cli/workflows/.github/workflows/ci.yml/badge.svg) [![Coverage Status](https://coveralls.io/repos/github/Sora233/buntdb-cli/badge.svg)](https://coveralls.io/github/Sora233/buntdb-cli)
+![build](https://github.com/Sora233/buntdb-cli/workflows/.github/workflows/ci.yml/badge.svg)
+[![Coverage Status](https://coveralls.io/repos/github/Sora233/buntdb-cli/badge.svg)](https://coveralls.io/github/Sora233/buntdb-cli)
 
 
 ----
@@ -30,6 +31,10 @@ go get -u -v github.com/Sora233/buntdb-cli
     * set
     * del
     * ttl
+    * rbegin (begin a readonly transaction)
+    * rwbegin (begin a read/write transaction)
+    * commit
+    * rollback
     * show
     * keys
     * use

cli/completer.go

@@ -2,6 +2,7 @@ package cli
 
 import (
 	"fmt"
+	"github.com/Sora233/buntdb-cli/db"
 	"github.com/c-bata/go-prompt"
 	"strings"
 )
@@ -33,6 +34,18 @@ func cmdCompleter(cmd string) []prompt.Suggest {
 		{Text: "use", Description: "change db"},
 		{Text: "exit", Description: "exit buntdb shell client"},
 	}
+	tx, _ := db.GetCurrentTransaction()
+	if tx == nil {
+		cmds = append(cmds,
+			prompt.Suggest{Text: "rbegin", Description: "open a readonly transaction"},
+			prompt.Suggest{Text: "rwbegin", Description: "open a read/write transaction"},
+		)
+	} else {
+		cmds = append(cmds,
+			prompt.Suggest{Text: "rollback", Description: "rollback a transaction"},
+			prompt.Suggest{Text: "commit", Description: "commit a transaction"},
+		)
+	}
 	return prompt.FilterHasPrefix(cmds, cmd, true)
 }
 
@@ -44,13 +57,18 @@ func optionCompleter(cmd string, args []string) []prompt.Suggest {
 	case "get":
 	case "set":
 	case "del":
+	case "ttl":
 	case "show":
 		return []prompt.Suggest{
 			{Text: "index"},
 			{Text: "db"},
 		}
 	case "keys":
 	case "use":
+	case "rbegin":
+	case "rwbegin":
+	case "rollback":
+	case "commit":
 	default:
 		return []prompt.Suggest{}
 	}

cli/executor.go

@@ -2,18 +2,15 @@ package cli
 
 import (
 	"fmt"
+	"github.com/Sora233/buntdb-cli/db"
 	"github.com/alecthomas/kong"
-	"runtime"
 	"strings"
 )
 
 func BuntdbExecutor(s string) {
 	s = strings.TrimSpace(s)
-	if s == "" {
+	if s == "" || s == "exit" {
 		return
-	} else if s == "exit" {
-		fmt.Println("bye")
-		runtime.Goexit()
 	}
 	args := ArgSplit(s)
 
@@ -34,8 +31,54 @@ func BuntdbExecutor(s string) {
 		fmt.Printf("ERR: %v\n", err)
 		return
 	}
-	err = ctx.Run()
-	if err != nil {
-		fmt.Printf("ERR: %v\n", err)
+	cmd := ctx.Selected().Name
+	if cmd == "rwbegin" || cmd == "rbegin" || cmd == "rollback" || cmd == "commit" {
+		err = ctx.Run()
+		if err != nil {
+			fmt.Printf("ERR: %v\n", err)
+		}
+		return
+	}
+	tx, rw := db.GetCurrentTransaction()
+	if ctx.Selected().Name == "use" {
+		fmt.Printf("tx %v", tx)
+		err = ctx.Run(tx)
+		if err != nil {
+			fmt.Printf("ERR: %v\n", err)
+		}
+		return
+	}
+	if tx != nil {
+		if Debug {
+			fmt.Printf("got current %v transaction\n", db.RWDescribe(rw))
+		}
+		err = ctx.Run(tx)
+		if err != nil {
+			fmt.Printf("ERR: %v\n", err)
+			return
+		}
+	} else {
+		if Debug {
+			fmt.Printf("no transaction, create a rw transaction\n")
+		}
+		tx, err := db.Begin(true)
+		if err != nil {
+			fmt.Printf("ERR: %v\n", err)
+			return
+		}
+		defer func() {
+			if Debug {
+				fmt.Printf("transaction commit\n")
+			}
+			err := db.Commit()
+			if err != nil {
+				fmt.Printf("ERR: commit error %v\n", err)
+			}
+		}()
+		err = ctx.Run(tx)
+		if err != nil {
+			fmt.Printf("ERR: %v\n", err)
+			return
+		}
 	}
 }

cli/executor_test.go

@@ -61,6 +61,53 @@ func TestBuntdbExecutor(t *testing.T) {
 		return nil
 	})
 	BuntdbExecutor("del b")
+	BuntdbExecutor("del a")
+
+	BuntdbExecutor("rwbegin")
+	BuntdbExecutor("set x y")
+	BuntdbExecutor("set y x")
+	BuntdbExecutor("commit")
+
+	bd.View(func(tx *buntdb.Tx) error {
+		val, err := tx.Get("x")
+		assert.Nil(t, err)
+		assert.Equal(t, "y", val)
+		val, err = tx.Get("y")
+		assert.Nil(t, err)
+		assert.Equal(t, "x", val)
+		return nil
+	})
+
+	BuntdbExecutor("rwbegin")
+	BuntdbExecutor("del x")
+	BuntdbExecutor("del y")
+	BuntdbExecutor("set x xxx")
+	BuntdbExecutor("rollback")
+	bd.View(func(tx *buntdb.Tx) error {
+		val, err := tx.Get("x")
+		assert.Nil(t, err)
+		assert.Equal(t, "y", val)
+		val, err = tx.Get("y")
+		assert.Nil(t, err)
+		assert.Equal(t, "x", val)
+		return nil
+	})
+
+	BuntdbExecutor("rbegin")
+	BuntdbExecutor("del x")
+	BuntdbExecutor("del y")
+	BuntdbExecutor("commit")
+	BuntdbExecutor("rollback")
+	bd.View(func(tx *buntdb.Tx) error {
+		val, err := tx.Get("x")
+		assert.Nil(t, err)
+		assert.Equal(t, "y", val)
+		val, err = tx.Get("y")
+		assert.Nil(t, err)
+		assert.Equal(t, "x", val)
+		return nil
+	})
+
 	BuntdbExecutor("use testcli-2")
 	BuntdbExecutor("use -c testcli-2")
 	assert.Equal(t, db.GetDbPath(), "testcli-2")