server: fix Setting wait_timeout to 0 doesn't take effect #23351 (#23443

)
pingcap · Apr 1, 2021 · 2c83f2d · 2c83f2d
1 parent 2e30ac3
commit 2c83f2d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/server/packetio.go b/server/packetio.go
@@ -108,6 +108,11 @@ func (p *packetIO) readOnePacket() ([]byte, error) {
 }
 
 func (p *packetIO) readPacket() ([]byte, error) {
+	if p.readTimeout == 0 {
+		if err := p.bufReadConn.SetReadDeadline(time.Time{}); err != nil {
+			return nil, errors.Trace(err)
+		}
+	}
 	data, err := p.readOnePacket()
 	if err != nil {
 		return nil, errors.Trace(err)