-
Notifications
You must be signed in to change notification settings - Fork 0
/
httpracer.go
78 lines (66 loc) · 1.38 KB
/
httpracer.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package main
import (
"fmt"
"net/http"
"net/url"
"os"
"sync"
"time"
)
type Response struct {
Error error
Location string
HTTPStatusCode int
TimeElapsed time.Duration
}
func fetcher(location string) *Response {
u, err := url.ParseRequestURI(location)
if err != nil {
return &Response{err, location, 0, time.Duration(0)}
}
startTime := time.Now()
resp, err := http.Get(u.String())
if err != nil {
return &Response{err, location, 0, time.Duration(0)}
}
defer resp.Body.Close()
endTime := time.Now()
elapsedTime := endTime.Sub(startTime)
return &Response{nil, u.String(), resp.StatusCode, elapsedTime}
}
func fetch(responses chan<- *Response, locations []string) {
var wg sync.WaitGroup
wg.Add(len(locations))
for _, location := range locations {
go func(location string) {
responses <- fetcher(location)
wg.Done()
}(location)
}
wg.Wait()
close(responses)
}
func printer(r <-chan *Response) {
for response := range r {
if response.Error != nil {
fmt.Printf("%s: %s\n",
os.Args[0],
response.Error,
)
} else {
fmt.Printf("%s\t%d\t%s\n",
response.TimeElapsed.String(),
response.HTTPStatusCode,
response.Location,
)
}
}
}
func main() {
if len(os.Args) < 2 {
fmt.Printf("%s: need at least one argument\n", os.Args[0])
}
responses := make(chan *Response)
go fetch(responses, os.Args[1:])
printer(responses)
}