feat: return begin err

MikyChow · Feb 13, 2023 · 7eed555 · 7eed555
1 parent c5b4756
commit 7eed555
Show file tree

Hide file tree

Showing 5 changed files with 30 additions and 10 deletions.
diff --git a/internal/template/query.go b/internal/template/query.go
@@ -84,10 +84,14 @@ func (q *Query) Transaction(fc func(tx *Query) error, opts ...*sql.TxOptions) er
 }
 
 func (q *Query) Begin(opts ...*sql.TxOptions) *QueryTx {
-	return &QueryTx{q.clone(q.db.Begin(opts...))}
+	tx := q.db.Begin(opts...)
+	return &QueryTx{Query: q.clone(tx), Error: tx.Error}
 }
 
-type QueryTx struct{ *Query }
+type QueryTx struct {
+	*Query
+	Error error
+}
 
 func (q *QueryTx) Commit() error {
 	return q.db.Commit().Error

diff --git a/tests/.expect/dal_1/query/gen.go b/tests/.expect/dal_1/query/gen.go
diff --git a/tests/.expect/dal_2/query/gen.go b/tests/.expect/dal_2/query/gen.go
diff --git a/tests/.expect/dal_3/query/gen.go b/tests/.expect/dal_3/query/gen.go
diff --git a/tests/.expect/dal_4/query/gen.go b/tests/.expect/dal_4/query/gen.go