温馨提示×

Go语言中怎么实现锁机制

小亿
85
2024-01-27 18:15:49
栏目: 编程语言

在Go语言中,可以使用sync包中的Mutex类型来实现锁机制。Mutex是互斥锁,可以用来保护临界区,防止多个goroutine同时访问共享资源。

下面是一个使用Mutex实现锁机制的示例代码:

package main

import (
	"fmt"
	"sync"
	"time"
)

var (
	count int
	lock  sync.Mutex
)

func main() {
	wg := sync.WaitGroup{}
	for i := 0; i < 10; i++ {
		wg.Add(1)
		go increment(&wg)
	}
	wg.Wait()
	fmt.Println("Final count:", count)
}

func increment(wg *sync.WaitGroup) {
	lock.Lock() // 加锁
	defer lock.Unlock() // 解锁
	defer wg.Done()

	// 模拟耗时操作
	time.Sleep(time.Millisecond * 100)

	count++
}

在上述代码中,首先创建了一个全局变量count和一个Mutex类型的lock变量。然后,在increment函数中,使用lock.Lock()方法对临界区进行加锁,确保只有一个goroutine可以进入临界区。在临界区内,对count进行了递增操作。最后,使用lock.Unlock()方法对临界区进行解锁。

运行以上代码,可以得到正确的递增结果,并且保证了多个goroutine不会同时访问临界区,实现了锁机制。

0