一个简单的例子阐述Go的对象复用思想

来源:互联网 发布:gui table 选中列 java 编辑:程序博客网 时间:2024/06/06 03:29

本来想写一个例子来给团队的人阐述一下Go中的对象复用思想,写完后发现这种思想也许某些同学还不知道,现分享出来。

// Go中为了减少海量的内存分配,控制垃圾回收时间,对于热点数据采用池的思想去处理是// 非常棒的,下面的例子就是一个简单的tcp读取缓存池复用// 实现了一个读取tcp长连接的buffer池,每次读取一条新连接时,先去池里读取// 如果不存在,则创建新的// 注意buf在放回池里时,没有进行归零处理,因为conn.Read每次读取时候,// 都会从buf.b[0]的位置开始写package mainimport ("fmt""net""sync")var bufPool sync.Pooltype 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 *bufv := 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.bfor {_, err := conn.Read(bf.b)//这里只是例子,因此没有对io.EOF做特殊处理if err != nil {fmt.Printf("conn error: %#v", err)break}}// bf使命完成,放入池中bufPool.Put(bf)}()}}

0 1
原创粉丝点击