- sync.WaitGroup 使用
sync.WaitGroup用于等待一组goroutine完成。Add方法设置计数器,Done方法减少计数器,Wait方法阻塞直到计数器为0。
package mainimport ("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.WaitGroupfor i := 1; i <= 5; i++ {wg.Add(1)go worker(i, &wg)}wg.Wait()fmt.Println("All workers done")
}
- sync.Mutex 锁的使用
sync.Mutex提供了基本的互斥锁功能,Lock加锁,Unlock解锁。它用于保护临界区,确保同时只有一个goroutine可以执行临界区代码。
package mainimport ("fmt""sync"
)type Counter struct {val intmux 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.WaitGroupcounter := Counter{}for i := 0; i < 1000; i++ {wg.Add(1)go func() {defer wg.Done()counter.Increment()}()}wg.Wait()fmt.Println(counter.Value())
}
- sync.Pool 对象复用
sync.Pool用于存储和复用临时对象,可以减少内存分配和降低GC压力。
package mainimport ("fmt""sync"
)func main() {var pool sync.Poolpool.New = func() interface{} {return "New object"}obj := pool.Get()fmt.Println(obj) // New objectpool.Put(obj)obj2 := pool.Get()fmt.Println(obj2) // New object, could be the same object as obj
}
- sync/atomic 原子操作
sync/atomic包提供了一系列原子操作函数,用于进行无锁的并发安全编程。
package mainimport ("fmt""sync""sync/atomic"
)func main() {var wg sync.WaitGroupvar counter int32for 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变量的增加操作是原子的,即使在并发环境下也能保证数据的一致性和完整性。