go語(yǔ)言TCP連接池rocket049/connpool使用

安裝:
go get -v -u github.com/rocket049/connpool
go get -v -u gitee.com/rocket049/connpool

rocket049/connpool包是本人用go語(yǔ)言開(kāi)發(fā)的,提供一個(gè)通用的TCP連接池,初始化參數(shù)包括最高連接數(shù)、超時(shí)秒數(shù)、連接函數(shù),放回連接池的連接被重新取出時(shí),如果已經(jīng)超時(shí),將會(huì)自動(dòng)重新連接;如果沒(méi)有超時(shí),連接將被復(fù)用。

可調(diào)用的函數(shù):

type Conn
    func (s *Conn) Read(p []byte) (int, error)
    func (s *Conn) Write(p []byte) (int, error)
    func (s *Conn) Close() error
    func (s *Conn) Timeout() bool
type Pool
    func NewPool(max, timeout int, factory func() (net.Conn, error)) *Pool
    func (s *Pool) Get() (*Conn, error)
    func (s *Pool) Put(conn1 *Conn)
    func (s *Pool) Close()

調(diào)用示例:

import "gitee.com/rocket049/connpool"

func factory() (net.Conn,error) {
    return net.Dial("tcp","127.0.0.1:7060")
}

func UsePool() {
    // 設(shè)置連接池大小為10,超時(shí)秒數(shù)為30,連接函數(shù)為 factory
    pool1 := connpool.NewPool(10, 30 ,factory)
    defer pool1.Close()
    var wg sync.WaitGroup
    for i:=0;i<50;i++ {
        wg.Add(1)
        go func(n int){
            // 取出連接
            conn ,err := pool1.Get()
            if err!=nil {
                ...
            }
            //發(fā)送數(shù)據(jù)
            _,err = conn.Write( msg )
            if err!=nil{
                ...
            }
            //讀取數(shù)據(jù)
            n1,err := conn.Read( buf )
            if err!=nil{
                ...
            }
            //超時(shí)檢測(cè)和重連
            if conn.Timeout() {
                pool1.Put(conn)
                conn ,err := pool1.Get()
                ...
            }
            //放回連接池
            pool1.Put(conn)
            wg.Done()
        }(i)
    }
    wg.Wait()

}
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書(shū)系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

友情鏈接更多精彩內(nèi)容