dnote/pkg/server/database/migrate/main.go
2019-07-22 13:41:09 +10:00

66 lines
1.6 KiB
Go

/* Copyright (C) 2019 Monomax Software Pty Ltd
*
* This file is part of Dnote.
*
* Dnote is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Dnote is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with Dnote. If not, see <https://www.gnu.org/licenses/>.
*/
package main
import (
"flag"
"fmt"
"os"
"github.com/dnote/dnote/pkg/server/database"
"github.com/joho/godotenv"
"github.com/pkg/errors"
"github.com/rubenv/sql-migrate"
)
var (
migrationDir = flag.String("migrationDir", "../migrations", "the path to the directory with migraiton files")
)
func init() {
fmt.Println("Migrating Dnote database...")
// Load env
if os.Getenv("GO_ENV") != "PRODUCTION" {
if err := godotenv.Load("../../api/.env.dev"); err != nil {
panic(err)
}
}
database.InitDB()
}
func main() {
flag.Parse()
db := database.DBConn
migrations := &migrate.FileMigrationSource{
Dir: *migrationDir,
}
migrate.SetTable("migrations")
n, err := migrate.Exec(db.DB(), "postgres", migrations, migrate.Up)
if err != nil {
panic(errors.Wrap(err, "executing migrations"))
}
fmt.Printf("Applied %d migrations\n", n)
}