refactor(server): move listener init out of server.go

etcd.go

@@ -18,6 +18,7 @@ package main
 
 import (
 	"fmt"
+	"net"
 	"os"
 	"runtime"
 	"time"
@@ -128,20 +129,29 @@ func main() {
 	sConfig := server.ServerConfig{
 		Name:     info.Name,
 		URL:      info.EtcdURL,
-		BindAddr: info.EtcdListenHost,
 		CORS:     corsInfo,
 	}
-	s := server.New(sConfig, &tlsConfig, &info.EtcdTLS, ps, registry, store, &mb)
+	s := server.New(sConfig, ps, registry, store, &mb)
 
 	if config.Trace() {
 		s.EnableTracing()
 	}
 
+	var sListener net.Listener
+	if tlsConfig.Scheme == "https" {
+		sListener, err = server.NewTLSListener(info.EtcdListenHost, info.EtcdTLS.CertFile, info.EtcdTLS.KeyFile)
+	} else {
+		sListener, err = server.NewListener(info.EtcdListenHost)
+	}
+	if err != nil {
+		panic(err)
+	}
+
 	ps.SetServer(s)
 
 	// Run peer server in separate thread while the client server blocks.
 	go func() {
 		log.Fatal(ps.ListenAndServe(config.Snapshot, config.Peers))
 	}()
-	log.Fatal(s.ListenAndServe())
+	log.Fatal(s.Serve(sListener))
 }

server/listener.go

@@ -0,0 +1,39 @@
+package server
+
+import (
+	"crypto/tls"
+	"net"
+)
+
+func NewListener(addr string) (net.Listener, error) {
+	if addr == "" {
+		addr = ":http"
+	}
+	l, e := net.Listen("tcp", addr)
+	if e != nil {
+		return nil, e
+	}
+	return l, nil
+}
+
+func NewTLSListener(addr, keyFile, certFile string) (net.Listener, error) {
+	if addr == "" {
+		addr = ":https"
+	}
+	config := &tls.Config{}
+	config.NextProtos = []string{"http/1.1"}
+
+	var err error
+	config.Certificates = make([]tls.Certificate, 1)
+	config.Certificates[0], err = tls.LoadX509KeyPair(certFile, keyFile)
+	if err != nil {
+		return nil, err
+	}
+
+	conn, err := net.Listen("tcp", addr)
+	if err != nil {
+		return nil, err
+	}
+
+	return tls.NewListener(conn, config), nil
+}

server/server.go

@@ -1,7 +1,6 @@
 package server
 
 import (
-	"crypto/tls"
 	"encoding/json"
 	"fmt"
 	"net"
@@ -26,7 +25,6 @@ import (
 type ServerConfig struct {
 	Name string
 	URL  string
-	BindAddr string
 	CORS *corsInfo
 }
 
@@ -36,17 +34,15 @@ type Server struct {
 	Config         ServerConfig
 	peerServer     *PeerServer
 	registry       *Registry
-	listener       net.Listener
 	store          store.Store
-	tlsConf        *TLSConfig
-	tlsInfo        *TLSInfo
 	router         *mux.Router
 	corsMiddleware *corsHTTPMiddleware
 	metrics     *metrics.Bucket
+	listener net.Listener
 }
 
 // Creates a new Server.
-func New(sConfig ServerConfig, tlsConf *TLSConfig, tlsInfo *TLSInfo, peerServer *PeerServer, registry *Registry, store store.Store, mb *metrics.Bucket) *Server {
+func New(sConfig ServerConfig, peerServer *PeerServer, registry *Registry, store store.Store, mb *metrics.Bucket) *Server {
 	r := mux.NewRouter()
 	cors := &corsHTTPMiddleware{r, sConfig.CORS}
 
@@ -54,13 +50,9 @@ func New(sConfig ServerConfig, tlsConf *TLSConfig, tlsInfo *TLSInfo, peerServer
 		Config: sConfig,
 		Server: http.Server{
 			Handler: cors,
-			TLSConfig: &tlsConf.Server,
-			Addr:      sConfig.BindAddr,
 		},
 		store:          store,
 		registry:       registry,
-		tlsConf:        tlsConf,
-		tlsInfo:        tlsInfo,
 		peerServer:     peerServer,
 		router:         r,
 		corsMiddleware: cors,
@@ -198,59 +190,10 @@ func (s *Server) handleFunc(path string, f func(http.ResponseWriter, *http.Reque
 }
 
 // Start to listen and response etcd client command
-func (s *Server) ListenAndServe() error {
+func (s *Server) Serve(listener net.Listener) error {
-	log.Infof("etcd server [name %s, listen on %s, advertised url %s]", s.Config.Name, s.Server.Addr, s.Config.URL)
+	log.Infof("etcd server [name %s, listen on %s, advertised url %s]", s.Config.Name, listener.Addr(), s.Config.URL)
-
+	s.listener = listener
-	if s.tlsConf.Scheme == "http" {
+	return s.Server.Serve(listener)
-		return s.listenAndServe()
-	} else {
-		return s.listenAndServeTLS(s.tlsInfo.CertFile, s.tlsInfo.KeyFile)
-	}
-}
-
-// Overridden version of net/http added so we can manage the listener.
-func (s *Server) listenAndServe() error {
-	addr := s.Server.Addr
-	if addr == "" {
-		addr = ":http"
-	}
-	l, e := net.Listen("tcp", addr)
-	if e != nil {
-		return e
-	}
-	s.listener = l
-	return s.Server.Serve(l)
-}
-
-// Overridden version of net/http added so we can manage the listener.
-func (s *Server) listenAndServeTLS(certFile, keyFile string) error {
-	addr := s.Server.Addr
-	if addr == "" {
-		addr = ":https"
-	}
-	config := &tls.Config{}
-	if s.Server.TLSConfig != nil {
-		*config = *s.Server.TLSConfig
-	}
-	if config.NextProtos == nil {
-		config.NextProtos = []string{"http/1.1"}
-	}
-
-	var err error
-	config.Certificates = make([]tls.Certificate, 1)
-	config.Certificates[0], err = tls.LoadX509KeyPair(certFile, keyFile)
-	if err != nil {
-		return err
-	}
-
-	conn, err := net.Listen("tcp", addr)
-	if err != nil {
-		return err
-	}
-
-	tlsListener := tls.NewListener(conn, config)
-	s.listener = tlsListener
-	return s.Server.Serve(tlsListener)
 }
 
 // Stops the server.

tests/server_utils.go

@@ -43,10 +43,13 @@ func RunServer(f func(*server.Server)) {
 	sConfig := server.ServerConfig{
 		Name: testName,
 		URL: "http://"+testClientURL,
-		BindAddr: testClientURL,
 		CORS: corsInfo,
 	}
-	s := server.New(sConfig, &server.TLSConfig{Scheme: "http"}, &server.TLSInfo{}, ps, registry, store, nil)
+	s := server.New(sConfig, ps, registry, store, nil)
+	sListener, err := server.NewListener(testClientURL)
+	if err != nil {
+		panic(err)
+	}
 
 	ps.SetServer(s)
 
@@ -61,7 +64,7 @@ func RunServer(f func(*server.Server)) {
 	// Start up etcd server.
 	go func() {
 		c <- true
-		s.ListenAndServe()
+		s.Serve(sListener)
 	}()
 	<-c