go鎖基礎 - atomic、sema

来源:https://www.cnblogs.com/studyios/archive/2023/12/01/17869424.html
-Advertisement-
Play Games

1、首先先把配置文件從jar中抽離 示例代碼: <plugin> <groupId>org.apache.maven.plugins</groupId> <artifactId>maven-jar-plugin</artifactId> <version>3.2.0</version> <confi ...


atomicsema是實現go中鎖的基礎,簡單看下他們的實現原理。

atomic

atomic 常用來作為保證原子性的操作。

當多個協程,同時一個數據進行操作時候,如果不加鎖,最終的很難得到想要的結果。

  var p int64 = 0
  func add() {
  	p = p + 1
  }

  func main() {

  	for i := 0; i < 1000; i++ {
  		go add()
  	}
  	time.Sleep(time.Second * 5)
  	fmt.Println(p) //982
  }

這種情況下,最終列印的 都不會是1000,每次不固定。

改成atomic 能解決
var p int64 = 0
func add() {
	atomic.AddInt64(&p, 1)
}
func main() {

	for i := 0; i < 1000; i++ {
		go add()
	}
	time.Sleep(time.Second * 5)
	fmt.Println(p)
}

atomic 為什麼能做到?

 TEXT	sync∕atomic·AddInt64(SB), NOSPLIT, $0-24
	GO_ARGS
	MOVD	$__tsan_go_atomic64_fetch_add(SB), R9
	BL	racecallatomic<>(SB)
	MOVD	add+8(FP), R0	// convert fetch_add to add_fetch
	MOVD	ret+16(FP), R1
	ADD	R0, R1, R0
	MOVD	R0, ret+16(FP)
	RET

老的版本中是能見到,lock 這種操作系統級別的鎖,新版的go已經改寫了這塊邏輯,但是能猜想到效果肯定一樣。 如果有理清楚的,評論區可以交流下。

小結:

原子操作是一種硬體層面加鎖的機制
保證操作一個變數的時候,其他協程、線程無法訪問

sema

幾乎在go的每個鎖的定義都能看到sema的身影,理解了sema再看 互斥鎖、讀寫鎖就會很好理解。

信號量鎖/信號鎖
核心是一個uint32值,含義是同時可併發的數量
每一個sema 鎖都對應一個SemaRoot結構體
SemaRoot中有一個平衡二叉樹用於協程排隊

例如:

 type Mutex struct {
  	state int32
  	sema  uint32 
 }

sema的uint32中的 每一個數,背後都對應一個  semaRoot的結構體
type semaRoot struct {
	lock  mutex
	treap *sudog        // root of balanced tree of unique waiters.
	nwait atomic.Uint32 // Number of waiters. Read w/o the lock.
}

type sudog struct {
        g *g              // 包含了 協程 g
	next *sudog   // 下一個
	prev *sudog  
	elem unsafe.Pointer // data element (may point to stack)
 }

結構如下:

這裡可以先講下,當這個 sema uint32 值,初始化時候,大於0 比如 賦值5,就代表著,併發時候,有5個協程可以獲取鎖。其他協程需要等待前面5個釋放了,才能進入。

sema 大於0

   // 獲取sema鎖。大於0的情況
func semacquire1(addr *uint32, lifo bool, profile semaProfileFlags, skipframes int, reason waitReason) {
    	gp := getg()
    	if gp != gp.m.curg {
    		throw("semacquire not on the G stack")
    	}
    	// Easy case. // 容易的情況
    	if cansemacquire(addr) {
    		return
    	}
      // 方法很長,先看簡單的部分。
}
  
func cansemacquire(addr *uint32) bool {
	for {
		v := atomic.Load(addr) // 根據sema的地址,獲取int32的值
		if v == 0 {   // 如果未0了,就獲取失敗了
			return false
		}
        // 大於0 ,則把 sema的值減去1
		if atomic.Cas(addr, v, v-1) { // cas 就是 CompareAndSwapInt 的底層實現
			return true
		}
	}
}

到此,對sema為什麼只是定義為一個 uint32的值有了大致理解,就是一個控制能有多少個協程同時獲取鎖的值。

看下釋放:

