Merge pull request #24 from twharmon/tx-fix

Fix tx bug
twharmon · Sep 3, 2020 · fc0d354 · fc0d354
2 parents 5308b6d + e0f004a
commit fc0d354
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/db.go b/db.go
@@ -82,8 +82,13 @@ func (db *DB) mustBeValid(m *model) error {
 }
 
 // Begin starts a transaction.
-func (db *DB) Begin() (*sql.Tx, error) {
-	return db.db.Begin()
+func (db *DB) Begin() (*Tx, error) {
+	sqlTx, err := db.db.Begin()
+	tx := Tx{
+		tx: sqlTx,
+		db: db,
+	}
+	return &tx, err
 }
 
 // Insert insterts a row in the database.