golang sync.Pool 指针数据覆盖问题

03-11 1219阅读

场景

golang sync.Pool 指针数据覆盖问题
(图片来源网络,侵删)

1. sync.Pool设置

var stringPool = sync.Pool{
	New: func() any {
		return new([]string)
	},
}
func NewString() *[]string {
	v := stringPool.Get().(*[]string)
	return v
}
func PutString(s *[]string) {
	if s == nil {
		return
	}
	if cap(*s) > 2048 {
		s = nil
	} else {
		*s = (*s)[:0]
		stringPool.Put(s)
	}
}

2.使用sync.Pool

func Test_Pool(t *testing.T) {
	dataSlice1 := demoData()
	dataSlice2 := demoData()
	dataSlice2[1] = "test4"
	fmt.Printf("dataSlice1:%v %p,dataSlice2:%v %p\n", dataSlice1, dataSlice1, dataSlice2, dataSlice2)
}
func demoData() []string {
	strsPtr := NewString()
	strs := *strsPtr
	defer func() {
		*strsPtr = strs
		PutString(strsPtr)
	}()
	strs = append(strs, "test1", "test2")
	return strs
}

打印结果:dataSlice1:[test1 test4] 0xc0000a6400,dataSlice2:[test1 test4] 0xc0000a6400

可以看到两个slice地址相同,内部使用同一个地址的数组,导致两次获取的数据互相影响

3.解决方法1

func Test_Pool(t *testing.T) {
	dataSlice1 := demoData()
	dataSlice2 := demoData()
	dataSlice2[1] = "test4"
	fmt.Printf("dataSlice1:%v %p,dataSlice2:%v %p\n", dataSlice1, dataSlice1, dataSlice2, dataSlice2)
}
func demoData() []string {
	strsPtr := NewString()
	strs := *strsPtr
	defer func() {
		*strsPtr = strs
		PutString(strsPtr)
	}()
	strs = append(strs, "test1", "test2")
	// 深复制
	var items = make([]string, len(strs))
	copy(items, strs)
	return items
}

使用深复制,在put回sync.Pool中之前把数据复制返回,但这样资源池失去了意义,获取到资源后有进行了一次内存的申请

4.解决方法2

我们看下golang语言源码怎么解决的

参考 go/src/fmt/print.go 302行 Fprintln方法

func Fprintln(w io.Writer, a ...any) (n int, err error) {
	p := newPrinter()
	p.doPrintln(a)
	n, err = w.Write(p.buf)
	p.free()
	return
}

可以看到306行有p.free()代码,newPrinter()和free()之间进行数据处理,数据处理完成之后再把资源返回给sync.Pool

总结:不是任何场景都适合用sync.Pool,需要关注并发情况下资源池中数据同步修改影响的问题。

VPS购买请点击我

文章版权声明:除非注明,否则均为主机测评原创文章,转载或复制请以超链接形式并注明出处。

目录[+]