func semrelease1(addr *uint32, handoff bool, skipframes int) {
	root := semtable.rootFor(addr)
	atomic.Xadd(addr, 1) // 給sema的值 加上1

	// Easy case: no waiters?
	// This check must happen after the xadd, to avoid a missed wakeup
	// (see loop in semacquire).
	if root.nwait.Load() == 0 { // 如果沒有 nwait在等待,就直接結束。 
		return
	}
}

nwait 就是等待協程的個數。

小結, 當sema的值大於0 :

獲取鎖:uint32減1 ,獲取成功

釋放鎖:uint32加1,釋放成功

sema值等於0

再看 semacquire1

func semacquire1(addr *uint32, lifo bool, profile semaProfileFlags, skipframes int, reason waitReason) {

     / / Harder case:
	//	increment waiter count
	s := acquireSudog()
	root := semtable.rootFor(addr) // 根據sema的地址,獲取了包含 sudog的隊列
	for {
		root.queue(addr, s, lifo) // 將新的協程放入這個等待隊列中
		goparkunlock(&root.lock, reason, traceBlockSync, 4+skipframes) 
        // 主動調用協程的gopark,讓它休眠,gopark的說明看 go GMP中有講
	}
	releaseSudog(s)
}

//再看釋放
func semrelease1(addr *uint32, handoff bool, skipframes int) {
	root := semtable.rootFor(addr)
	atomic.Xadd(addr, 1)
	if root.nwait.Load() == 0 {
		return
	}
    // 如果等待隊列不是0 ,就需要釋放一個
	// Harder case: search for a waiter and wake it.
	lockWithRank(&root.lock, lockRankRoot)
	
	s, t0 := root.dequeue(addr) // 從全局的隊列中,取出一個
	if s != nil {
		root.nwait.Add(-1) // 把等待的數量減一
	}
	unlock(&root.lock) //操作全局隊列都需要加鎖
}

小結,當sema的值等於0時候:

獲取鎖:協程休眠,進入堆樹等待
釋放鎖:從堆樹中取出一個協程,喚醒
sema 鎖退化成一個專用休眠隊列

有沒有可能sema的值,小於0 ?

看看sema的定義 `uint32` 所以 不可能。

總結下:

atomic原子操作是一種硬體層面的加鎖機制。

sema 背後是一整套鎖的管理和等待的機制,開發者在使用時候,感知不到。

sema的值就是能同時獲取鎖協程的個數。sema的地址作為了休眠等待隊列(平衡樹)的key。

您的分享是我們最大的動力!

