Merge pull request #4710 from jimenez/4680-timeout_flag-fix

Disable timeout
This commit is contained in:
unclejack 2014-03-18 00:38:39 +02:00
commit e4ef28500c

View file

@ -5,15 +5,10 @@
*/ */
package listenbuffer package listenbuffer
import ( import "net"
"fmt"
"net"
"time"
)
// NewListenBuffer returns a listener listening on addr with the protocol. It sets the // NewListenBuffer returns a listener listening on addr with the protocol.
// timeout to wait on first connection before an error is returned func NewListenBuffer(proto, addr string, activate chan struct{}) (net.Listener, error) {
func NewListenBuffer(proto, addr string, activate chan struct{}, timeout time.Duration) (net.Listener, error) {
wrapped, err := net.Listen(proto, addr) wrapped, err := net.Listen(proto, addr)
if err != nil { if err != nil {
return nil, err return nil, err
@ -22,7 +17,6 @@ func NewListenBuffer(proto, addr string, activate chan struct{}, timeout time.Du
return &defaultListener{ return &defaultListener{
wrapped: wrapped, wrapped: wrapped,
activate: activate, activate: activate,
timeout: timeout,
}, nil }, nil
} }
@ -30,7 +24,6 @@ type defaultListener struct {
wrapped net.Listener // the real listener to wrap wrapped net.Listener // the real listener to wrap
ready bool // is the listner ready to start accpeting connections ready bool // is the listner ready to start accpeting connections
activate chan struct{} activate chan struct{}
timeout time.Duration // how long to wait before we consider this an error
} }
func (l *defaultListener) Close() error { func (l *defaultListener) Close() error {
@ -47,15 +40,7 @@ func (l *defaultListener) Accept() (net.Conn, error) {
if l.ready { if l.ready {
return l.wrapped.Accept() return l.wrapped.Accept()
} }
<-l.activate
select { l.ready = true
case <-time.After(l.timeout): return l.Accept()
// close the connection so any clients are disconnected
l.Close()
return nil, fmt.Errorf("timeout (%s) reached waiting for listener to become ready", l.timeout.String())
case <-l.activate:
l.ready = true
return l.Accept()
}
panic("unreachable")
} }