From 639f24ecbd2ee44f22b706c5627e1e71e33d0ccd Mon Sep 17 00:00:00 2001
From: karashiiro <49822414+karashiiro@users.noreply.github.com>
Date: Fri, 6 May 2022 17:50:33 -0700
Subject: [PATCH] Release database connections instead of closing them

---
 cmd/operator/main.go | 2 +-
 pkg/jobs/report.go   | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/cmd/operator/main.go b/cmd/operator/main.go
index d836aa6..c7bfa10 100644
--- 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
index 3e6b9fb..113e44e 100644
--- 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