golang mysql 非阻塞_golang非阻塞锁的简单实现

对于复杂类型如 container/list ，需要在所有读写操作上使用 sync.mutex 互斥锁以保证数据一致性，互斥锁并发情况下，Lock 操作会阻塞，一直等到其他线程Unlock，但是有的时候因为有一个耗时比较长的操作一直占用锁，我们想让其他线程不在Lock上一直阻塞，而是直接走其他业务流程。

举一个很简单的场景：在多人并发抽奖环节，为了保证不出现负库存，我们可以通过竞争锁，第一个获取锁的人可能中奖，而其他并发过来的请求获取锁失败(而不是一直阻塞在Lock())，直接当做未中奖处理。

思考了一下，有一个思路，我们可以用两道锁，第一道锁用来判断锁状态，第二道锁才是真正的耗时任务用的锁。

直接上代码

type NBLocker struct{

l1 sync.Mutex

l2 sync.Mutex

locked bool

}

func (NBLocker *NBLocker) Lock() (success bool) {

NBLocker.l1.Lock()

defer NBLocker.l1.Unlock()

if NBLocker.locked == false {

NBLocker.locked = true

success = true

NBLocker.l2.Lock()

}

return

}

func (NBLocker *NBLocker) Unlock() {

NBLocker.l1.Lock()

defer NBLocker.l1.Unlock()

NBLocker.locked = false

NBLocker.l2.Unlock()

}

使用

type foo struct {

mux NBLocker

}

func (self *foo) Bong(wg *sync.WaitGroup) {

defer wg.Done()

if !self.mux.Lock() {

fmt.Println("获取锁失败")

return

}

defer self.mux.Unlock()

time.Sleep(time.Second) //停顿一秒

fmt.Println("bong~")

}

func main() {

f := &foo{}

wg := &sync.WaitGroup{}

wg.Add(4)

go f.Bong(wg)

go f.Bong(wg)

go f.Bong(wg)

time.Sleep(time.Second *2)

go f.Bong(wg)

wg.Wait()

}

如果获取第二道锁失败，NBLocker.Lock() 方法会直接返回false，这时候只需要判断一下就可以直接跳过"抽奖环节"

以上例程输出

获取锁失败

获取锁失败

bong~

bong~

有疑问加站长微信联系(非本文作者)