[go学习笔记]三十一、sync.pool对象缓存

示例代码请访问：github.com/wenjianzhan…

sync.Pool 对象获取

• 尝试从私有对象获取
• 似有对象不存在，尝试从当前 Processor 的共享池获取
• 如果当前Processer 共享池也是空的，那么就尝试去其他Processor 的共享池获取
• 如果所有子池都是空的，最后就用用户指定的new 函数产生一个新对象返回

使用 sync.Pool

pool := &sync.Pool {
  New: func() interface{} {
    return 0
  }
}

arry := poolGet().(int)

...

pool.Put(10)

sync.Pool 对象的生命周期

• GC 会清除 sync.pool 缓存的对象
• 对象的缓存有效期为下一次 GC 之前

package obj_cache

import (
	"fmt"
	"runtime"
	"sync"
	"testing"
)

func TestSyncPool(t *testing.T) {
	pool := &sync.Pool{
		New: func() interface{} {
			fmt.Println("Create a new object.")
			return 100
		},
	}
	v := pool.Get().(int)
	fmt.Println(v)
	pool.Put(3)
	runtime.GC()
	v1, _ := pool.Get().(int)
	fmt.Println(v1)
}

func TestSyncPoolInMultiGroutine(t *testing.T) {
	pool := &sync.Pool{
		New: func() interface{} {
			fmt.Println("Create a new object.")
			return 10
		},
	}
	pool.Put(100)
	pool.Put(100)
	pool.Put(100)

	var wg sync.WaitGroup
	for i := 0; i < 10; i++ {
		wg.Add(1)
		go func(id int) {
			fmt.Println(pool.Get())
			wg.Done()
		}(1)
	}
	wg.Wait()
}

输出

=== RUN   TestSyncPoolInMultiGroutine
100
100
100
Create a new object.
Create a new object.
10
Create a new object.
10
Create a new object.
10
Create a new object.
10
10
Create a new object.
10
Create a new object.
10
--- PASS: TestSyncPoolInMultiGroutine (0.00s)
PASS

Process finished with exit code 0

sync.Pool 总结

• 适合于通用复用，降低复杂对象的创建和 GC 代价
• 协程安全，会有锁的开销
• 生命周期 GC 影响，不适合于做连接池等，需要自己管理生命周期的资源的池化

示例代码请访问：github.com/wenjianzhan…