Refactoring

Signed-off-by: Vishal Rana <[email protected]>
firevenus · Mar 8, 2016 · 79f7be2 · 79f7be2
1 parent 6ed0b57
commit 79f7be2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/engine/fasthttp/server.go b/engine/fasthttp/server.go
@@ -15,8 +15,8 @@ type (
 	Server struct {
 		config  *engine.Config
 		handler engine.Handler
-		pool    *Pool
 		logger  *log.Logger
+		pool    *Pool
 	}
 
 	Pool struct {

diff --git a/engine/standard/server.go b/engine/standard/server.go
@@ -14,8 +14,8 @@ type (
 		server  *http.Server
 		config  *engine.Config
 		handler engine.Handler
-		pool    *Pool
 		logger  *log.Logger
+		pool    *Pool
 	}
 
 	Pool struct {
@@ -42,7 +42,10 @@ func NewWithTLS(addr, certfile, keyfile string) *Server {
 
 func NewWithConfig(c *engine.Config) (s *Server) {
 	s = &Server{
-		server: new(http.Server),
+		server: &http.Server{
+			Addr:    c.Address,
+			Handler: s,
+		},
 		config: c,
 		pool: &Pool{
 			request: sync.Pool{
@@ -83,8 +86,6 @@ func (s *Server) SetLogger(l *log.Logger) {
 }
 
 func (s *Server) Start() {
-	s.server.Addr = s.config.Address
-	s.server.Handler = s
 	certfile := s.config.TLSCertfile
 	keyfile := s.config.TLSKeyfile
 	if certfile != "" && keyfile != "" {