Add method DBC.TransactionContext

db.go

@@ -1,6 +1,7 @@
 package db
 
 import (
+	"context"
 	"database/sql"
 	"reflect"
 	"time"
@@ -108,6 +109,29 @@ func (dbc *DBC) Transaction(txFunc func(dbc *DBC) error) (err error) {
 	return err
 }
 
+func (dbc *DBC) TransactionContext(
+	ctx context.Context,
+	txFunc func(ctx context.Context, dbc *DBC) error,
+) (err error) {
+
+	err = dbc.Begin()
+	if err != nil {
+		return
+	}
+	defer func() {
+		if p := recover(); p != nil {
+			dbc.Rollback()
+			panic(p) // re-throw panic after Rollback
+		} else if err != nil {
+			dbc.Rollback() // err is non-nil; don't change it
+		} else {
+			err = dbc.Commit() // err is nil; if Commit returns error update err
+		}
+	}()
+	err = txFunc(ctx, dbc)
+	return err
+}
+
 func (dbc *DBC) Begin() error {
 	//check if we are already in a transaction
 	if dbc.trx != nil {
@@ -247,8 +271,8 @@ func (dbc *DBC) SelectBySQL(sql string, args ...interface{}) *dbr.SelectBuilder
 	}
 }
 
-//Note: Creates a new slice of dbr.SelectBuilder for the given SQL string
-//Supported chunking only for first IN-list
+// Note: Creates a new slice of dbr.SelectBuilder for the given SQL string
+// Supported chunking only for first IN-list
 func (dbc *DBC) SelectBySQLWithChunkedIN(sql string, chunkSize int, args ...interface{}) []*dbr.SelectBuilder {
 	var builders []*dbr.SelectBuilder
 	listsForIN := make(map[int]reflect.Value)