gosora/query_gen/transaction.go

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    Adapter
	textToStmt map[string]*transactionStmt
}

func (b *TransactionBuilder) SimpleDelete(table string, where string) (stmt *sql.Stmt, err error) {
	res, err := b.adapter.SimpleDelete("", table, where)
	if err != nil {
		return stmt, err
	}
	return b.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 (b *TransactionBuilder) QuickDelete(table string, where string) *transactionStmt {
	res, err := b.adapter.SimpleDelete("", table, where)
	if err != nil {
		return newTransactionStmt(nil, err)
	}

	stmt, ok := b.textToStmt[res]
	if ok {
		return stmt
	}
	stmt = newTransactionStmt(b.tx.Prepare(res))
	b.textToStmt[res] = stmt
	return stmt
}

func (b *TransactionBuilder) SimpleInsert(table string, columns string, fields string) (stmt *sql.Stmt, err error) {
	res, err := b.adapter.SimpleInsert("", table, columns, fields)
	if err != nil {
		return stmt, err
	}
	return b.tx.Prepare(res)
}

func (b *TransactionBuilder) QuickInsert(table string, where string) *transactionStmt {
	res, err := b.adapter.SimpleDelete("", table, where)
	if err != nil {
		return newTransactionStmt(nil, err)
	}

	stmt, ok := b.textToStmt[res]
	if ok {
		return stmt
	}
	stmt = newTransactionStmt(b.tx.Prepare(res))
	b.textToStmt[res] = stmt
	return stmt
}