Merge pull request #18241 from LK4D4/remove_listenbuffer
Remove listenbuffer
This commit is contained in:
commit
e868bb4183
5 changed files with 4 additions and 155 deletions
|
@ -1,27 +0,0 @@
|
||||||
# listenbuffer
|
|
||||||
|
|
||||||
listenbuffer uses the kernel's listening backlog functionality to queue
|
|
||||||
connections, allowing applications to start listening immediately and handle
|
|
||||||
connections later. This is signaled by closing the activation channel passed to
|
|
||||||
the constructor.
|
|
||||||
|
|
||||||
The maximum amount of queued connections depends on the configuration of your
|
|
||||||
kernel (typically called SOMAXXCON) and cannot be configured in Go with the
|
|
||||||
net package. See `src/net/sock_platform.go` in the Go tree or consult your
|
|
||||||
kernel's manual.
|
|
||||||
|
|
||||||
activator := make(chan struct{})
|
|
||||||
buffer, err := NewListenBuffer("tcp", "localhost:4000", activator)
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// will block until activator has been closed or is sent an event
|
|
||||||
client, err := buffer.Accept()
|
|
||||||
|
|
||||||
Somewhere else in your application once it's been booted:
|
|
||||||
|
|
||||||
close(activator)
|
|
||||||
|
|
||||||
`buffer.Accept()` will return the first client in the kernel listening queue, or
|
|
||||||
continue to block until a client connects or an error occurs.
|
|
|
@ -1,76 +0,0 @@
|
||||||
/*
|
|
||||||
Package listenbuffer uses the kernel's listening backlog functionality to queue
|
|
||||||
connections, allowing applications to start listening immediately and handle
|
|
||||||
connections later. This is signaled by closing the activation channel passed to
|
|
||||||
the constructor.
|
|
||||||
|
|
||||||
The maximum amount of queued connections depends on the configuration of your
|
|
||||||
kernel (typically called SOMAXXCON) and cannot be configured in Go with the
|
|
||||||
net package. See `src/net/sock_platform.go` in the Go tree or consult your
|
|
||||||
kernel's manual.
|
|
||||||
|
|
||||||
activator := make(chan struct{})
|
|
||||||
buffer, err := NewListenBuffer("tcp", "localhost:4000", activator)
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// will block until activator has been closed or is sent an event
|
|
||||||
client, err := buffer.Accept()
|
|
||||||
|
|
||||||
Somewhere else in your application once it's been booted:
|
|
||||||
|
|
||||||
close(activator)
|
|
||||||
|
|
||||||
`buffer.Accept()` will return the first client in the kernel listening queue, or
|
|
||||||
continue to block until a client connects or an error occurs.
|
|
||||||
*/
|
|
||||||
package listenbuffer
|
|
||||||
|
|
||||||
import "net"
|
|
||||||
|
|
||||||
// NewListenBuffer returns a net.Listener listening on addr with the protocol
|
|
||||||
// passed. The channel passed is used to activate the listenbuffer when the
|
|
||||||
// caller is ready to accept connections.
|
|
||||||
func NewListenBuffer(proto, addr string, activate <-chan struct{}) (net.Listener, error) {
|
|
||||||
wrapped, err := net.Listen(proto, addr)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return &defaultListener{
|
|
||||||
wrapped: wrapped,
|
|
||||||
activate: activate,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// defaultListener is the buffered wrapper around the net.Listener
|
|
||||||
type defaultListener struct {
|
|
||||||
wrapped net.Listener // The net.Listener wrapped by listenbuffer
|
|
||||||
ready bool // Whether the listenbuffer has been activated
|
|
||||||
activate <-chan struct{} // Channel to control activation of the listenbuffer
|
|
||||||
}
|
|
||||||
|
|
||||||
// Close closes the wrapped socket.
|
|
||||||
func (l *defaultListener) Close() error {
|
|
||||||
return l.wrapped.Close()
|
|
||||||
}
|
|
||||||
|
|
||||||
// Addr returns the listening address of the wrapped socket.
|
|
||||||
func (l *defaultListener) Addr() net.Addr {
|
|
||||||
return l.wrapped.Addr()
|
|
||||||
}
|
|
||||||
|
|
||||||
// Accept returns a client connection on the wrapped socket if the listen buffer
|
|
||||||
// has been activated. To active the listenbuffer the activation channel passed
|
|
||||||
// to NewListenBuffer must have been closed or sent an event.
|
|
||||||
func (l *defaultListener) Accept() (net.Conn, error) {
|
|
||||||
// if the listen has been told it is ready then we can go ahead and
|
|
||||||
// start returning connections
|
|
||||||
if l.ready {
|
|
||||||
return l.wrapped.Accept()
|
|
||||||
}
|
|
||||||
<-l.activate
|
|
||||||
l.ready = true
|
|
||||||
return l.Accept()
|
|
||||||
}
|
|
|
@ -1,41 +0,0 @@
|
||||||
package listenbuffer
|
|
||||||
|
|
||||||
import (
|
|
||||||
"io/ioutil"
|
|
||||||
"net"
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestListenBufferAllowsAcceptingWhenActivated(t *testing.T) {
|
|
||||||
lock := make(chan struct{})
|
|
||||||
buffer, err := NewListenBuffer("tcp", "", lock)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("Unable to create listen buffer: ", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
go func() {
|
|
||||||
conn, err := net.Dial("tcp", buffer.Addr().String())
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("Client failed to establish connection to server: ", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
conn.Write([]byte("ping"))
|
|
||||||
conn.Close()
|
|
||||||
}()
|
|
||||||
|
|
||||||
close(lock)
|
|
||||||
|
|
||||||
client, err := buffer.Accept()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("Failed to accept client: ", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
response, err := ioutil.ReadAll(client)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("Failed to read from client: ", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if string(response) != "ping" {
|
|
||||||
t.Fatal("Expected to receive ping from client, received: ", string(response))
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -7,17 +7,13 @@ import (
|
||||||
"net"
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/docker/docker/pkg/listenbuffer"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// NewTCPSocket creates a TCP socket listener with the specified address and
|
// NewTCPSocket creates a TCP socket listener with the specified address and
|
||||||
// and the specified tls configuration. If TLSConfig is set, will encapsulate the
|
// and the specified tls configuration. If TLSConfig is set, will encapsulate the
|
||||||
// TCP listener inside a TLS one.
|
// TCP listener inside a TLS one.
|
||||||
// The channel passed is used to activate the listenbuffer when the caller is ready
|
func NewTCPSocket(addr string, tlsConfig *tls.Config) (net.Listener, error) {
|
||||||
// to accept connections.
|
l, err := net.Listen("tcp", addr)
|
||||||
func NewTCPSocket(addr string, tlsConfig *tls.Config, activate <-chan struct{}) (net.Listener, error) {
|
|
||||||
l, err := listenbuffer.NewListenBuffer("tcp", addr, activate)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
|
@ -10,20 +10,17 @@ import (
|
||||||
"syscall"
|
"syscall"
|
||||||
|
|
||||||
"github.com/Sirupsen/logrus"
|
"github.com/Sirupsen/logrus"
|
||||||
"github.com/docker/docker/pkg/listenbuffer"
|
|
||||||
"github.com/opencontainers/runc/libcontainer/user"
|
"github.com/opencontainers/runc/libcontainer/user"
|
||||||
)
|
)
|
||||||
|
|
||||||
// NewUnixSocket creates a unix socket with the specified path and group.
|
// NewUnixSocket creates a unix socket with the specified path and group.
|
||||||
// The channel passed is used to activate the listenbuffer when the caller is ready
|
func NewUnixSocket(path, group string) (net.Listener, error) {
|
||||||
// to accept connections.
|
|
||||||
func NewUnixSocket(path, group string, activate <-chan struct{}) (net.Listener, error) {
|
|
||||||
if err := syscall.Unlink(path); err != nil && !os.IsNotExist(err) {
|
if err := syscall.Unlink(path); err != nil && !os.IsNotExist(err) {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
mask := syscall.Umask(0777)
|
mask := syscall.Umask(0777)
|
||||||
defer syscall.Umask(mask)
|
defer syscall.Umask(mask)
|
||||||
l, err := listenbuffer.NewListenBuffer("unix", path, activate)
|
l, err := net.Listen("unix", path)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue