one timeout to rule them all

master
Bel LaPointe 2022-12-23 12:41:02 -05:00
parent 0f85b99815
commit 50781b6ee5
1 changed files with 2 additions and 2 deletions

View File

@ -26,7 +26,7 @@ func NewServer(c *Config) *Server {
resolver := &net.Resolver{ resolver := &net.Resolver{
PreferGo: true, PreferGo: true,
Dial: func(ctx context.Context, network, addr string) (net.Conn, error) { 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) return d.DialContext(ctx, network, c.DNS)
}, },
} }
@ -74,7 +74,7 @@ func (s *Server) Connect(w http.ResponseWriter, r *http.Request) {
return return
} }
dest, err := net.DialTimeout("tcp", host, 30*time.Second) dest, err := net.DialTimeout("tcp", host, s.Timeout)
if err != nil { if err != nil {
s.Error(r, w, err) s.Error(r, w, err)
return return