87 lines
1.7 KiB
Go
87 lines
1.7 KiB
Go
package sql
|
|
|
|
import (
|
|
"database/sql"
|
|
|
|
"git.sr.ht/~cco/go-scopes/logging/log"
|
|
)
|
|
|
|
type Cfg struct {
|
|
Driver, Connstr string
|
|
}
|
|
|
|
type Rows = sql.Rows
|
|
|
|
type rowsProc = func(*sql.Rows) error
|
|
type Scanner[T any] interface{ Scan(*Rows) (T, error) }
|
|
|
|
type Storage struct {
|
|
*sql.DB
|
|
}
|
|
|
|
func Open(cfg *Cfg) *Storage {
|
|
db, err := sql.Open(cfg.Driver, cfg.Connstr)
|
|
if err != nil {
|
|
log.Error(err).Msg("sql.Open")
|
|
return nil
|
|
}
|
|
return &Storage{db}
|
|
}
|
|
|
|
func QueryData[T Scanner[T]](db *Storage, q string, args ...interface{}) []T {
|
|
var data []T
|
|
proc := func(r *sql.Rows) error {
|
|
var rec T
|
|
rec, err := rec.Scan(r)
|
|
data = append(data, rec)
|
|
return err
|
|
}
|
|
db.Query(proc, q, args...) // ?? check error
|
|
return data
|
|
}
|
|
|
|
func QueryCol[T any](db *Storage, q string, args ...interface{}) []T {
|
|
var data []T
|
|
proc := func(r *sql.Rows) error {
|
|
var rec T
|
|
err := r.Scan(&rec)
|
|
data = append(data, rec)
|
|
return err
|
|
}
|
|
db.Query(proc, q, args...) // ?? check error
|
|
return data
|
|
}
|
|
|
|
func (db *Storage) Query(process rowsProc, q string, args ...interface{}) error {
|
|
info := "sql.Storage.Query"
|
|
//log.Debug().Str("query", q).Msg(info)
|
|
logErr := func(err error) { log.Error(err).Str("query", q).Msg(info) }
|
|
rows, err := db.DB.Query(q, args...)
|
|
if err != nil {
|
|
logErr(err)
|
|
return err
|
|
}
|
|
for rows.Next() {
|
|
if err := process(rows); err != nil {
|
|
logErr(err)
|
|
return err
|
|
}
|
|
}
|
|
if err := rows.Err(); err != nil {
|
|
logErr(err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (db *Storage) Exec(q string, args ...interface{}) (int64, error) {
|
|
info := "sql.Storage.Exec"
|
|
//log.Debug().Str("query", q).Msg(info)
|
|
res, err := db.DB.Exec(q, args...)
|
|
if err != nil {
|
|
log.Error(err).Str("query", q).Msg(info)
|
|
return 0, err
|
|
}
|
|
nrows, _ := res.RowsAffected()
|
|
return nrows, nil
|
|
}
|