From e5c155fafc3dd255309916f1c6ab8184a15e2217 Mon Sep 17 00:00:00 2001 From: Dimitri Sokolyuk Date: Wed, 20 Jul 2016 21:23:53 +0200 Subject: Rename Get to Send, add timeout --- cmd/btcheck/main.go | 2 +- tracker/messages.go | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/cmd/btcheck/main.go b/cmd/btcheck/main.go index d4c30b0..21cc317 100644 --- a/cmd/btcheck/main.go +++ b/cmd/btcheck/main.go @@ -35,7 +35,7 @@ func main() { Compact: true, Port: 6881, } - resp, err := req.Get(tor.Announce) + resp, err := req.Send(tor.Announce) if err != nil { log.Fatal(err) } diff --git a/tracker/messages.go b/tracker/messages.go index 8a09674..96cc069 100644 --- a/tracker/messages.go +++ b/tracker/messages.go @@ -51,7 +51,7 @@ type Response struct { WarningMessage string `bencode:"warning message"` } -func (r Request) Get(announce string) (Response, error) { +func (r Request) Send(announce string) (Response, error) { fail := func(err error) (Response, error) { return Response{}, err } @@ -59,7 +59,8 @@ func (r Request) Get(announce string) (Response, error) { if err != nil { return fail(err) } - resp, err := http.Get(announce + q) + client := &http.Client{Timeout: time.Minute} + resp, err := client.Get(announce + q) if err != nil { return fail(err) } -- cgit v1.2.3