Add error handling for Setenv

m-mizutani · Aug 27, 2023 · 9f358ab · 9f358ab
1 parent f4603b5
commit 9f358ab
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pkg/controller/cli/migrate/migrate.go b/pkg/controller/cli/migrate/migrate.go
@@ -57,7 +57,9 @@ func New() *cli.Command {
 				Host:     dbConfig.Host,
 				Port:     dbConfig.Port,
 			}
-			os.Setenv("PGSSLMODE", dbConfig.SSLMode)
+			if err := os.Setenv("PGSSLMODE", dbConfig.SSLMode); err != nil {
+				return goerr.Wrap(err, "failed to set PGSSLMODE")
+			}
 
 			db, err := postgres.NewDatabase(config)
 			if err != nil {