Tx func return error instead bool

Author: xfali

sqlrunner.go

@@ -104,9 +104,9 @@ func (this *Session) GetContext() context.Context {
 }
 
 //开启事务执行语句
-//返回true则提交，返回false回滚
+//返回nil则提交，返回error回滚
 //抛出异常错误触发回滚
-func (this *Session) Tx(txFunc func(session *Session) bool) {
+func (this *Session) Tx(txFunc func(session *Session) error) {
 	this.session.Begin()
 	defer func() {
 		if r := recover(); r != nil {
@@ -115,7 +115,7 @@ func (this *Session) Tx(txFunc func(session *Session) bool) {
 		}
 	}()
 
-	if txFunc(this) != true {
+	if txFunc(this) != nil {
 		this.session.Rollback()
 	} else {
 		this.session.Commit()

test/cmd/test_table_proxy.go

@@ -18,9 +18,9 @@ func New(proxyMrg *gobatis.SessionManager) *TestTableCallProxy {
 	return (*TestTableCallProxy)(proxyMrg.NewSession())
 }
 
-func (proxy *TestTableCallProxy) Tx(txFunc func(s *TestTableCallProxy) bool) {
+func (proxy *TestTableCallProxy) Tx(txFunc func(s *TestTableCallProxy) error) {
 	sess := (*gobatis.Session)(proxy)
-	sess.Tx(func(session *gobatis.Session) bool {
+	sess.Tx(func(session *gobatis.Session) error {
 		return txFunc(proxy)
 	})
 }

test/runner_test.go

@@ -230,16 +230,19 @@ func TestTx1(t *testing.T) {
 
 	var testList []TestTable
 
-	mgr.NewSession().Tx(func(session *gobatis.Session) bool {
+	mgr.NewSession().Tx(func(session *gobatis.Session) error {
 		ret := 0
-		session.Insert("insert_id").Param(testV).Result(&ret)
+		err := session.Insert("insert_id").Param(testV).Result(&ret)
+		if err != nil {
+			return err
+		}
 		t.Logf("ret %d\n", ret)
 		session.Select("select_id").Param().Result(&testList)
 		for _, v := range testList {
 			t.Logf("data: %v", v)
 		}
 		//commit
-		return true
+		return nil
 	})
 }
 
@@ -269,9 +272,13 @@ func TestTx2(t *testing.T) {
 
 	var testList []TestTable
 
-	mgr.NewSession().Tx(func(session *gobatis.Session) bool {
+	mgr.NewSession().Tx(func(session *gobatis.Session) error {
 		ret := 0
-		session.Insert("insert_id").Param(testV).Result(&ret)
+		err := session.Insert("insert_id").Param(testV).Result(&ret)
+		if err != nil {
+			t.Logf("error %v\n", err)
+			return err
+		}
 		t.Logf("ret %d\n", ret)
 		session.Select("select_id").Param().Result(&testList)
 		for _, v := range testList {
@@ -281,6 +288,6 @@ func TestTx2(t *testing.T) {
 		panic("ROLLBACK panic!!")
 
 		//rollback
-		return false
+		return nil
 	})
 }