Go基础--19 共享内存机制

最新推荐文章于 2024-12-05 09:00:00 发布

原创最新推荐文章于 2024-12-05 09:00:00 发布 · 272 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#多线程 #golang

Go 专栏收录该内容

29 篇文章

订阅专栏

本文通过三个并发编程示例，对比分析了使用Lock、WaitGroup和互斥锁实现线程安全计数器的方法。在TestCounter函数中，未使用同步机制导致计数不准确。而在TestCounterThreadSafe中，引入Mutex解决了并发问题。最后，TestCounterWaitGroup利用WaitGroup确保所有goroutine执行完毕后再进行计数，确保结果正确。

Lock 与 waitGroup

package share_memory

import (
	"sync"
	"testing"
	"time"
)

func TestCounter(t *testing.T){
	counter := 0
	for i:=0; i<5000; i++{
		go func() {
			counter++
		}()
	}
	time.Sleep( 1 * time.Second)
	t.Logf("counter = %d",counter)
}

func TestCounterThreadSafe(t *testing.T){
	var mut sync.Mutex
	counter := 0
	for i:=0; i<5000; i++{
		go func() {
			defer func() {
				mut.Unlock()
			}()
			mut.Lock()
			counter++
		}()
	}
	//time.Sleep( 1 * time.Second)
	t.Logf("counter = %d",counter)
}

func TestCounterWaitGroup(t *testing.T){
	var wg sync.WaitGroup
	var mux sync.Mutex
	counter := 0
	for i:=0; i<5000; i++{
		wg.Add(2)
		go func() {
			defer func() {
				mux.Unlock()
			}()
			mux.Lock()
			counter++
			wg.Done()
		}()
		go func() {
			defer func() {
				mux.Unlock()
			}()
			mux.Lock()
			counter++
			wg.Done()
		}()
		wg.Wait()
	}
	//time.Sleep( 1 * time.Second)
	t.Logf("counter = %d",counter)

}