fix reading in udp conn

pull/1132/head
Darien Raymond 2018-05-27 13:23:41 +02:00
parent 13f3c356ca
commit 46dbbfff8c
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169
1 changed files with 7 additions and 4 deletions

View File

@ -131,8 +131,13 @@ func (c *udpConn) ReadMultiBuffer() (buf.MultiBuffer, error) {
select { select {
case in := <-c.input: case in := <-c.input:
payload.Append(in) payload.Append(in)
case <-c.done.Wait(): default:
return nil, io.EOF select {
case in := <-c.input:
payload.Append(in)
case <-c.done.Wait():
return nil, io.EOF
}
} }
L: L:
@ -140,8 +145,6 @@ L:
select { select {
case in := <-c.input: case in := <-c.input:
payload.Append(in) payload.Append(in)
case <-c.done.Wait():
break L
default: default:
break L break L
} }