arno/clair
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

database: Look for migrations relative to executable

Browse Source
This commit is contained in:
Andrew Lewis 2016-03-03 15:02:02 +02:00
parent 5a716f93ad
commit 35297b2461
1 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
database/pgsql/pgsql.go
View File

@ -21,7 +21,7 @@ import (
"io/ioutil"
"os"
"path"
"runtime"
"path/filepath"
"strings"
"time"
@ -123,8 +123,11 @@ func Open(config *config.DatabaseConfig) (database.Datastore, error) {
func migrate(dataSource string) error {
log.Info("running database migrations")
_, filename, _, _ := runtime.Caller(1)
migrationDir := path.Join(path.Dir(filename), "/migrations/")
exeDir, err := filepath.Abs(filepath.Dir(os.Args[0]))
if err != nil {
return err
}
migrationDir := path.Join(exeDir, "/migrations/")
conf := &goose.DBConf{
MigrationsDir: migrationDir,
Driver: goose.DBDriver{
@ -234,8 +237,13 @@ func OpenForTest(name string, withTestData bool) (*pgSQLTest, error) {
// Load test data if specified.
if withTestData {
_, filename, _, _ := runtime.Caller(0)
d, _ := ioutil.ReadFile(path.Join(path.Dir(filename)) + "/testdata/data.sql")
exeDir, err := filepath.Abs(filepath.Dir(os.Args[0]))
if err != nil {
dropDatabase(dataSourceDefaultDatabase, dbName)
log.Error(err)
return nil, err
}
d, _ := ioutil.ReadFile(path.Join(exeDir, "testdata/data.sql"))
_, err = db.(*pgSQL).Exec(string(d))
if err != nil {
dropDatabase(dataSourceDefaultDatabase, dbName)