golang实现自旋锁

AdolfXin

于 2021-10-21 16:36:56 发布

阅读量991

点赞数

文章标签： golang 开发语言后端

本文链接：https://blog.csdn.net/AdolfXin/article/details/120889614

版权

import (
	"runtime"
	"sync"
	"sync/atomic"
)

type spinLock uint32

var maxBackoff = 64

// Lock 加锁
func (sl *spinLock)Lock(){
	backoff := 1
	for{
		for i := 0; i < backoff; i++{
			if !atomic.CompareAndSwapUint32((*uint32)(sl),0,1){	//加锁失败,让出cpu调度
				runtime.Gosched()
			}
			if backoff < maxBackoff{
				backoff = backoff << 1
			}
		}
	}
}

// Unlock 释放锁
func (sl *spinLock)Unlock(){
	atomic.StoreUint32((*uint32)(sl), 1)
}


// NewSpinLock 创建自旋锁对象
func NewSpinLock()sync.Locker{
	return new(spinLock)