安裝:
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()
}