go语言学习-编程案例1.6

并发操作

package main

import (
	"fmt"
	"io"
	"io/ioutil"
	"net/http"
	"time"
)

var https = []string{"https://www.baidu.com", "https://github.com/Addision"}

func main() {
	start := time.Now()
	ch := make(chan string) //创建一个字符串通道
	for _, url := range https {
		fmt.Println("url ===", url)
		go fetch(url, ch) //go 创建协程
	}
	for range https {
		fmt.Println(<-ch) //从通道读取信息
	}
	fmt.Printf("%.2fs elapsed\n", time.Since(start).Seconds())
}

func fetch(url string, ch chan<- string) {
	start := time.Now()
	resp, err := http.Get(url)
	if err != nil {
		ch <- fmt.Sprint(err)
		return
	}
	nbytes, err := io.Copy(ioutil.Discard, resp.Body)
	resp.Body.Close()    //及时关闭，释放资源
	if err != nil {
		ch <- fmt.Sprintf("while reading %s: %v", url, err)
		return
	}
	secs := time.Since(start).Seconds()
	ch <- fmt.Sprintf("%.2fs %7d %s", secs, nbytes, url) //将格式化信息写入通道
}