hangar-tech
/
listmonk
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add a DB setup check on boot

This commit is contained in:
Kailash Nadh 2020-11-10 22:21:25 +05:30
parent 7a9d11d426
commit cde0b4b42a
2 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
cmd/install.go
View File

@ -180,3 +180,14 @@ func newConfigFile() error {
return ioutil.WriteFile("config.toml", b, 0644) return ioutil.WriteFile("config.toml", b, 0644)
} }
// checkSchema checks if the DB schema is installed.
func checkSchema(db *sqlx.DB) (bool, error) {
if _, err := db.Exec(`SELECT id FROM templates LIMIT 1`); err != nil {
if isTableNotExistErr(err) {
return false, nil
}
return false, err
}
return true, nil
}

8
cmd/main.go
View File

@ -113,6 +113,14 @@ func init() {
install(migList[len(migList)-1].version, db, fs, !ko.Bool("yes")) install(migList[len(migList)-1].version, db, fs, !ko.Bool("yes"))
os.Exit(0) os.Exit(0)
} }
// Check if the DB schema is installed.
if ok, err := checkSchema(db); err != nil {
log.Fatalf("error checking schema in DB: %v", err)
} else if !ok {
lo.Fatal("the database does not appear to be setup. Run --install.")
}
if ko.Bool("upgrade") { if ko.Bool("upgrade") {
upgrade(db, fs, !ko.Bool("yes")) upgrade(db, fs, !ko.Bool("yes"))
os.Exit(0) os.Exit(0)