74 lines
1.4 KiB
Go
74 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"path"
|
|
|
|
"modernc.org/ql"
|
|
)
|
|
|
|
func openDB() *sql.DB {
|
|
ql.RegisterDriver()
|
|
pigeonPath := maybeSetupPigeonDir()
|
|
dbPath := path.Join(pigeonPath, "db")
|
|
db, err0 := sql.Open("ql", dbPath)
|
|
|
|
if err0 != nil {
|
|
panicf("failed to open db: %s", err0)
|
|
}
|
|
|
|
err1 := db.Ping()
|
|
|
|
if err1 != nil {
|
|
panicf("failed to ping db: %s", err1)
|
|
}
|
|
|
|
migrateUp(db)
|
|
|
|
return db
|
|
}
|
|
|
|
// Database is a database object. Currently using modernc.org/ql
|
|
var database *sql.DB
|
|
|
|
func getDB() *sql.DB {
|
|
if database != nil {
|
|
return database
|
|
}
|
|
|
|
database = openDB()
|
|
return database
|
|
}
|
|
|
|
// SetConfig will write a key/value pair to the `configs`
|
|
// table
|
|
func SetConfig(key string, value []byte) {
|
|
tx, err := getDB().Begin()
|
|
if err != nil {
|
|
panicf("Failed to SetConfig (0): %s", err)
|
|
}
|
|
_, err2 := tx.Exec("INSERT INTO configs(key, value) VALUES(?1, ?2)", key, string(value))
|
|
if err2 != nil {
|
|
panicf("Failed to SetConfig (1): %s", err2)
|
|
}
|
|
err1 := tx.Commit()
|
|
if err1 != nil {
|
|
panicf("Failed to SetConfig (2): %s", err)
|
|
}
|
|
}
|
|
|
|
// GetConfig retrieves a key/value pair from the database.
|
|
func GetConfig(key string) []byte {
|
|
var result string
|
|
row := getDB().QueryRow("SELECT value FROM configs WHERE key=$1", key)
|
|
err := row.Scan(&result)
|
|
if err != nil {
|
|
if err == sql.ErrNoRows {
|
|
panicf("CONFIG MISSING: %s", key)
|
|
} else {
|
|
panic(err)
|
|
}
|
|
}
|
|
return []byte(result)
|
|
}
|