summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorDimitri Sokolyuk <demon@dim13.org>2018-07-24 17:03:12 +0200
committerDimitri Sokolyuk <demon@dim13.org>2018-07-24 17:03:12 +0200
commita42145425fec5817e50d729caa495df45318c6ed (patch)
tree4837cb6f12058da3d377e65a99945e7094329314
parent57bf770b53b0cb0eb928bde2e0328c7936e462df (diff)
set timeout on request
-rw-r--r--href.go11
1 files changed, 10 insertions, 1 deletions
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
}