Update server_socket.go

Author: qr243vbi

lib/go/thrift/server_socket.go

@@ -26,11 +26,9 @@ import (
 	"time"
 )
 
-// TServerSocketListenerFactory abstracts how listeners are created.
-type TServerSocketListenerFactory func(listen bool) (net.Addr, net.Listener, error)
-
 type TServerSocket struct {
-	factory       TServerSocketListenerFactory
+	// TServerSocketListenerFactory abstracts how listeners are created.
+	factory       func(listen bool) (net.Addr, net.Listener, error)
 	clientTimeout time.Duration
 
 	mu          sync.RWMutex
@@ -67,7 +65,20 @@ func NewTServerSocketFromAddrTimeout(addr net.Addr, clientTimeout time.Duration)
 }
 
 // Allows full customization (TLS, mocks, unix sockets, windows named pipes, etc.)
-func NewTServerSocketFromFactoryTimeout(factory TServerSocketListenerFactory, clientTimeout time.Duration) *TServerSocket {
+func NewTServerSocketFromFactoryAddrTimeout(proc func(addr net.Addr) (net.Listener, error),addr net.Addr, clientTimeout time.Duration) *TServerSocket {
+	factory := func(listen bool) (net.Addr, net.Listener, error) {
+		var listener net.Listener
+		var err error
+		if (listen){
+			listener, err = proc(addr)
+		}
+		return addr, listener, err
+	}
+	return NewTServerSocketFromFactoryTimeout(factory, clientTimeout)
+}
+
+// Allows full customization (TLS, mocks, unix sockets, windows named pipes, etc.)
+func NewTServerSocketFromFactoryTimeout(factory func(listen bool) (net.Addr, net.Listener, error), clientTimeout time.Duration) *TServerSocket {
 	return &TServerSocket{
 		factory:       factory,
 		clientTimeout: clientTimeout,