golang并发http请求几种方式

概述

go中方便的协程写法, 让并发变得很简单, 让开发者很多时候不需要关心协程(线程)管理的问题, 不过也是有值得注意的地方.
下面以一个http请求的例子, 请求的服务是进制数据-手机号码归属地查询.(找了很久才找到方便学demo的免费API, 想跑demo的朋友注册一下在代码中填上自己的appkey即可)

并发方式

平时常用到3种方式做并发请求.

  1. 简单粗暴, 一次请求开一个线程, 线程中完成请求与响应处理.
  2. 优雅一点, 限制线程池, 以master-worker的方式处理并发, 最后又把响应统一处理.
  3. 优雅地限制线程池(2020.7.15更新)

1. 简单并发

代码

package main

import (
	"fmt"
	"io/ioutil"
	"net/http"
	"sync"
	"time"
)
func main()  {
	var numberTasks = [5]string{"13456755448", " 13419385751", "13419317885", " 13434343439","13438522395"}
	client = &http.Client{}
	beg := time.Now()
	wg := &sync.WaitGroup{}
	for _, keyword := range numberTasks{
		wg.Add(1)
		go func(keyword string, group *sync.WaitGroup) {
			body, err := NumberQueryRequest(keyword)
			if err != nil{
				fmt.Printf("error occurred in query keyword: %s, error: %s\n",keyword, err.Error() )
			}else{
				fmt.Printf("search %s success, data size is %d\n, body is %s\n",keyword, len(body), string(body))
			}
			group.Done()
		}(keyword, wg)
	}
	wg.Wait() //waitGroup在主线程等待所有任务完成
	fmt.Printf("time consumed: %fs", time.Now().Sub(beg).Seconds())
}

var client *http.Client

//GET请求的通用写法, 注意复用client即可.
func NumberQueryRequest(keyword string)(body []byte, err error){
	url := fmt.Sprintf("https://api.binstd.com/shouji/query?appkey=df2720f76a0991fa&shouji=%s", keyword)
	req, err := http.NewRequest("GET", url, nil)
	if err != nil{
		return nil, err
	}
	resp, err := client.Get(url)
	if err != nil{
		return nil, err
	}
	if resp.StatusCode != http.StatusOK{
		data, _ := ioutil.ReadAll(resp.Body)
		return nil, fmt.Errorf("response status code is not OK, response code is %d, body:%s", resp.StatusCode, string(data))
	}
	if resp != nil && resp.Body != nil{
		defer resp.Body.Close()
	}
	body, err = ioutil.ReadAll(resp.Body)
	if err != nil{
		return nil, err
	}
	return body, nil
}

解析

  • 在for循环中有多少个任务就go出去多少个协程,没有限制,在查询量固定或者不大的时候没什么问题.
  • 没有用到 channel的特性, 仅利用了多核调度.

2. master-worker形式

代码

package main

import (
	"fmt"
	"io/ioutil"
	"net/http"
	"sync"
	"time"
)
const (
	routineCountTotal = 5 //限制线程数
)

func main() {
	var numberTasks = [5]string{"13456755448", " 13419385751", "13419317885", " 13434343439", "13438522395"}
	client = &http.Client{}
	beg := time.Now()
	wg := &sync.WaitGroup{}
	tasks := make(chan string)
	results := make(chan string)
	//receiver接受响应并处理的函数块, 也可以单独写在一个函数
	go func() {
		for result := range results{
			if result == ""{
				close(results)
			}else{
				fmt.Println("result:", result)
			}
		}
	}()
	for i:= 0; i < routineCountTotal; i++{
		wg.Add(1)
		go worker(wg, tasks, results)
	}
	//分发任务
	for _, task := range numberTasks{
		tasks <- task
	}
	tasks <- "" //worker结束标志
	wg.Wait() //同步结束
	results <- "" // result结束标志
	fmt.Printf("time consumed: %fs", time.Now().Sub(beg).Seconds())
}

