mirror of
https://github.com/dnote/dnote
synced 2026-03-14 22:45:50 +01:00
2c82e36f7a* Require explicit configuration for database connection * Test validation * Remove stutter * Only use packr for self hosting * Restart server upon change
73 lines
1.7 KiB
Go
73 lines
1.7 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("../../.env.dev"); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
c := database.Config{
|
|
Host: os.Getenv("DBHost"),
|
|
Port: os.Getenv("DBPort"),
|
|
Name: os.Getenv("DBName"),
|
|
User: os.Getenv("DBUser"),
|
|
Password: os.Getenv("DBPassword"),
|
|
}
|
|
database.Open(c)
|
|
}
|
|
|
|
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)
|
|
}
|