golang sync.Pool包的使用和一些注意地方

Posted 两脚任从行处来,一灵常与气相随。有时四大熏熏醉,借问青天我是

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了golang sync.Pool包的使用和一些注意地方相关的知识,希望对你有一定的参考价值。

package main;

import (
	"sync"
	"fmt"
	"net"
	"runtime"
)

//sync.Pool是一个可以存或取的临时对象集合
//sync.Pool可以安全被多个线程同时使用,保证线程安全
//注意、注意、注意,sync.Pool中保存的任何项都可能随时不做通知的释放掉,所以不适合用于像socket长连接或数据库连接池。
//sync.Pool主要用途是增加临时对象的重用率,减少GC负担。

func testTcpConnPool() {
	sp2 := sync.Pool{
		New: func() interface{} {
			conn, err := net.Dial("tcp", ":80");
			if err != nil {
				return nil;
			}
			return conn;
		},
	};
	buf := make([]byte, 1024);
	//获取对象
	conn := sp2.Get().(net.Conn);
	//使用对象
	conn.Write([]byte("GET / HTTP/1.1 \\r\\n\\r\\n"));
	n, _ := conn.Read(buf);
	fmt.Println("conn read : ", string(buf[:n]));
	//打印conn的地址
	fmt.Println(conn);
	//把对象放回池中
	sp2.Put(conn);
	//我们人为的进行一次垃圾回收
	runtime.GC();
	//再次获取池中的对象
	conn2 := sp2.Get().(net.Conn);
	//这时发现conn2的地址与上面的conn的地址不一样了
	//说明池中我们之前放回的对象被全部清除了,显然这并不是我们想看到的
	//所以sync.Pool不适合用于scoket长连接或数据库连接池
	fmt.Println(conn2);
}

func main() {
	//我们创建一个Pool,并实现New()函数
	sp := sync.Pool{
		//New()函数的作用是当我们从Pool中Get()对象时,如果Pool为空,则先通过New创建一个对象,插入Pool中,然后返回对象。
		New: func() interface{} {
			return make([]int, 16);
		},
	};
	item := sp.Get();
	//打印可以看到,我们通过New返回的大小为16的[]int
	fmt.Println("item : ", item);

	//然后我们对item进行操作
	//New()返回的是interface{},我们需要通过类型断言来转换
	for i := 0; i < len(item.([]int)); i++ {
		item.([]int)[i] = i;
	}
	fmt.Println("item : ", item);

	//使用完后,我们把item放回池中,让对象可以重用
	sp.Put(item);

	//再次从池中获取对象
	item2 := sp.Get();
	//注意这里获取的对象就是上面我们放回池中的对象
	fmt.Println("item2 : ", item2);
	//我们再次获取对象
	item3 := sp.Get();
	//因为池中的对象已经没有了,所以又重新通过New()创建一个新对象,放入池中,然后返回
	//所以item3是大小为16的空[]int
	fmt.Println("item3 : ", item3);

	//测试sync.Pool保存socket长连接池
	testTcpConnPool();
}

以上是关于golang sync.Pool包的使用和一些注意地方的主要内容,如果未能解决你的问题,请参考以下文章

由浅入深聊聊Golang的sync.Pool

由浅入深聊聊Golang的sync.Pool

深入Golang之sync.Pool详解

golang sync.Pool的用法及实现

golang sync.Pool在1.14中的优化

golang sync.Pool在1.14中的优化