3152eed170
* Where statement OR/AND support * Added null-extended library to use different data types * Added disable triggers function for test main
54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
package boil
|
|
|
|
import (
|
|
"database/sql"
|
|
"os"
|
|
)
|
|
|
|
type Executor interface {
|
|
Exec(query string, args ...interface{}) (sql.Result, error)
|
|
Query(query string, args ...interface{}) (*sql.Rows, error)
|
|
QueryRow(query string, args ...interface{}) *sql.Row
|
|
}
|
|
|
|
type Transactor interface {
|
|
Commit() error
|
|
Rollback() error
|
|
|
|
Executor
|
|
}
|
|
|
|
type Creator interface {
|
|
Begin() (*sql.Tx, error)
|
|
}
|
|
|
|
var currentDB Executor
|
|
|
|
// DebugMode is a flag controlling whether generated sql statements and
|
|
// debug information is outputted to the DebugWriter handle
|
|
//
|
|
// NOTE: This should be disabled in production to avoid leaking sensitive data
|
|
var DebugMode = false
|
|
|
|
// DebugWriter is where the debug output will be sent if DebugMode is true
|
|
var DebugWriter = os.Stdout
|
|
|
|
func Begin() (Transactor, error) {
|
|
creator, ok := currentDB.(Creator)
|
|
if !ok {
|
|
panic("Your database does not support transactions.")
|
|
}
|
|
|
|
return creator.Begin()
|
|
}
|
|
|
|
// SetDB initializes the database handle for all template db interactions
|
|
func SetDB(db Executor) {
|
|
currentDB = db
|
|
}
|
|
|
|
// GetDB retrieves the global state database handle
|
|
func GetDB() Executor {
|
|
return currentDB
|
|
}
|