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

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

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

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

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

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

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

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

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