Remove rxMsg
This commit is contained in:
@@ -708,22 +708,6 @@ func (c *Conn) processContextFreeMsg(msg pgproto3.BackendMessage) (err error) {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Conn) rxMsg() (pgproto3.BackendMessage, error) {
|
|
||||||
if !c.IsAlive() {
|
|
||||||
return nil, ErrDeadConn
|
|
||||||
}
|
|
||||||
|
|
||||||
msg, err := c.pgConn.ReceiveMessage()
|
|
||||||
if err != nil {
|
|
||||||
if netErr, ok := err.(net.Error); !(ok && netErr.Timeout()) {
|
|
||||||
c.die(err)
|
|
||||||
}
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return msg, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Conn) rxErrorResponse(msg *pgproto3.ErrorResponse) *pgconn.PgError {
|
func (c *Conn) rxErrorResponse(msg *pgproto3.ErrorResponse) *pgconn.PgError {
|
||||||
err := &pgconn.PgError{
|
err := &pgconn.PgError{
|
||||||
Severity: msg.Severity,
|
Severity: msg.Severity,
|
||||||
@@ -975,7 +959,7 @@ func (c *Conn) waitForPreviousCancelQuery(ctx context.Context) error {
|
|||||||
|
|
||||||
func (c *Conn) ensureConnectionReadyForQuery() error {
|
func (c *Conn) ensureConnectionReadyForQuery() error {
|
||||||
for c.pendingReadyForQueryCount > 0 {
|
for c.pendingReadyForQueryCount > 0 {
|
||||||
msg, err := c.rxMsg()
|
msg, err := c.pgConn.ReceiveMessage()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|||||||
+1
-1
@@ -79,7 +79,7 @@ func (f *fastpath) Call(oid pgtype.OID, args []fpArg) (res []byte, err error) {
|
|||||||
f.cn.pendingReadyForQueryCount++
|
f.cn.pendingReadyForQueryCount++
|
||||||
|
|
||||||
for {
|
for {
|
||||||
msg, err := f.cn.rxMsg()
|
msg, err := f.cn.pgConn.ReceiveMessage()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|||||||
+2
-2
@@ -219,7 +219,7 @@ func (rc *ReplicationConn) GetConnInfo() *pgtype.ConnInfo {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (rc *ReplicationConn) readReplicationMessage() (r *ReplicationMessage, err error) {
|
func (rc *ReplicationConn) readReplicationMessage() (r *ReplicationMessage, err error) {
|
||||||
msg, err := rc.c.rxMsg()
|
msg, err := rc.c.pgConn.ReceiveMessage()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -344,7 +344,7 @@ func (rc *ReplicationConn) sendReplicationModeQuery(sql string) (*Rows, error) {
|
|||||||
rows.fatal(err)
|
rows.fatal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
msg, err := rc.c.rxMsg()
|
msg, err := rc.c.pgConn.ReceiveMessage()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user