address comment

iotexproject · Jul 15, 2022 · b29714f · b29714f
1 parent 6f4b538
commit b29714f
Show file tree

Hide file tree

Showing 5 changed files with 20 additions and 21 deletions.
diff --git a/api/http.go b/api/http.go
@@ -35,7 +35,7 @@ func NewHTTPServer(route string, port int, handler http.Handler) *HTTPServer {
 	mux := http.NewServeMux()
 	mux.Handle("/"+route, handler)
 
-	svr := httputil.Server(":"+strconv.Itoa(port), mux, httputil.HeaderTimeout(10*time.Second))
+	svr := httputil.NewServer(":"+strconv.Itoa(port), mux, httputil.HeaderTimeout(10*time.Second))
 	return &HTTPServer{
 		svr: &svr,
 	}

diff --git a/pkg/probe/probe.go b/pkg/probe/probe.go
@@ -55,7 +55,7 @@ func New(port int, opts ...Option) *Server {
 	mux.HandleFunc("/health", readiness)
 	mux.Handle("/metrics", promhttp.Handler())
 
-	s.server = httputil.Server(fmt.Sprintf(":%d", port), mux)
+	s.server = httputil.NewServer(fmt.Sprintf(":%d", port), mux)
 	return s
 }
 

diff --git a/pkg/util/httputil/httputil.go b/pkg/util/httputil/httputil.go
@@ -9,11 +9,7 @@ import (
 )
 
 const (
-	_connectionCount   = 400
-	_readHeaderTimeout = 5 * time.Second
-	_readTimeout       = 30 * time.Second
-	_writeTimeout      = 30 * time.Second
-	_idleTimeout       = 120 * time.Second
+	_connectionCount = 400
 )
 
 type (
@@ -28,21 +24,23 @@ type (
 	}
 )
 
+var DefaultServerConfig = serverConfig{
+	ReadHeaderTimeout: 5 * time.Second,
+	ReadTimeout:       30 * time.Second,
+	WriteTimeout:      30 * time.Second,
+	IdleTimeout:       120 * time.Second,
+}
+
 // HeaderTimeout sets header timeout
 func HeaderTimeout(h time.Duration) ServerOption {
 	return func(cfg *serverConfig) {
 		cfg.ReadHeaderTimeout = h
 	}
 }
 
-// Server creates a HTTP server with time out settings.
-func Server(addr string, handler http.Handler, opts ...ServerOption) http.Server {
-	cfg := serverConfig{
-		ReadHeaderTimeout: _readHeaderTimeout,
-		ReadTimeout:       _readTimeout,
-		WriteTimeout:      _writeTimeout,
-		IdleTimeout:       _idleTimeout,
-	}
+// NewServer creates a HTTP server with time out settings.
+func NewServer(addr string, handler http.Handler, opts ...ServerOption) http.Server {
+	cfg := DefaultServerConfig
 	for _, opt := range opts {
 		opt(&cfg)
 	}

diff --git a/pkg/util/httputil/httputil_test.go b/pkg/util/httputil/httputil_test.go
@@ -4,6 +4,7 @@ import (
 	"errors"
 	"net/http"
 	"testing"
+	"time"
 
 	"github.com/stretchr/testify/require"
 )
@@ -14,14 +15,14 @@ func TestServer(t *testing.T) {
 		addr := "myAddress"
 
 		expectValue := http.Server{
-			ReadHeaderTimeout: _readHeaderTimeout,
-			ReadTimeout:       _readTimeout,
-			WriteTimeout:      _writeTimeout,
-			IdleTimeout:       _idleTimeout,
+			ReadHeaderTimeout: 2 * time.Second,
+			ReadTimeout:       DefaultServerConfig.ReadTimeout,
+			WriteTimeout:      DefaultServerConfig.WriteTimeout,
+			IdleTimeout:       DefaultServerConfig.IdleTimeout,
 			Addr:              addr,
 			Handler:           handler,
 		}
-		result := Server(addr, handler)
+		result := NewServer(addr, handler, HeaderTimeout(2*time.Second))
 		require.Equal(t, expectValue, result)
 	})
 }

diff --git a/server/itx/server.go b/server/itx/server.go
@@ -243,7 +243,7 @@ func StartServer(ctx context.Context, svr *Server, probeSvr *probe.Server, cfg c
 		mux.Handle("/debug/pprof/trace", http.HandlerFunc(pprof.Trace))
 
 		port := fmt.Sprintf(":%d", cfg.System.HTTPAdminPort)
-		adminserv = httputil.Server(port, mux)
+		adminserv = httputil.NewServer(port, mux)
 		defer func() {
 			if err := adminserv.Shutdown(ctx); err != nil {
 				log.L().Error("Error when serving metrics data.", zap.Error(err))