mirror of
https://github.com/sigcn/pg.git
synced 2025-12-24 11:31:03 +08:00
connmux: avoid explicitly closing l.accept to prevent a panic (#36)
This commit is contained in:
@@ -221,11 +221,8 @@ func (l *MuxSession) Accept() (net.Conn, error) {
|
||||
select {
|
||||
case <-l.exit:
|
||||
return nil, io.ErrClosedPipe
|
||||
case c, ok := <-l.accept:
|
||||
if ok {
|
||||
return c, nil
|
||||
}
|
||||
return nil, io.ErrClosedPipe
|
||||
case c := <-l.accept:
|
||||
return c, nil
|
||||
}
|
||||
}
|
||||
|
||||
@@ -234,7 +231,6 @@ func (l *MuxSession) Accept() (net.Conn, error) {
|
||||
func (l *MuxSession) Close() error {
|
||||
l.closeOnce.Do(func() {
|
||||
close(l.exit)
|
||||
close(l.accept)
|
||||
l.closed.Store(true)
|
||||
})
|
||||
return l.c.Close()
|
||||
|
||||
Reference in New Issue
Block a user