版权声明:本文为Sunface(孙飞)原创文章,转载请注明出处 https://blog.csdn.net/abv123456789/article/details/51153474
本来想写一个例子来给团队的人阐述一下Go中的对象复用思想,写完后发现这种思想也许某些同学还不知道,现分享出来。
// Go中为了减少海量的内存分配,控制垃圾回收时间,对于热点数据采用池的思想去处理是
// 非常棒的,下面的例子就是一个简单的tcp读取缓存池复用
// 实现了一个读取tcp长连接的buffer池,每次读取一条新连接时,先去池里读取
// 如果不存在,则创建新的
// 注意buf在放回池里时,没有进行归零处理,因为conn.Read每次读取时候,
// 都会从buf.b[0]的位置开始写
package main
import (
"fmt"
"net"
"sync"
)
var bufPool sync.Pool
type buf struct {
b []byte
}
func main() {
ln, _ := net.Listen("tcp", ":8082")
for {
conn, err := ln.Accept()
if err != nil {
fmt.Println("accept error:", err)
}
//为每一条进来的连接都开启一个goroutine处理
go func() {
var bf *buf
v := bufPool.Get()
if v == nil {
//若不存在buf,创建新的
fmt.Println("no buffer ,need create!")
bf = &buf{
b: make([]byte, 10),
}
} else {
// 池里存在buf,v这里是interface{},需要做类型转换
bf = v.(*buf)
}
//从conn中连续读取数据
//注意conn.Read有个特点,如果bf.b容量不足,那么会分两次读取
//而不会去扩展bf.b
for {
_, err := conn.Read(bf.b)
//这里只是例子,因此没有对io.EOF做特殊处理
if err != nil {
fmt.Printf("conn error: %#v", err)
break
}
}
// bf使命完成,放入池中
bufPool.Put(bf)
}()
}
}
有疑问加站长微信联系(非本文作者)