GOMODULES
/
goose
mirror of https://github.com/pressly/goose.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add GOOSE_MIGRATION_DIR env variable (#337)

pull/342/head
Michael Fridman 2022-04-27 22:58:27 -04:00 committed by GitHub
parent c05d3375c8
commit 1311a0eb13
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
cmd/goose/main.go
View File

@ -15,7 +15,7 @@ import (
var ( var (
flags = flag.NewFlagSet("goose", flag.ExitOnError) flags = flag.NewFlagSet("goose", flag.ExitOnError)
dir = flags.String("dir", ".", "directory with migration files") dir = flags.String("dir", defaultMigrationDir, "directory with migration files")
table = flags.String("table", "goose_db_version", "migrations table name") table = flags.String("table", "goose_db_version", "migrations table name")
verbose = flags.Bool("v", false, "enable verbose mode") verbose = flags.Bool("v", false, "enable verbose mode")
help = flags.Bool("h", false, "print help") help = flags.Bool("h", false, "print help")
@ -27,7 +27,6 @@ var (
sslkey = flags.String("ssl-key", "", "file path to SSL key in pem format (only support on mysql)") sslkey = flags.String("ssl-key", "", "file path to SSL key in pem format (only support on mysql)")
noVersioning = flags.Bool("no-versioning", false, "apply migration commands with no versioning, in file order, from directory pointed to") noVersioning = flags.Bool("no-versioning", false, "apply migration commands with no versioning, in file order, from directory pointed to")
) )
var ( var (
gooseVersion = "" gooseVersion = ""
) )
@ -56,6 +55,11 @@ func main() {
flags.Usage() flags.Usage()
return return
} }
// The -dir option has not been set, check whether the env variable is set
// before defaulting to ".".
if *dir == defaultMigrationDir && os.Getenv(envGooseMigrationDir) != "" {
*dir = os.Getenv(envGooseMigrationDir)
}
switch args[0] { switch args[0] {
case "init": case "init":
@ -123,8 +127,13 @@ func main() {
} }
const ( const (
envGooseDriver = "GOOSE_DRIVER" envGooseDriver = "GOOSE_DRIVER"
envGooseDBString = "GOOSE_DBSTRING" envGooseDBString = "GOOSE_DBSTRING"
envGooseMigrationDir = "GOOSE_MIGRATION_DIR"
)
const (
defaultMigrationDir = "."
) )
func mergeArgs(args []string) []string { func mergeArgs(args []string) []string {
@ -234,7 +243,7 @@ SELECT 'down SQL query';
// initDir will create a directory with an empty SQL migration file. // initDir will create a directory with an empty SQL migration file.
func gooseInit(dir string) error { func gooseInit(dir string) error {
if dir == "" || dir == "." { if dir == "" || dir == defaultMigrationDir {
dir = "migrations" dir = "migrations"
} }
_, err := os.Stat(dir) _, err := os.Stat(dir)