Go语言并发与管道测试（VSCode for mac）

Go语言并发与管道测试（VSCode for mac）

package main

//导入的包如下
import (
	"fmt"
	"io"
	"io/ioutil"
	"net/http"
	"strings"
	"time"
)

func main() {

	//预定义需要并发获取的三个网页信息
	urlList := []string{"www.163.com", "www.yy.com", "www.qq.com"}
   
	//为了计算main执行时间，先记录开始时间
	start := time.Now()
    
	//创建一个string类型的管道
	ch := make(chan string) 
	
	for _, url := range urlList[0:] {
	   //启动并发执行例程routine
		go fetch(url, ch) 
	}
	for range urlList[0:] {
		//打印管道中收到的数据，如果当前没有数据则会阻塞，直到rotine写入新的数据才会继续执行
		fmt.Println(<-ch)
	}
	
	//计算从start开始，到当前为止经过的时间，转换为秒数
	fmt.Printf("%.2fs total elapsed]\n", time.Since(start).Seconds())
}

//获取指定的网页信息routine
func fetch(url string, ch chan<- string) {
	//为了计算routine执行时间，先记录开始时间
	start := time.Now()
	if !strings.HasPrefix(url, "http://") {
		url = "http://" + url
	}
	resp, err := http.Get(url)
	if err != nil {
		ch <- fmt.Sprint(err)
		return
	}

	//我们使用Copy方法来统计字节数，内容直接扔掉了
	nbytes, err := io.Copy(ioutil.Discard, resp.Body)
	resp.Body.Close()
	if err != nil {
		ch <- fmt.Sprintf("while reading %s:%v", url, err)
		return
	}
   
	//计算从start开始，到当前为止经过的时间，转换为秒数
	secs := time.Since(start).Seconds()
	ch <- fmt.Sprintf("%.2fs  %7d  %s", secs, nbytes, url)
}

程序的运行结果如下