fix(pkg/pgsql/server): close row readers to release resources

Signed-off-by: Stefano Scafiti <stefano.scafiti96@gmail.com>
codenotary · Aug 7, 2024 · 8b5f651 · 8b5f651
1 parent 76cd08a
commit 8b5f651
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/pgsql/server/query_machine.go b/pkg/pgsql/server/query_machine.go
@@ -312,6 +312,7 @@ func (s *session) query(st *sql.SelectStmt, parameters []*schema.NamedParam, res
 	if err != nil {
 		return err
 	}
+	defer reader.Close()
 
 	cols, err := reader.Columns(s.ctx)
 	if err != nil {
@@ -372,10 +373,13 @@ func (s *session) inferParamAndResultCols(stmt sql.SQLStmt) ([]sql.ColDescriptor
 		if err != nil {
 			return nil, nil, err
 		}
+
 		resCols, err = rr.Columns(s.ctx)
 		if err != nil {
 			return nil, nil, err
 		}
+
+		rr.Close()
 	}
 
 	r, err := s.db.InferParametersPrepared(s.ctx, s.tx, stmt)