func worker(group *sync.WaitGroup, tasks chan string, result chan string){
	for task := range tasks{
		if task == ""{
			close(tasks)
		}else{
			respBody, err := NumberQueryRequest(task)
			if err != nil{
				fmt.Printf("error occurred in NumberQueryRequest: %s\n", task)
				result <- err.Error()
			}else{
				result <- string(respBody)
			}
		}
	}
	group.Done()
}

var client *http.Client

func NumberQueryRequest(keyword string) (body []byte, err error) {
	url := fmt.Sprintf("https://api.binstd.com/shouji/query?appkey=df2720f76a0991fa&shouji=%s", keyword)
	req, err := http.NewRequest("GET", url, nil)
	if err != nil {
		return nil, err
	}
	req.Header.Set("User-Agent", "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/69.0.3497.100 Safari/537.36")
	resp, err := client.Get(url)
	if err != nil {
		return nil, err
	}
	if resp.StatusCode != http.StatusOK {
		data, _ := ioutil.ReadAll(resp.Body)
		return nil, fmt.Errorf("response status code is not OK, response code is %d, body:%s", resp.StatusCode, string(data))
	}
	if resp != nil && resp.Body != nil {
		defer resp.Body.Close()
	}
	body, err = ioutil.ReadAll(resp.Body)
	if err != nil {
		return nil, err
	}
	return body, nil
}

输出

result: {"status":0,"msg":"ok","result":{"shouji":"13434343439","province":"广东","city":"广州","company":"中国移动","cardtype":null,"areacode":"020"}}
result: {"status":0,"msg":"ok","result":{"shouji":"13438522395","province":"四川","city":"攀枝花","company":"中国移动","cardtype":null,"areacode":"0812"}}
result: {"status":0,"msg":"ok","result":{"shouji":"13419317885","province":"四川","city":"广安","company":"中国移动","cardtype":null,"areacode":"0826"}}
result: {"status":0,"msg":"ok","result":{"shouji":"13456755448","province":"浙江","city":"杭州","company":"中国移动","cardtype":null,"areacode":"0571"}}
result: {"status":0,"msg":"ok","result":{"shouji":"13419385751","province":"四川","city":"遂宁","company":"中国移动","cardtype":null,"areacode":"0825"}}
time consumed: 0.399975s

3. 使用带缓冲区的chan控制

完整代码见: https://github.com/chrispinkyang/golimitDemo

先上代码

package golimit

type Glimit struct {
	Num int
	C   chan struct{}
}

func NewG(num int) *Glimit {
	return &Glimit{
		Num: num,
		C : make(chan struct{}, num),
	}
}

func (g *Glimit) Run(f func()){
	g.C <- struct{}{}
	go func() {
		f()
		<-g.C
	}()
}

使用方式

func main() {
	number := 10
	// 限制线程数2
	g := NewG(2)
	wg := sync.WaitGroup{}
	for i := 0; i < number; i++ {
		wg.Add(1)
		value :=i // 闭包传值方式
		goFunc := func() {
			// 做一些业务逻辑处理
			fmt.Printf("go func: %d\n", value)
			time.Sleep(time.Second)
			wg.Done()
		}
		g.Run(goFunc)
	}
	wg.Wait()
}

并发请求的用法

package main

import (
	"fmt"
	"io/ioutil"
	"net/http"
	"sync"
	"time"
	"./golimit"
)

const (
	routineCountTotal = 5 //限制线程数
)

func main() {
	var numberTasks = [5]string{"13456755448", " 13419385751", "13419317885", " 13434343439", "13438522395"}

	g := golimit.NewG(routineCountTotal)
	wg := &sync.WaitGroup{}
	client = &http.Client{}
	beg := time.Now()
	for i := 0; i < len(numberTasks); i++ {
		wg.Add(1)
		task := numberTasks[i]
		g.Run(func() {
			respBody, err := NumberQueryRequest(task)
			if err != nil {
				fmt.Printf("error occurred in NumberQueryRequest: %s\n", task)
			} else {
				fmt.Printf("response data: %s\n", string(respBody))
			}
			wg.Done()
		})
	}
	wg.Wait()
	fmt.Printf("time consumed: %fs", time.Now().Sub(beg).Seconds())
}
var client *http.Client

