From c3513fba5400801154c4c2d10f64b69e49c152a9 Mon Sep 17 00:00:00 2001
From: Azareal <azareal@users.noreply.github.com>
Date: Tue, 31 Oct 2017 00:57:57 +0000
Subject: [PATCH] We forgot to commit this before, oops.

---
 query_gen/lib/transaction.go | 72 ++++++++++++++++++++++++++++++++++++
 1 file changed, 72 insertions(+)
 create mode 100644 query_gen/lib/transaction.go

diff --git a/query_gen/lib/transaction.go b/query_gen/lib/transaction.go
new file mode 100644
index 00000000..f04d1fcb
--- /dev/null
+++ b/query_gen/lib/transaction.go
@@ -0,0 +1,72 @@
+package qgen
+
+import "database/sql"
+
+type transactionStmt struct {
+	stmt     *sql.Stmt
+	firstErr error // This'll let us chain the methods to reduce boilerplate
+}
+
+func newTransactionStmt(stmt *sql.Stmt, err error) *transactionStmt {
+	return &transactionStmt{stmt, err}
+}
+
+func (stmt *transactionStmt) Exec(args ...interface{}) (*sql.Result, error) {
+	if stmt.firstErr != nil {
+		return nil, stmt.firstErr
+	}
+	return stmt.Exec(args...)
+}
+
+type TransactionBuilder struct {
+	tx         *sql.Tx
+	adapter    DB_Adapter
+	textToStmt map[string]*transactionStmt
+}
+
+func (build *TransactionBuilder) SimpleDelete(table string, where string) (stmt *sql.Stmt, err error) {
+	res, err := build.adapter.SimpleDelete("_builder", table, where)
+	if err != nil {
+		return stmt, err
+	}
+	return build.tx.Prepare(res)
+}
+
+// Quick* versions refer to it being quick to type not the performance. For performance critical transactions, you might want to use the Simple* methods or the *Tx methods on the main builder. Alternate suggestions for names are welcome :)
+func (build *TransactionBuilder) QuickDelete(table string, where string) *transactionStmt {
+	res, err := build.adapter.SimpleDelete("_builder", table, where)
+	if err != nil {
+		return newTransactionStmt(nil, err)
+	}
+
+	stmt, ok := build.textToStmt[res]
+	if ok {
+		return stmt
+	}
+	stmt = newTransactionStmt(build.tx.Prepare(res))
+	build.textToStmt[res] = stmt
+	return stmt
+}
+
+func (build *TransactionBuilder) SimpleInsert(table string, columns string, fields string) (stmt *sql.Stmt, err error) {
+	res, err := build.adapter.SimpleInsert("_builder", table, columns, fields)
+	if err != nil {
+		return stmt, err
+	}
+	return build.tx.Prepare(res)
+}
+
+func (build *TransactionBuilder) QuickInsert(table string, where string) *transactionStmt {
+	res, err := build.adapter.SimpleDelete("_builder", table, where)
+	if err != nil {
+		return newTransactionStmt(nil, err)
+	}
+
+	stmt, ok := build.textToStmt[res]
+	if ok {
+		return stmt
+	}
+	stmt = newTransactionStmt(build.tx.Prepare(res))
+	build.textToStmt[res] = stmt
+	return stmt
+}