Go项目中如何限制并发数?Atomic必须掌握

Go语言越来越受到大家的青睐,它的高并发在同类语言中表现突出。所以,很多具有高并发属性的项目都开始用Go语言开发。很多时候,我们用Go语言开发一个web server,主要用来接收外部的http请求。那在面对外部高并发的http请求的时候,我们如何保证系统不被击垮呢?这就是今天我们要说的限制并发数。

在Go语言的标准库中,为我们提供了sync/atomic包。它将底层硬件提供原子操作封装成了go的函数。借助它我们就可以实现并发数控制。直接上代码:

首选,我们定义一个原子操作的变量

接着,我们定义两个方法,分别对原子变量进行加减操作

注意:原子操作的变量是需要传入指针。这里我们定义了IncrAndGetConcurrentCount方法 和 DecrAndGetConcurrentCount方法。用于对concurrentCount变量进行原子操作,并且返回操作之后的值。

最后,我们在业务方法中使用。

每一次请求进来以后,我们调用IncrAndGetConcurrentCount方法进行原子的加操作,然后判断返回的值是否大于我们设定的并发数,如果大于则直接返回,否则执行正常的逻辑。请求完成以后我们调用DecrAndGetConcurrentCount,对原子变量进行减的操作。这里为了防止异常出现不能对原子变量进行减的操作,我们将DecrAndGetConcurrentCount放到defer之后。

到此,我们就可以控制系统的并发数,防止瞬间流量过高而导致系统宕机。

  • 发表于:
  • 原文链接https://kuaibao.qq.com/s/20191024A08A4A00?refer=cp_1026
  • 腾讯「云+社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。
  • 如有侵权,请联系 yunjia_community@tencent.com 删除。

扫码关注云+社区

领取腾讯云代金券