Golang之不可重入函数实现

女主宣言

一个不可重入的函数就是一个在任何时间点只能执行一次的函数，不管它被调用了多少次，以及有多少goroutines。

本篇文章说明了阻塞不可重入函数，并在golang中产生不可重入的函数实现。

某个服务是对某些条件进行轮询，每秒监视一些状态。我们希望每个状态都可以独立地检查，而不需要阻塞。实现可能是这样的：

func main() {
    tick := time.Tick(time.Second)
    go func() {
        for range tick {
            go CheckSomeStatus()
            go CheckAnotherStatus()
        }
    }()
}

我们选择在自己的goroutine中运行每个状态检查，以便 CheckAnotherStatus() 不会等待 CheckSomeStatus() 完成。

每一项检查通常都要花费很短的时间，而且比一秒要少得多。但是，如果 CheckAnotherStatus() 本身需要超过一秒的时间运行，会发生什么呢？可能会有一个意外的网络或磁盘延迟影响检查的执行时间。

在同一时间执行两次的函数是否有意义？如果没有，我们希望它是不可重入的。

防止函数多次运行的简单方法是使用sync.Mutex。

假设我们只关心从上面的循环调用这个函数，我们可以从函数外面实现锁：

import (
    "sync"
    "time"
)

func main() {
    tick := time.Tick(time.Second)
    var mu sync.Mutex
    go func() {
        for range tick {
            go CheckSomeStatus()
            go func() {
                mu.Lock()
                defer mu.Unlock()
                CheckAnotherStatus()
            }()
        }
    }()
}

上面的代码保证了 CheckAnotherStatus() 不是由循环的多次迭代执行的。在以前执行 CheckAnotherStatus() 的时候，循环的任何后续迭代都会被互斥锁阻塞。

阻塞解决方案具有以下属性：

• 它确保了许多“CheckAnotherStatus()”的调用作为循环迭代的次数。

• 假设一个执行“CheckAnotherStatus()”的停顿，随后的迭代会导致请求调用相同函数的请求。

import (
    "sync/atomic"
    "time"
)

func main() {
    tick := time.Tick(time.Second)
    var reentranceFlag int64
    go func() {
        for range tick {
            go CheckSomeStatus()
            go func() {
                if atomic.CompareAndSwapInt64(&reentranceFlag, 0, 1) {
                    defer atomic.StoreInt64(&reentranceFlag, 0)
                } else {
                     return
                }
                CheckAnotherStatus()
            }()
        }
    }()
}

HULK一线技术杂谈

由360云平台团队打造的技术分享公众号，内容涉及 云计算 、 数据库 、 大数据 、 监控 、 泛前端 、 自动化测试 等众多技术领域，通过夯实的技术积累和丰富的一线实战经验，为你带来最有料的技术分享