add: db migrations & update
All checks were successful
continuous-integration/drone/push Build is passing
All checks were successful
continuous-integration/drone/push Build is passing
This commit is contained in:
@@ -3,11 +3,13 @@ package database
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"embed"
|
||||
_ "embed"
|
||||
"fmt"
|
||||
"path"
|
||||
"time"
|
||||
|
||||
"github.com/pressly/goose/v3"
|
||||
log "github.com/sirupsen/logrus"
|
||||
_ "modernc.org/sqlite"
|
||||
"reichard.io/antholume/config"
|
||||
@@ -22,6 +24,9 @@ type DBManager struct {
|
||||
//go:embed schema.sql
|
||||
var ddl string
|
||||
|
||||
//go:embed migrations/*
|
||||
var migrations embed.FS
|
||||
|
||||
func NewMgr(c *config.Config) *DBManager {
|
||||
// Create Manager
|
||||
dbm := &DBManager{
|
||||
@@ -38,16 +43,27 @@ func NewMgr(c *config.Config) *DBManager {
|
||||
var err error
|
||||
dbm.DB, err = sql.Open("sqlite", dbLocation)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
log.Fatalf("[NewMgr] Unable to open DB: %v", err)
|
||||
}
|
||||
|
||||
// Single Open Connection
|
||||
dbm.DB.SetMaxOpenConns(1)
|
||||
|
||||
// Execute DDL
|
||||
if _, err := dbm.DB.Exec(ddl, nil); err != nil {
|
||||
log.Info("Exec Error:", err)
|
||||
log.Fatalf("[NewMgr] Error executing schema: %v", err)
|
||||
}
|
||||
|
||||
// Perform Migrations
|
||||
err = dbm.performMigrations()
|
||||
if err != nil && err != goose.ErrNoMigrationFiles {
|
||||
log.Fatalf("[NewMgr] Error running DB migrations: %v", err)
|
||||
}
|
||||
|
||||
// Cache Tables
|
||||
dbm.CacheTempTables()
|
||||
} else {
|
||||
log.Fatal("Unsupported Database")
|
||||
log.Fatal("[NewMgr] Unsupported Database")
|
||||
}
|
||||
|
||||
dbm.Queries = New(dbm.DB)
|
||||
@@ -82,3 +98,13 @@ func (dbm *DBManager) CacheTempTables() error {
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (dbm *DBManager) performMigrations() error {
|
||||
// Set DB Migration
|
||||
goose.SetBaseFS(migrations)
|
||||
|
||||
// Run Migrations
|
||||
goose.SetLogger(log.StandardLogger())
|
||||
goose.SetDialect("sqlite")
|
||||
return goose.Up(dbm.DB, "migrations")
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user