Run goose migration on application startup

1 files changed, 22 insertions, 0 deletions

diff --git a/models.go b/models.go
index d124387..f6dc884 100644
--- a/models.go
+++ b/models.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"bitbucket.org/liamstask/goose/lib/goose"
 	"database/sql"
 	"fmt"
 	"github.com/jmoiron/sqlx"
@@ -133,6 +134,27 @@ func init() {
 		}
 		stmt.Close()
 	}
+
+	migrateConf := &goose.DBConf{
+		MigrationsDir: config.MigrationsPath,
+		Env:           "production",
+		Driver: goose.DBDriver{
+			Name:    "sqlite3",
+			OpenStr: config.DatabaseFile,
+			Import:  "github.com/mattn/go-sqlite3",
+			Dialect: &goose.Sqlite3Dialect{},
+		},
+	}
+
+	latest, err := goose.GetMostRecentDBVersion(migrateConf.MigrationsDir)
+	if err != nil {
+		logger.Panicln(err)
+	}
+
+	err = goose.RunMigrationsOnDb(migrateConf, migrateConf.MigrationsDir, latest, db.DB)
+	if err != nil {
+		logger.Panicln(err)
+	}
 }
 
 type VoteType uint8