67 lines
1.1 KiB
Go
67 lines
1.1 KiB
Go
|
package database
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"database/sql"
|
||
|
|
_ "embed"
|
||
|
|
"path"
|
||
|
|
|
||
|
|
sqlite "github.com/mattn/go-sqlite3"
|
||
|
|
log "github.com/sirupsen/logrus"
|
||
|
|
"reichard.io/bbank/config"
|
||
|
|
)
|
||
|
|
|
||
|
|
type DBManager struct {
|
||
|
|
DB *sql.DB
|
||
|
|
Ctx context.Context
|
||
|
|
Queries *Queries
|
||
|
|
}
|
||
|
|
|
||
|
|
//go:embed schema.sql
|
||
|
|
var ddl string
|
||
|
|
|
||
|
|
func foobar() string {
|
||
|
|
log.Info("WTF")
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewMgr(c *config.Config) *DBManager {
|
||
|
|
// Create Manager
|
||
|
|
dbm := &DBManager{
|
||
|
|
Ctx: context.Background(),
|
||
|
|
}
|
||
|
|
|
||
|
|
// Create Database
|
||
|
|
if c.DBType == "SQLite" {
|
||
|
|
|
||
|
|
sql.Register("sqlite3_custom", &sqlite.SQLiteDriver{
|
||
|
|
ConnectHook: func(conn *sqlite.SQLiteConn) error {
|
||
|
|
if err := conn.RegisterFunc("test_func", foobar, false); err != nil {
|
||
|
|
log.Info("Error Registering")
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
},
|
||
|
|
})
|
||
|
|
|
||
|
|
dbLocation := path.Join(c.ConfigPath, "bbank.db")
|
||
|
|
|
||
|
|
var err error
|
||
|
|
dbm.DB, err = sql.Open("sqlite3_custom", dbLocation)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
log.Fatal("Unsupported Database")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Create Tables
|
||
|
|
if _, err := dbm.DB.ExecContext(dbm.Ctx, ddl); err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
dbm.Queries = New(dbm.DB)
|
||
|
|
|
||
|
|
return dbm
|
||
|
|
}
|