migrate at db init

repo/db.go

@@ -8,14 +8,16 @@ import (
 
 	"github.com/jackc/pgx/v5/stdlib"
 
+	_ "github.com/lib/pq"
+
 	"github.com/nbtca/saturday/util"
 	"github.com/qustavo/sqlhooks/v2"
 	"github.com/sirupsen/logrus"
 
 	"github.com/jmoiron/sqlx"
 
 	"github.com/golang-migrate/migrate/v4"
-	_ "github.com/golang-migrate/migrate/v4/database/postgres"
+	"github.com/golang-migrate/migrate/v4/database/postgres"
 	_ "github.com/golang-migrate/migrate/v4/source/file"
 )
 
@@ -43,12 +45,22 @@ func (h *Hooks) After(ctx context.Context, query string, args ...interface{}) (c
 
 func InitDB() {
 	var err error
-	sql.Register("mysqlWithHooks", sqlhooks.Wrap(&stdlib.Driver{},&Hooks{}))
+	sql.Register("mysqlWithHooks", sqlhooks.Wrap(&stdlib.Driver{}, &Hooks{}))
 	db, err = sqlx.Connect("mysqlWithHooks", os.Getenv("DB_URL"))
+	if err != nil {
+		util.Logger.Fatal(err)
+	}
+
+	driver, err := postgres.WithInstance(db.DB, &postgres.Config{})
+	if err != nil {
+		util.Logger.Fatal(err)
+	}
+
 	m, err := migrate.NewWithDatabaseInstance(
-		"file:///migrations",
+		"file://migrations",
 		"postgres", driver)
 	m.Up() // or m.Step(2) if you want to explicitly set the number of migrations to run
+	m.Close()
 
 	if err != nil {
 		util.Logger.Fatal(err)