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

erlib · · 792 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

本来想写一个例子来给团队的人阐述一下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)
		}()
	}
}


有疑问加站长微信联系(非本文作者)

本文来自:CSDN博客

感谢作者:erlib

查看原文:一个简单的例子阐述Go的对象复用思想

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

792 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传