-
Notifications
You must be signed in to change notification settings - Fork 0
/
main2.go
46 lines (40 loc) · 1.25 KB
/
main2.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
package main
// 练习 1.8: 修改fetch这个范例,如果输入的url参数没有 http:// 前缀的话,
// 为这个url加上该前缀。你可能会用到strings.HasPrefix这个函数。
// EX: 顺手支持了https的情况
////////////////////////////////////////////////////
// <===================OUTPUT===================> //
// go build -o fetch2 main2.go //
// <====================TEST====================> //
// go build -o fetch2 main2.go //
////////////////////////////////////////////////////
import (
"fmt"
"io"
"net/http"
"os"
"strings"
)
const (
httpPrefix = "http://"
httpsPrefix = "https://"
)
func main() {
for _, url := range os.Args[1:] {
if !strings.HasPrefix(url, httpPrefix) && !strings.HasPrefix(url, httpsPrefix) {
url = httpPrefix + url
}
fmt.Printf("Ready to fetch: %s\n", url)
resp, err := http.Get(url)
if err != nil {
fmt.Fprintf(os.Stderr, "fetch: %v\n", err)
os.Exit(1)
}
//练习 1.9: 修改fetch打印出HTTP协议的状态码,可以从resp.Status变量得到该状态码。
fmt.Printf("HTTP STATUS: %d\n", resp.StatusCode)
w, err := io.Copy(os.Stdout, resp.Body)
if err != nil {
fmt.Fprintf(os.Stderr, "fetch: copy in %d: %v", w, err)
}
}
}