diff --git a/cmd/goose/main.go b/cmd/goose/main.go index b1ca141..724d7e6 100644 --- a/cmd/goose/main.go +++ b/cmd/goose/main.go @@ -61,6 +61,11 @@ func main() { if err != nil { log.Fatalf("-dbstring=%q: %v\n", dbstring, err) } + defer func() { + if err := db.Close(); err != nil { + log.Fatalf("goose: failed to close DB: %v\n", err) + } + }() arguments := []string{} if len(args) > 3 { diff --git a/examples/go-migrations/main.go b/examples/go-migrations/main.go index 77c9997..d9b3470 100644 --- a/examples/go-migrations/main.go +++ b/examples/go-migrations/main.go @@ -33,6 +33,12 @@ func main() { log.Fatalf("goose: failed to open DB: %v\n", err) } + defer func() { + if err := db.Close(); err != nil { + log.Fatalf("goose: failed to close DB: %v\n", err) + } + }() + arguments := []string{} if len(args) > 3 { arguments = append(arguments, args[3:]...)