《go语言圣经》练习答案--1.5. 获取URL
练习 1.7: 函数调用io.Copy(dst, src)会从src中读取内容,并将读到的结果写入到dst中,使用这个函数替代掉例子中的ioutil.ReadAll来拷贝响应结构体到os.Stdout,避免申请一个缓冲区(例子中的b)来存储。记得处理io.Copy返回结果中的错误。
package main
import (
"fmt"
"io"
"net/http"
"os"
)
func main() {
for _, url := range os.Args[1:] {
resp, err := http.Get(url)
if err != nil {
fmt.Fprintf(os.Stderr, "fetch: %v\n", err)
os.Exit(1)
}
_, err = io.Copy(os.Stdout, resp.Body)
resp.Body.Close()
if err != nil {
fmt.Fprintf(os.Stderr, "fetch: %v\n", err)
os.Exit(1)
}
}
}
练习 1.8: 修改fetch这个范例,如果输入的url参数没有 http://
前缀的话,为这个url加上该前缀。你可能会用到strings.HasPrefix这个函数。
package main
import (
"fmt"
"io"
"net/http"
"os"
"strings"
)
func main() {
const prefix = "http://"
for _, url := range os.Args[1:] {
if !strings.HasPrefix(url, prefix) {
url = prefix + url
}
resp, err := http.Get(url)
if err != nil {
fmt.Fprintf(os.Stderr, "fetch: %v\n", err)
os.Exit(1)
}
_, err = io.Copy(os.Stdout, resp.Body)
resp.Body.Close()
if err != nil {
fmt.Fprintf(os.Stderr, "fetch: %v\n", err)
os.Exit(1)
}
}
}
练习 1.9: 修改fetch打印出HTTP协议的状态码,可以从resp.Status变量得到该状态码。
package main
import (
"fmt"
"net/http"
"os"
"strings"
)
func main() {
const prefix = "http://"
for _, url := range os.Args[1:] {
if !strings.HasPrefix(url, prefix) {
url = prefix + url
}
resp, err := http.Get(url)
if err != nil {
fmt.Fprintf(os.Stderr, "fetch: %v\n", err)
os.Exit(1)
}
fmt.Println(resp.Status)
}
}