From 1bf83dd65faaafbc6a59ee2afa4a95a7fb2b3ba2 Mon Sep 17 00:00:00 2001
From: Ariel Mashraki <mashraki@fb.com>
Date: Mon, 25 Nov 2019 07:23:10 -0800
Subject: [PATCH] dialect/sql: add BeginTx for sql driver

Summary: Pull Request resolved: https://github.com/facebookincubator/ent/pull/193

Reviewed By: dlvhdr

Differential Revision: D18685934

fbshipit-source-id: abc4bf0b14ece3827bc20ecb864e7ca0594c2759
---
 dialect/sql/driver.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/dialect/sql/driver.go b/dialect/sql/driver.go
index 9edec5620..27daedc0c 100644
--- a/dialect/sql/driver.go
+++ b/dialect/sql/driver.go
@@ -51,7 +51,12 @@ func (d Driver) Dialect() string {
 
 // Tx starts and returns a transaction.
 func (d *Driver) Tx(ctx context.Context) (dialect.Tx, error) {
-	tx, err := d.ExecQuerier.(*sql.DB).BeginTx(ctx, &sql.TxOptions{})
+	return d.BeginTx(ctx, &sql.TxOptions{})
+}
+
+// BeginTx starts a transaction with options.
+func (d *Driver) BeginTx(ctx context.Context, opts *sql.TxOptions) (dialect.Tx, error) {
+	tx, err := d.ExecQuerier.(*sql.DB).BeginTx(ctx, opts)
 	if err != nil {
 		return nil, err
 	}