Go如何实现自旋锁

自旋锁简介

在并发编程中,互斥锁(Mutex)是一种常用的同步机制,用于保护临界资源,防止数据竞争。而在某些特定场景下,尤其是当锁的持有时间很短且线程数量有限的情况下,一种更为轻量级的锁------自旋锁(Spin Lock)可以提供更高的性能。

什么是自旋锁

自旋锁是一种忙等待锁,当一个线程尝试获取一个已经被其它线程持有的锁时,这个线程会持续循环检查锁的状态(即"自旋") ,直到锁被释放后获得所有权。这种等待方式避免了线程上下文切换带来的开销,因此比较适用于锁竞争不激烈且锁定时间非常短的场景。

自旋锁原理

当一个线程尝试获取自旋锁时,如果发现锁已被占用,则该线程会进入一个循环,不断检查锁是否已被释放。一旦锁的持有者完成操作并释放锁后,正在自旋的线程即可立即获得锁并继续执行。

自旋锁的优缺点

避免了线程切换和调度的开销,持锁时间短性能好

持锁时间长或锁竞争频繁会导致CPU

如何选择合适的锁

如果不确定锁的持有时间,则选择互斥锁或者读写锁

如果确定锁的持有时间很短,则可以考虑使用自旋锁

如果读操作比较比多,则优先考虑读写锁

自旋锁实例

go 复制代码
package main

import (
	"log"
	"sync"
	"sync/atomic"
)

func init() {
	log.SetFlags(log.LstdFlags)
}

type SpinLock struct {
	state int32
}

func (s *SpinLock) Lock() {
	for !atomic.CompareAndSwapInt32(&s.state, 0, 1) {
		//自旋,不执行任何操作
	}
}
func (s *SpinLock) unLock() {
	atomic.StoreInt32(&s.state, 0)
}
func main() {
	var cnt int
	spinLock := SpinLock{}
	wg := sync.WaitGroup{}

	for i := 0; i < 1000; i++ {
		wg.Add(1)
		go func() {
			spinLock.Lock()
			cnt++
			spinLock.unLock()
			wg.Done()
		}()
	}
	for i := 0; i < 1000; i++ {
		wg.Add(1)
		go func() {
			spinLock.Lock()
			cnt++
			spinLock.unLock()
			wg.Done()
		}()
	}
	wg.Wait()
	log.Println("cnt:", cnt)
}
相关推荐
杜杜的man28 分钟前
【go从零单排】go中的结构体struct和method
开发语言·后端·golang
幼儿园老大*29 分钟前
走进 Go 语言基础语法
开发语言·后端·学习·golang·go
半桶水专家29 分钟前
go语言中package详解
开发语言·golang·xcode
llllinuuu30 分钟前
Go语言结构体、方法与接口
开发语言·后端·golang
cookies_s_s31 分钟前
Golang--协程和管道
开发语言·后端·golang
王大锤439133 分钟前
golang通用后台管理系统07(后台与若依前端对接)
开发语言·前端·golang
产幻少年33 分钟前
golang函数
golang
为什么这亚子33 分钟前
九、Go语言快速入门之map
运维·开发语言·后端·算法·云原生·golang·云计算
半桶水专家35 分钟前
用go实现创建WebSocket服务器
服务器·websocket·golang
材料苦逼不会梦到计算机白富美36 分钟前
golang分布式缓存项目 Day 1
分布式·缓存·golang