-Advertisement-
Play Games
更多相關文章
  • acwing week2 基礎演算法3總結 總結點1:雙指針演算法 //常用模版框架 for (int i = 0, j = 0; i < n; i ++ ) { while (j < i && check(i, j)) j ++ ; } 常見問題分類: (1) 對於一個序列,用兩個指針維護一段區間 ( ...
  • 十九、函數(二) 1、函數參數之接受不定量參數 1)普通函數不定量傳參用法 //接受不定量參數的函數 #include <cstdarg> //引入頭文件cstdarg int Add(unsigned count, ...) //第一個參數為參數的個數,第二個參數為三個. { int rt{}; ...
  • 小市值選股策略的核心在於通過綜合分析公司的基本面、行業定位、財務健康狀況以及市場趨勢, 來尋找那些被市場低估但具備顯著成長潛力的股票,同時也要重視風險管理和投資組合的多樣化。 今天來給大家分享下小市值策略代碼如下: # 顯式導入 BigQuant 相關 SDK 模塊 from bigdatasour ...
  • 工作中,經常遇到需要重試的場景,最簡單的方式可以用try...catch...加while迴圈來實現。那麼,有沒有統一的、優雅一點兒的處理方式呢?有的,Spring Retry就可以幫我們搞定重試問題。 關於重試,我們可以關註以下以下幾個方面: 什麼情況下去觸發重試機制 重試多少次,重試的時間間隔 ...
  • CompletableFuture非同步回調 CompletableFuture簡介 CompletableFuture被用於非同步編程,非同步通常意味著非阻塞,可以使得任務單獨允許在與主線程分離的其他線程中,並且通過回調可以在主線程中得到非同步任務的執行狀態,是否完成,和是否異常信息。 Completab ...
  • hutool工具包可以幫我們完成這件事,幾行代碼可以實現,我們提供兩種方式,壓縮本地文件和壓縮記憶體流。 壓縮本地文件 @Test public void zip(){ String entryName = "d:\\codegen\\1"; String zipFilePath = "d:\\cod ...
  • C語言迴圈結構詳解 在C語言中,迴圈結構是一種重要的控制結構,它允許我們重覆執行一段代碼,以達到特定的目的。迴圈結構可以幫助我們簡化重覆性的任務,提高代碼的效率。本篇文章將深入探討C語言中的迴圈結構,包括 while 迴圈、for 迴圈、do-while 迴圈以及迴圈中的控制語句。 1. while ...
  • Fork/Join框架簡介 Fork/Join框架簡介 Fork/Join它可以將一個大的任務拆分成多個子任務並行處理,最後將子任務結果合成並最後的計算結果,併進行輸出。FOrk/Join框架要完成兩件事情。Fork:把一個複雜的任務進行分析,大任務拆分成小任務;Join:把拆分的結果進行合併。 1 ...
一周排行
    -Advertisement-
    Play Games
  • 示例項目結構 在 Visual Studio 中創建一個 WinForms 應用程式後,項目結構如下所示: MyWinFormsApp/ │ ├───Properties/ │ └───Settings.settings │ ├───bin/ │ ├───Debug/ │ └───Release/ ...
  • [STAThread] 特性用於需要與 COM 組件交互的應用程式,尤其是依賴單線程模型(如 Windows Forms 應用程式)的組件。在 STA 模式下,線程擁有自己的消息迴圈,這對於處理用戶界面和某些 COM 組件是必要的。 [STAThread] static void Main(stri ...
  • 在WinForm中使用全局異常捕獲處理 在WinForm應用程式中,全局異常捕獲是確保程式穩定性的關鍵。通過在Program類的Main方法中設置全局異常處理,可以有效地捕獲並處理未預見的異常,從而避免程式崩潰。 註冊全局異常事件 [STAThread] static void Main() { / ...
  • 前言 給大家推薦一款開源的 Winform 控制項庫,可以幫助我們開發更加美觀、漂亮的 WinForm 界面。 項目介紹 SunnyUI.NET 是一個基於 .NET Framework 4.0+、.NET 6、.NET 7 和 .NET 8 的 WinForm 開源控制項庫,同時也提供了工具類庫、擴展 ...
  • 說明 該文章是屬於OverallAuth2.0系列文章,每周更新一篇該系列文章(從0到1完成系統開發)。 該系統文章,我會儘量說的非常詳細,做到不管新手、老手都能看懂。 說明:OverallAuth2.0 是一個簡單、易懂、功能強大的許可權+可視化流程管理系統。 有興趣的朋友,請關註我吧(*^▽^*) ...
  • 一、下載安裝 1.下載git 必須先下載並安裝git,再TortoiseGit下載安裝 git安裝參考教程:https://blog.csdn.net/mukes/article/details/115693833 2.TortoiseGit下載與安裝 TortoiseGit,Git客戶端,32/6 ...
  • 前言 在項目開發過程中,理解數據結構和演算法如同掌握蓋房子的秘訣。演算法不僅能幫助我們編寫高效、優質的代碼,還能解決項目中遇到的各種難題。 給大家推薦一個支持C#的開源免費、新手友好的數據結構與演算法入門教程:Hello演算法。 項目介紹 《Hello Algo》是一本開源免費、新手友好的數據結構與演算法入門 ...
  • 1.生成單個Proto.bat內容 @rem Copyright 2016, Google Inc. @rem All rights reserved. @rem @rem Redistribution and use in source and binary forms, with or with ...
  • 一:背景 1. 講故事 前段時間有位朋友找到我,說他的窗體程式在客戶這邊出現了卡死,讓我幫忙看下怎麼回事?dump也生成了,既然有dump了那就上 windbg 分析吧。 二:WinDbg 分析 1. 為什麼會卡死 窗體程式的卡死,入口門檻很低,後續往下分析就不一定了,不管怎麼說先用 !clrsta ...
  • 前言 人工智慧時代,人臉識別技術已成為安全驗證、身份識別和用戶交互的關鍵工具。 給大家推薦一款.NET 開源提供了強大的人臉識別 API,工具不僅易於集成,還具備高效處理能力。 本文將介紹一款如何利用這些API,為我們的項目添加智能識別的亮點。 項目介紹 GitHub 上擁有 1.2k 星標的 C# ...