advsql/db.go

package advsql

import (
	"database/sql"

	"git.tordarus.net/Tordarus/adverr"
	_ "github.com/go-sql-driver/mysql"
)

type Database struct {
	db         *sql.DB
	closefuncs []func() error
}

func NewDatabase(conn *sql.DB) *Database {
	return &Database{
		db:         conn,
		closefuncs: make([]func() error, 0),
	}
}

func NewMysqlDatabase(host string, port uint16, user, pass, db string) (*Database, error) {
	conn, err := sql.Open("mysql", connString(host, port, user, pass, db))
	if err != nil {
		return nil, adverr.Wrap("could not connect to database", err)
	}
	return NewDatabase(conn), nil
}

func (db *Database) prepare(query string) (*sql.Stmt, error) {
	s, err := db.db.Prepare(query)
	if err != nil {
		return nil, err
	}
	db.closefuncs = append(db.closefuncs, s.Close)
	return s, nil
}

func (db *Database) Close() error {
	for _, close := range db.closefuncs {
		close()
	}
	return db.db.Close()
}