add timeout

xxf098 · Feb 16, 2021 · 1b42d51 · 1b42d51
1 parent 5e1e096
commit 1b42d51
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/download/download.go b/download/download.go
@@ -84,7 +84,7 @@ func Download(link string, timeout time.Duration, resultChan chan<- int64) (int6
 func downloadInternal(ctx context.Context, url string, timeout time.Duration, resultChan chan<- int64, dial func(network, addr string) (net.Conn, error)) (int64, error) {
 	var max int64 = 0
 	httpTransport := &http.Transport{}
-	httpClient := &http.Client{Transport: httpTransport}
+	httpClient := &http.Client{Transport: httpTransport, Timeout: timeout}
 	if dial != nil {
 		httpTransport.Dial = dial
 	}