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