package main import ( "fmt" "io/ioutil" "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) } b, err := ioutil.ReadAll(resp.Body) resp.Body.Close() if err != nil { fmt.Fprintf(os.Stderr, "fetch: reading %s: %v\n", url, err) os.Exit(1) } fmt.Printf("%s", b) } }
执行命令:
go run main.go http://gopl.io
这个程序从两个package中导入了函数,net/http和io/ioutil包,http.Get函数是创建HTTP请求
的函数,如果获取过程没有出错,那么会在resp这个结构体中得到访问的请求结果。resp的
Body字段包括一个可读的服务器响应流。ioutil.ReadAll函数从response中读取到全部内容;
将其结果保存在变量b中。resp.Body.Close关闭resp的Body流,防止资源泄露,Printf函数会
将结果b写出到标准输出流中。
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) } //b, err := ioutil.ReadAll(resp.Body) io.Copy(os.Stdout, resp.Body) resp.Body.Close() if err != nil { fmt.Fprintf(os.Stderr, "fetch: reading %s: %v\n", url, err) os.Exit(1) } //fmt.Printf("%s", data) } }
函数调用io.Copy(dst, src)会从src中读取内容,并将读到的结果写入到dst中,使用
这个函数替代掉例子中的ioutil.ReadAll来拷贝响应结构体到os.Stdout,避免申请一个缓冲区
(例子中的b)来存储。记得处理io.Copy返回结果中的错误。
如果输入的url参数没有 http:// 前缀的话,为这个url加上该
前缀。用到strings.HasPrefix这个函数。
package main import ( "fmt" "io" "net/http" "os" "strings" ) func main() { for _, url := range os.Args[1:] { if strings.HasPrefix(url, "http://") == false { url = "http://" + url } resp, err := http.Get(url) if err != nil { fmt.Fprintf(os.Stderr, "fetch: %v\n", err) os.Exit(1) } //b, err := ioutil.ReadAll(resp.Body) io.Copy(os.Stdout, resp.Body) resp.Body.Close() if err != nil { fmt.Fprintf(os.Stderr, "fetch: reading %s: %v\n", url, err) os.Exit(1) } //fmt.Printf("%s", data) } }
打印出HTTP协议的状态码,可以从resp.Status变量得到该状态码。
package main import ( "fmt" "net/http" "os" "strings" ) func main() { for _, url := range os.Args[1:] { if strings.HasPrefix(url, "http://") == false { url = "http://" + url } resp, err := http.Get(url) if err != nil { fmt.Fprintf(os.Stderr, "fetch: %v\n", err) os.Exit(1) } status := resp.Status fmt.Printf("%s", status) if err != nil { fmt.Fprintf(os.Stderr, "fetch: reading %s: %v\n", url, err) os.Exit(1) } } }