IT博客汇
  • 首页
  • 精华
  • 技术
  • 设计
  • 资讯
  • 扯淡
  • 权利声明
  • 登录 注册

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

    abv123456789发表于 2016-04-14 17:40:18
    love 0

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



沪ICP备19023445号-2号
友情链接