sync包常用并发安全数据结构

  1. sync.WaitGroup 使用
    sync.WaitGroup用于等待一组goroutine完成。Add方法设置计数器,Done方法减少计数器,Wait方法阻塞直到计数器为0。
package main

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

func worker(id int, wg *sync.WaitGroup) {
    defer wg.Done()
    fmt.Printf("Worker %d starting\n", id)
    time.Sleep(time.Second)
    fmt.Printf("Worker %d done\n", id)
}

func main() {
    var wg sync.WaitGroup
    for i := 1; i <= 5; i++ {
        wg.Add(1)
        go worker(i, &wg)
    }
    wg.Wait()
    fmt.Println("All workers done")
}
  1. sync.Mutex 锁的使用
    sync.Mutex提供了基本的互斥锁功能,Lock加锁,Unlock解锁。它用于保护临界区,确保同时只有一个goroutine可以执行临界区代码。
package main

import (
    "fmt"
    "sync"
)

type Counter struct {
    val int
    mux sync.Mutex
}

func (c *Counter) Increment() {
    c.mux.Lock()
    c.val++
    c.mux.Unlock()
}

func (c *Counter) Value() int {
    return c.val
}

func main() {
    var wg sync.WaitGroup
    counter := Counter{}

    for i := 0; i < 1000; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            counter.Increment()
        }()
    }

    wg.Wait()
    fmt.Println(counter.Value())
}
  1. sync.Pool 对象复用
    sync.Pool用于存储和复用临时对象,可以减少内存分配和降低GC压力。
package main

import (
    "fmt"
    "sync"
)

func main() {
    var pool sync.Pool

    pool.New = func() interface{} {
        return "New object"
    }

    obj := pool.Get()
    fmt.Println(obj) // New object
    pool.Put(obj)

    obj2 := pool.Get()
    fmt.Println(obj2) // New object, could be the same object as obj
}
  1. sync/atomic 原子操作
    sync/atomic包提供了一系列原子操作函数,用于进行无锁的并发安全编程。
package main

import (
    "fmt"
    "sync"
    "sync/atomic"
)

func main() {
    var wg sync.WaitGroup
    var counter int32

    for i := 0; i < 1000; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            atomic.AddInt32(&counter, 1)
        }()
    }

    wg.Wait()
    fmt.Println(counter) // 1000
}

在这个示例中,我们使用atomic.AddInt32来确保对counter变量的增加操作是原子的,即使在并发环境下也能保证数据的一致性和完整性。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值