Esempio n. 1
0
func NewFlumePoolLink(hp config.HostPort) (error, *FlumePoolLink) {
	err, pool := newFlumeClientPool(hp, 10, 10, 20, 10*time.Minute, 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()}
}
Esempio n. 2
0
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()

}