ReceiveMessage returns context error instead of io error on cancel

2019-08-22 20:11:27 -05:00
parent 11255efe7a
commit 1558987979
1 changed files with 3 additions and 0 deletions
@@ -361,6 +361,9 @@ func (pgConn *PgConn) ReceiveMessage(ctx context.Context) (pgproto3.BackendMessa
 	defer pgConn.contextWatcher.Unwatch()

 	msg, err := pgConn.receiveMessage()
+	if err != nil {
+		err = linkErrors(ctx.Err(), err)
+	}
 	return msg, err
 }