41 lines
786 B
Go
41 lines
786 B
Go
package boil
|
|
|
|
import "database/sql"
|
|
|
|
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
|
|
|
|
func Begin() (Transactor, error) {
|
|
creator, ok := currentDB.(Creator)
|
|
if !ok {
|
|
panic("Your database handle 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
|
|
}
|