-
Notifications
You must be signed in to change notification settings - Fork 2
/
types.go
79 lines (67 loc) · 1.67 KB
/
types.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
package qbdb
import (
"database/sql"
"git.ultraware.nl/NiseVoid/qb"
)
// Target is a target for a query, either a plain DB or a Tx
type Target interface {
Render(qb.Query) (string, []interface{})
Query(qb.SelectQuery) (*sql.Rows, error)
QueryRow(qb.SelectQuery) *sql.Row
Exec(q qb.Query) (sql.Result, error)
RawExec(s string, v ...interface{}) (sql.Result, error)
Prepare(q qb.Query) (*Stmt, error)
}
// Tx is a transaction
type Tx struct {
QueryTarget
tx *sql.Tx
}
// Commit applies all the changes from the transaction
func (t Tx) Commit() error {
return t.tx.Commit()
}
// MustCommit is the same as Commit, but it panics if an error occurred
func (t Tx) MustCommit() {
err := t.Commit()
if err != nil {
panic(err)
}
}
// Rollback reverts all the changes from the transaction
func (t Tx) Rollback() error {
return t.tx.Rollback()
}
// QueryTarget is a sql.DB or sql.Tx
type QueryTarget struct {
src interface {
Exec(string, ...interface{}) (sql.Result, error)
Query(string, ...interface{}) (*sql.Rows, error)
QueryRow(string, ...interface{}) *sql.Row
Prepare(string) (*sql.Stmt, error)
}
Driver qb.Driver
Debug bool
}
// DB wraps sql.DB to support qb queries
type DB struct {
QueryTarget
DB *sql.DB
}
// Begin starts a transaction
func (db *DB) Begin() (Tx, error) {
tx, err := db.DB.Begin()
return Tx{QueryTarget{tx, db.QueryTarget.Driver, db.QueryTarget.Debug}, tx}, err
}
// MustBegin is the same as Begin, but it panics if an error occurred
func (db *DB) MustBegin() Tx {
tx, err := db.Begin()
if err != nil {
panic(err)
}
return tx
}
// New returns a new DB
func New(driver qb.Driver, db *sql.DB) *DB {
return &DB{QueryTarget{db, driver, false}, db}
}