func NewFlumePoolLink(hp config.HostPort) (error, *FlumePoolLink) {
err, pool := newFlumeClientPool(hp, 10, 30, 50, 10*time.Second, func() (error, *client.FlumeClient) {
flumeclient := client.NewFlumeClient(hp.Host, hp.Port)
err := flumeclient.Connect()
return err, flumeclient
})
//将此pool封装为Link
return err, &FlumePoolLink{FlumePool: pool, BusinessLink: list.New()}
}
func Test_Pool(t *testing.T) {
errs, clientPool := newFlumeClientPool(config.NewHostPort("localhost:44444"),
10, 20, 30, 10*time.Second, func() (error, *client.FlumeClient) {
flumeclient := client.NewFlumeClient("localhost", 44444)
err := flumeclient.Connect()
return err, flumeclient
})
idlecount := clientPool.CorePoolSize()
log.Println(errs)
//默认初始化10个最小连接
if idlecount != 10 {
t.Fail()
}
var err error
checkout := list.New()
for i := 0; i < 45; i++ {
fclient, perr := clientPool.Get(5 * time.Second)
if nil == perr && nil != fclient {
checkout.PushFront(fclient)
}
err = perr
}
if nil == err {
//达到最大的连接数后应该会报错
t.Fail()
}
log.Printf("------------get |active:%d,core:%d,max:%d", clientPool.ActivePoolSize(), clientPool.CorePoolSize(), clientPool.maxPoolSize)
//如果活动线程数等于现在持有的则成功，反则失败
if clientPool.ActivePoolSize() != checkout.Len() {
t.Logf("client pool |active ne checkout|%d,%d", clientPool.ActivePoolSize(), checkout.Len())
t.Fail()
}
if clientPool.CorePoolSize() != clientPool.maxPoolSize {
t.Logf("client pool |active ne checkout|%d,%d", clientPool.CorePoolSize(), checkout.Len())
t.Fail()
}
func() {
for e := checkout.Front(); nil != e; e = e.Next() {
clientPool.Release(e.Value.(*client.FlumeClient))
}
}()
log.Printf("------------release conn |active:%d,core:%d,max:%d", clientPool.ActivePoolSize(), clientPool.CorePoolSize(), clientPool.maxPoolSize)
for i := 0; i < 20; i++ {
client, err := clientPool.Get(5 * time.Second)
if nil != err {
log.Println(err)
} else {
time.Sleep(2 * time.Second)
clientPool.Release(client)
log.Printf("test release checkout client|active:%d,core:%d,max:%d", clientPool.ActivePoolSize(), clientPool.CorePoolSize(), clientPool.maxPoolSize)
}
}
time.Sleep(5 * time.Second)
log.Printf("end ------------active:%d,core:%d,max:%d", clientPool.ActivePoolSize(), clientPool.CorePoolSize(), clientPool.maxPoolSize)
clientPool.Destroy()
}