first commit

transaction.go

@@ -0,0 +1,96 @@
+package db
+
+import (
+	"database/sql"
+)
+
+type Transaction struct {
+	Tx             *sql.Tx
+	Rx             *sql.DB
+	Config         *DBConfig
+	ConnectionName string
+}
+type TxClosure func(tx *Transaction) (interface{}, error)
+
+func (t *Transaction) Table(tableName string) *Builder {
+	builder := NewTxBuilder(t)
+	if t.Config.Driver == DriverMysql {
+		builder.Grammar = &MysqlGrammar{}
+	} else if t.Config.Driver == DriverSqlite3 {
+		builder.Grammar = &Sqlite3Grammar{}
+	} else {
+		panic("不支持的数据库类型")
+	}
+	builder.Grammar.SetTablePrefix(t.Config.Prefix)
+	builder.Tx = t
+	builder.Grammar.SetBuilder(builder)
+	builder.From(tableName)
+	return builder
+}
+
+func (t *Transaction) Select(query string, bindings []interface{}, dest interface{}) (result sql.Result, err error) {
+	var stmt *sql.Stmt
+	var rows *sql.Rows
+	stmt, err = t.Rx.Prepare(query)
+	if err != nil {
+		return
+	}
+	defer stmt.Close()
+	rows, err = stmt.Query(bindings...)
+	if err != nil {
+		return
+	}
+	defer rows.Close()
+
+	return ScanResult(rows, dest), nil
+}
+func (t *Transaction) AffectingStatement(query string, bindings []interface{}) (result sql.Result, err error) {
+	stmt, errP := t.Tx.Prepare(query)
+	if errP != nil {
+		err = errP
+		return
+	}
+	defer stmt.Close()
+	result, err = stmt.Exec(bindings...)
+	if err != nil {
+		return
+	}
+	return
+}
+func (t *Transaction) Insert(query string, bindings []interface{}) (sql.Result, error) {
+	return t.AffectingStatement(query, bindings)
+}
+
+func (t *Transaction) Update(query string, bindings []interface{}) (sql.Result, error) {
+	return t.AffectingStatement(query, bindings)
+}
+
+func (t *Transaction) Delete(query string, bindings []interface{}) (sql.Result, error) {
+	return t.AffectingStatement(query, bindings)
+}
+
+func (t *Transaction) Statement(query string, bindings []interface{}) (sql.Result, error) {
+	return t.AffectingStatement(query, bindings)
+}
+
+func (t *Transaction) Commit() error {
+	return t.Tx.Commit()
+}
+
+func (t *Transaction) Rollback() error {
+	return t.Tx.Rollback()
+}
+
+func (t *Transaction) Query() *Builder {
+	builder := NewTxBuilder(t)
+	if t.Config.Driver == DriverMysql {
+		builder.Grammar = &MysqlGrammar{}
+	} else if t.Config.Driver == DriverSqlite3 {
+		builder.Grammar = &Sqlite3Grammar{}
+	} else {
+		panic("不支持的数据库类型")
+	}
+	builder.Grammar.SetBuilder(builder)
+	builder.Grammar.SetTablePrefix(t.Config.Prefix)
+	return builder
+}