golang学习 ----获取URL

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)
		}

	}
}

  

猜你喜欢

转载自www.cnblogs.com/saryli/p/9991226.html