Release database connections instead of closing them

goatcorp · May 7, 2022 · 639f24e · 639f24e
1 parent 6a4f7ea
commit 639f24e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/cmd/operator/main.go b/cmd/operator/main.go
@@ -19,7 +19,7 @@ func applyMigrations(pool *pgx.ConnPool) {
 		log.Printf("Failed to acquire database connection: %v\n", err)
 		os.Exit(1)
 	}
-	defer conn.Close()
+	defer pool.Release(conn)
 
 	err = db.ApplyMigrations(conn, "./sql")
 	if err != nil {

diff --git a/pkg/jobs/report.go b/pkg/jobs/report.go
@@ -57,7 +57,7 @@ func (j *ReportJob) Execute() {
 		log.Printf("Failed to acquire database connection: %v\n", err)
 		return
 	}
-	defer readerConn.Close()
+	defer j.Pool.Release(readerConn)
 
 	rows, err := getReadersToNotify(readerConn)
 	if err != nil {
@@ -73,7 +73,7 @@ func (j *ReportJob) Execute() {
 		log.Printf("Failed to acquire database connection: %v\n", err)
 		return
 	}
-	defer reportConn.Close()
+	defer j.Pool.Release(reportConn)
 
 	for rows.Next() {
 		// Read the next row from the database