disconnect on fail

tidepool-org · Dec 7, 2023 · 5b38ecc · 5b38ecc
1 parent e46a5af
commit 5b38ecc
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/migrations/20231128_jellyfish_migration/20231128_jellyfish_migration.go b/migrations/20231128_jellyfish_migration/20231128_jellyfish_migration.go
@@ -73,7 +73,7 @@ func (m *Migration) RunAndExit() {
 	}
 
 	m.CLI().Action = func(ctx *cli.Context) error {
-		log.Println("prepare")
+		log.Println("running")
 		var err error
 		m.client, err = mongo.Connect(m.ctx, options.Client().ApplyURI(m.config.uri))
 		if err != nil {
@@ -95,6 +95,10 @@ func (m *Migration) RunAndExit() {
 	}
 
 	if err := m.CLI().Run(os.Args); err != nil {
+		if m.client != nil {
+			m.client.Disconnect(m.ctx)
+		}
+
 		os.Exit(1)
 	}
 }