Restore simple protocol support

2019-05-20 20:36:03 -05:00
parent 6d23b58b01
commit 29f02807b0
7 changed files with 861 additions and 3 deletions
@@ -86,7 +86,8 @@ type connRows struct {
 	args      []interface{}
 	closed    bool

-	resultReader *pgconn.ResultReader
+	resultReader      *pgconn.ResultReader
+	multiResultReader *pgconn.MultiResultReader
 }

 func (rows *connRows) FieldDescriptions() []pgproto3.FieldDescription {
@@ -107,6 +108,13 @@ func (rows *connRows) Close() {
 		}
 	}

+	if rows.multiResultReader != nil {
+		closeErr := rows.multiResultReader.Close()
+		if rows.err == nil {
+			rows.err = closeErr
+		}
+	}
+
 	if rows.logger != nil {
 		if rows.err == nil {
 			if rows.logger.shouldLog(LogLevelInfo) {