golang快速入门（四）

提示：本系列文章适合有其他语音基础并对Go有持续冲动的读者

一、golang获取HTTP请求

1.在golang标准库中提供了net包来处理网络连接，通过http.Get创建http请求并返回服务器响应流。再通过ReadAll读取response全部内容。

package main
import (

"fmt"

"io/ioutil"

"net/http"

"os"

)
func main() {

for _, arg := range os.Args[1:] {

res, err := http.Get(arg)

if err != nil {

fmt.Println(err)

os.Exit(1)

}

b, err := ioutil.ReadAll(res.Body)

res.Body.Close()

if err != nil {

fmt.Println(err)

os.Exit(1)

}

fmt.Printf("%s", b)

}

}

以访问360为例

超时会当成错误被捕获

二、练习

io.Copy(dst, src)

package main
import (

"bufio"

"fmt"

"io"

"net/http"

"os"

)
func main() {

for _, arg := range os.Args[1:] {

res, err := http.Get(arg)

if err != nil {

fmt.Println(err)

os.Exit(1)

}

out, err := os.Create("/tmp/buf_file2.txt")

// 初始化一个 io.Writer

wt := bufio.NewWriter(out)

result, err := io.Copy(wt, res.Body)

defer res.Body.Close()
if err != nil {

fmt.Println(err)

os.Exit(1)

}

fmt.Println(result)

wt.Flush()

}

}

http://strings.HasPrefix

strings.Hasprefix(s, prefix)

func HasPrefix(s, prefix string) bool {

return len(s) >= len(prefix) && s[0:len(prefix)] == prefix //通过切片处理

}

我们只需对参数做判断即可。

package main
import (

"bufio"

"fmt"

"io"

"net/http"

"os"

"strings"

)
func main() {

for _, arg := range os.Args[1:] {

if strings.HasPrefix(arg, "http://") {

fmt.Println(arg)

get_txt(arg)

} else {

arg = "http://" + arg

get_txt(arg)

fmt.Println(arg)

}

}

}
func get_txt(arg string) {

res, err := http.Get(arg)

if err != nil {

fmt.Println(err)

os.Exit(1)

}

out, err := os.Create("/tmp/buf_file2.txt")

// 初始化一个 io.Writer

wt := bufio.NewWriter(out)

result, err := io.Copy(wt, res.Body)

defer res.Body.Close()
if err != nil {

fmt.Println(err)

os.Exit(1)

}

fmt.Println(result)

wt.Flush()

}

resp.Status

我们增加一个get_code函数如下，在main将get_txt替换为get_code即可：

func get_code(arg string) {

res, err := http.Get(arg)

if err != nil {

fmt.Println(err)

os.Exit(1)

}

status := res.Status

fmt.Println("Http Code :", status)

}

结果如下：

三、初尝golang并发（扩展）

goroutime

new_routime(arg string, ch chan<- stringstart_timeend_time

package main
import (

"bufio"

"fmt"

"io"

"io/ioutil"

"net/http"

"os"

"strings"

"time"

)
func get_txt(arg string) {

// ...

}

func get_code(arg string) {

// ...

}

func main() {

start := time.Now() //记录开始时间

ch := make(chan string) //创建一个字符信道
for _, arg := range os.Args[1:] {

//根据参数开启协程

if strings.HasPrefix(arg, "http://") {

go new_routime(arg, ch)

} else {

arg = "http://" + arg

go new_routime(arg, ch)

}

}

//读取信道数据

for range os.Args[1:] {

fmt.Println(<-ch)

}

end := time.Since(start).Seconds() //结束时间

fmt.Printf("time used :%.2fs\n", end)

}

func new_routime(arg string, ch chan<- string) {

start_time := time.Now()

res, err := http.Get(arg)

if err != nil {

ch <- fmt.Sprintf("err:", err)

return

}

size_bytes, err := io.Copy(ioutil.Discard, res.Body)

res.Body.Close()

if err != nil {

ch <- fmt.Sprintf("reading usl:%v", err)

return

}

end_time := time.Since(start_time).Seconds()

ch <- fmt.Sprintf("%.2fs %10d %s", end_time, size_bytes, arg) //执行时间、请求大小、url

}

Sprintf

ioutil.Discard

执行后结果如下：

书籍参考：Go语言圣经

文章有不足的地方欢迎在评论区指出。

欢迎收藏、点赞、提问。关注顶级饮水机管理员，除了管烧热水，有时还做点别的。