gopenfusion/internal/db/schema.go

package db

/*
	This database has been based off of openfusion's. Databases should be completely interchangable between
	openfusion and gopenfusion.
*/

import (
	"database/sql"
	_ "embed"
	"fmt"

	"github.com/CPunch/gopenfusion/config"
	_ "github.com/lib/pq"
)

type DBHandler struct {
	db *sql.DB
}

//go:embed migrations/new.sql
var createDBQuery string

func OpenFromConnectionString(driverName, connectionString string) (*DBHandler, error) {
	db, err := sql.Open(driverName, connectionString)
	if err != nil {
		return nil, err
	}

	return &DBHandler{db}, nil
}

func OpenPostgresDB(dbAddr string) (*DBHandler, error) {
	fmt := fmt.Sprintf("postgresql://%s:%s@%s/%s?sslmode=disable", config.GetDBUser(), config.GetDBPass(), dbAddr, config.GetDBName())
	return OpenFromConnectionString("postgres", fmt)
}

func (db *DBHandler) Query(query string, args ...any) (*sql.Rows, error) {
	return db.db.Query(query, args...)
}

func (db *DBHandler) Exec(query string, args ...any) (sql.Result, error) {
	return db.db.Exec(query, args...)
}

func (db *DBHandler) Close() error {
	return db.db.Close()
}

func (db *DBHandler) Setup() error {
	// create db tables
	_, err := db.db.Exec(createDBQuery)
	return err
}

// calls transaction, if transaction returns a non-nil error the transaction is rolled back. otherwise the transaction is committed
func (db *DBHandler) Transaction(transaction func(*sql.Tx) error) (err error) {
	tx, err := db.db.Begin()
	if err != nil {
		return
	}
	defer func() {
		if p := recover(); p != nil {
			// we panic'd ??? rollback and rethrow
			tx.Rollback()
			panic(p)
		} else if err != nil {
			tx.Rollback()
		} else {
			err = tx.Commit()
		}
	}()

	err = transaction(tx)
	return
}