From a42145425fec5817e50d729caa495df45318c6ed Mon Sep 17 00:00:00 2001 From: Dimitri Sokolyuk Date: Tue, 24 Jul 2018 17:03:12 +0200 Subject: set timeout on request --- href.go | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/href.go b/href.go index 9e0139b..0869afd 100644 --- a/href.go +++ b/href.go @@ -1,10 +1,12 @@ package main import ( + "context" "errors" "io" "net/http" "strings" + "time" "unicode/utf8" "golang.org/x/net/html" @@ -47,7 +49,14 @@ func title(r io.Reader) (string, error) { } func getTitle(uri string) (string, error) { - resp, err := http.Get(uri) + req, err := http.NewRequest("GET", uri, nil) + if err != nil { + return "", err + } + ctx, cancel := context.WithTimeout(req.Context(), time.Minute) + defer cancel() + + resp, err := http.DefaultClient.Do(req.WithContext(ctx)) if err != nil { return "", err } -- cgit v1.2.3