diff --git a/server.go b/server.go index 8d83aa5..8f84250 100644 --- a/server.go +++ b/server.go @@ -26,7 +26,7 @@ func NewServer(c *Config) *Server { resolver := &net.Resolver{ PreferGo: true, Dial: func(ctx context.Context, network, addr string) (net.Conn, error) { - d := net.Dialer{Timeout: time.Second * 10} + d := net.Dialer{Timeout: c.Timeout} return d.DialContext(ctx, network, c.DNS) }, } @@ -74,7 +74,7 @@ func (s *Server) Connect(w http.ResponseWriter, r *http.Request) { return } - dest, err := net.DialTimeout("tcp", host, 30*time.Second) + dest, err := net.DialTimeout("tcp", host, s.Timeout) if err != nil { s.Error(r, w, err) return