Handle db close errors (#1370)

namespacelabs · Jan 10, 2025 · 74989fe · 74989fe
1 parent a77e0c5
commit 74989fe
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/library/oss/postgres/prepare/database/main.go b/library/oss/postgres/prepare/database/main.go
@@ -75,7 +75,11 @@ func main() {
 		if err != nil {
 			log.Fatalf("unable to open connection: %v", err)
 		}
-		defer db.Close()
+		defer func() {
+			if err := db.Close(); err != nil {
+				log.Printf("unable to close database connection: %v", err)
+			}
+		}()
 
 		for _, schema := range p.Intent.Schema {
 			if _, err := universepg.ReturnFromReadWriteTx(ctx, db, backOff{
@@ -99,7 +103,11 @@ func ensureDatabase(ctx context.Context, cluster *postgresclass.ClusterInstance,
 	if err != nil {
 		return false, err
 	}
-	defer postgresConn.Close(ctx)
+	defer func() {
+		if err := postgresConn.Close(ctx); err != nil {
+			log.Printf("unable to close database connection: %v", err)
+		}
+	}()
 
 	exists, err := existsDatabase(ctx, postgresConn, name)
 	if err != nil {