Golang sync.Pool 简介与用法-CSDN博客

本文链接：https://blog.csdn.net/K346K346/article/details/95660971

文章目录

1.简介
2.应用
- 2.1 标准库的应用
- 2.2 复用 bytes.Buffer
参考文章

1.简介

Pool 是可伸缩、并发安全的临时对象池，用来存放已经分配但暂时不用的临时对象，通过对象重用机制，缓解 GC 压力，提高程序性能。

2.应用

2.1 标准库的应用

一个比较好的例子是 fmt 包，fmt 包总是需要使用一些 []byte 之类的对象，Golang 建立了一个临时对象池，存放着这些对象，如果需要使用一个 []byte，就去 Pool 中取，如果拿不到就分配一个。这比起不停生成新的 []byte，用完了再等待 GC 回收要高效得多。

注意，sync.Pool 是一个临时对象池，适用于储存一些会在 goroutine 间共享的临时对象，其中保存的任何项都可能随时不做通知地释放掉，所以不适合用于存放诸如 socket 长连接或数据库连接的对象。

type Pool struct {
	// New optionally specifies a function to generate
    // a value when Get would otherwise return nil.
	// It may not be changed concurrently with calls to Get.
	New func() interface{}
	// contains filtered or unexported fields
}

// Get 从 Pool 中获取元素。当 Pool 中没有元素时，会调用 New 生成元素，新元素不会放入 Pool 中。若 New 未定义，则返回 nil。
func (p *Pool) Get() interface{}

// Put 往 Pool 中添加元素 x。
func (p *Pool) Put(x interface{})

2.2 复用 bytes.Buffer

bytes.Buffer 是一个具有读写方法的大小可变的字节缓冲器。其管理的是一个字节切片，在如果不复用，则每次向其中写入数据都会涉及到内存的申请，比较耗时。因此我们可以利用 sync.Pool 来复用它，提高程序性能。

package main

import (
    "bytes"
    "io"
    "os"
    "sync"
    "time"
)

var bufPool = sync.Pool {
    New: func() interface{} {
        return new(bytes.Buffer)
    },
}

func PoolTest(w io.Writer, key, val string) {
    b, _ := bufPool.Get().(*bytes.Buffer)
    b.Reset()
    b.WriteString(time.Now().UTC().Format("2006-01-02 15:04:05"))
    b.WriteByte('|')
    b.WriteString(key)
    b.WriteByte('=')
    b.WriteString(val)
    w.Write(b.Bytes())
	w.Write([]byte("\n"))
    bufPool.Put(b)
}

func main() {
    PoolTest(os.Stdout, "dablelv", "monkey")
}