9.2 共享内存并发机制

peter
• 阅读 1076
// 本节可以通过下面的代码来说明
package share_mem

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)
    // 输出:counter = 4625
    // 这不是线程安全的程序
}

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++
        }()
    }
    // 防止外面的程序执行的太快,导致内部的协程还没有执行完,从而无法得到正确的结果
    // 也可以使用 waitGroup来做这件事
    time.Sleep(1 * time.Second)
    t.Logf("counter = %d", counter)
    // 输出:counter = 5000
}

func TestCounterWaitGroup(t *testing.T) {
    var mut sync.Mutex
    var wg sync.WaitGroup
    counter := 0
    for i := 0; i < 5000; i++ {
        wg.Add(1)
        go func() {
            defer func() {
                mut.Unlock()
            }()
            mut.Lock()
            counter++
            wg.Done()
        }()
    }
    // Wait() 方法会被阻塞在这里
    wg.Wait()
    t.Logf("counter = %d", counter)
    // 输出:counter = 5000
}
点赞
收藏
评论区
推荐文章

暂无数据