add NewDocumentWithTimeout

type.go

@@ -2,13 +2,13 @@ package goquery
 
 import (
 	"errors"
+	"github.com/andybalholm/cascadia"
+	"golang.org/x/net/html"
 	"io"
+	"net"
 	"net/http"
 	"net/url"
-
-	"github.com/andybalholm/cascadia"
-
-	"golang.org/x/net/html"
+	"time"
 )
 
 // Document represents an HTML document to be manipulated. Unlike jQuery, which
@@ -40,6 +40,34 @@ func NewDocument(url string) (*Document, error) {
 	return NewDocumentFromResponse(res)
 }
 
+func NewDocumentWithTimeout(url string, timeout time.Duration) (*Document, error) {
+	// Load the URL
+	var netTransport = &http.Transport{
+		Dial: (&net.Dialer{
+			Timeout: timeout,
+		}).Dial,
+		TLSHandshakeTimeout: 60 * time.Second,
+	}
+	client := &http.Client{
+		Timeout:   timeout,
+		Transport: netTransport,
+		CheckRedirect: func(req *http.Request, via []*http.Request) error {
+			return http.ErrUseLastResponse
+		},
+	}
+	req, e := http.NewRequest(http.MethodGet, url, nil)
+	if e != nil {
+		return nil, e
+	}
+
+	res, e := client.Do(req)
+	//res, e := http.Get(url)
+	if e != nil {
+		return nil, e
+	}
+	return NewDocumentFromResponse(res)
+}
+
 // NewDocumentFromReader returns a Document from a generic reader.
 // It returns an error as second value if the reader's data cannot be parsed
 // as html. It does *not* check if the reader is also an io.Closer, so the