Continue receiving and ignoring messages on SelectFunc until readyForQuery

connection.go

@@ -128,33 +128,24 @@ func (c *Connection) SelectFunc(sql string, onDataRow func(*DataRowReader) error
 		return
 	}
 
-	var callbackError error
-
 	for {
-		var t byte
+		if t, r, rxErr := c.rxMsg(); rxErr == nil {
-		var r *MessageReader
-		if t, r, err = c.rxMsg(); err == nil {
 			switch t {
 			case readyForQuery:
-				if err == nil {
-					err = callbackError
-				}
 				return
 			case rowDescription:
 				fields = c.rxRowDescription(r)
 			case dataRow:
-				if callbackError == nil {
+				if err == nil {
-					callbackError = onDataRow(newDataRowReader(r, fields))
+					err = onDataRow(newDataRowReader(r, fields))
 				}
 			case commandComplete:
 			case bindComplete:
 			default:
-				if err = c.processContextFreeMsg(t, r); err != nil {
+				err = c.processContextFreeMsg(t, r)
-					return
-				}
 			}
 		} else {
-			return
+			return rxErr
 		}
 	}
 }