func NumberQueryRequest(keyword string) (body []byte, err error) {
	url := fmt.Sprintf("https://api.binstd.com/shouji/query?appkey=df2720f76a0991fa&shouji=%s", keyword)
	req, err := http.NewRequest("GET", url, nil)
	if err != nil {
		return nil, err
	}
	req.Header.Set("User-Agent", "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/69.0.3497.100 Safari/537.36")
	resp, err := client.Get(url)
	if err != nil {
		return nil, err
	}
	if resp.StatusCode != http.StatusOK {
		data, _ := ioutil.ReadAll(resp.Body)
		return nil, fmt.Errorf("response status code is not OK, response code is %d, body:%s", resp.StatusCode, string(data))
	}
	if resp != nil && resp.Body != nil {
		defer resp.Body.Close()
	}
	body, err = ioutil.ReadAll(resp.Body)
	if err != nil {
		return nil, err
	}
	return body, nil
}

输出

response data: {"status":0,"msg":"ok","result":{"shouji":"13456755448","province":"浙江","city":"杭州","company":"中国移动","cardtype":null,"areacode":"0571"}}
response data: {"status":0,"msg":"ok","result":{"shouji":"13438522395","province":"四川","city":"攀枝花","company":"中国移动","cardtype":null,"areacode":"0812"}}
response data: {"status":0,"msg":"ok","result":{"shouji":"13434343439","province":"广东","city":"广州","company":"中国移动","cardtype":null,"areacode":"020"}}
response data: {"status":0,"msg":"ok","result":{"shouji":"13419385751","province":"四川","city":"遂宁","company":"中国移动","cardtype":null,"areacode":"0825"}}
response data: {"status":0,"msg":"ok","result":{"shouji":"13419317885","province":"四川","city":"广安","company":"中国移动","cardtype":null,"areacode":"0826"}}
time consumed: 0.937952s%  

解析

  • 比第一种方式更规范一些, 线程数量可以控制.
  • 两处close通道的地方都用了传特殊空值的方式, 似乎不太优雅, 欢迎有更好的关闭通道的方式互相交流, 谢谢.

第三种方式解析(2020.7.15更新)

  • 巧妙地使用go带缓冲区的通道来实现goroutine控制, 更加简洁, 并且这种方式在多个项目中都可以复用. 不必像第二种方式一样每次都实现一个worker函数.

参考

Go开发中,如何有效控制Goroutine的并发数量

  • 14
    点赞
  • 38
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Golang 短信通道是使用 Go 语言开发的一种用于发送短信的通信工具。在现代社会中,短信通道是一种非常重要和常用的通信方式,它不仅可以用于个人之间的短信互通,还用于企业之间或企业与用户之间的信息传递。 使用 Golang 进行短信通道的开发具有以下几个优点: 1. 快速高效:Go 语言是一种编译型语言,具有非常高的执行效率和处理速度。在短信通道中,高效的执行速度和处理能力可以确保短信的及时发送和接收,保证通信的实时性。 2. 并发支持:Go 语言内置并发机制,使用轻量级的 goroutine 和通道(channel) 可以很方便地实现并发操作。在短信通道中,这种并发支持可以同时处理多个短信请求,提高短信的发送速度和通道的处理能力。 3. 代码简洁易读:Go 语言注重代码的简洁性和可读性,在语法设计上追求简单清晰。使用 Go 语言进行短信通道的开发,代码量相对较少,易于维护和理解。 4. 跨平台兼容:Go 语言的编译器能够生成可执行文件,使得开发的应用程序可以在不同的操作系统平台上运行。这样,使用 Golang 开发的短信通道可以在多个平台上使用,提高了通信的灵活性和可用性。 总的来说,Golang 短信通道是一种高效、可靠、并发支持的通信工具,在各种应用场景中都能展现出良好的性能和稳定性。它不仅可以用于个人之间的短信交流,还广泛应用于企业之间的业务通信和各种机构与用户之间的信息传递中。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值