Function BenchmarkSessionPool_Get

rpc/mux/pool_test.go:68–104

(b *testing.B)

}

func BenchmarkSessionPool_Get(b *testing.B) {
	Convey("session pool", b, func(c C) {
		log.SetLevel(log.FatalLevel)
		p := &SessionPool{
			sessions: make(map[proto.NodeID]*Session),
		}
		defer withTCPDialer()()
		defer func() { _ = p.Close() }()

		l, err := net.Listen("tcp", ":0")
		if err != nil {
			log.Fatal(err)
		}
		server, err := NewServerWithService(ServiceMap{"Test": NewTestService()})
		server.SetListener(l)
		go server.WithAcceptConnFunc(rpc.AcceptRawConn).Serve()
		wg := &sync.WaitGroup{}
		b.ResetTimer()

		wg.Add(concurrency)
		for i := 0; i < concurrency; i++ {
			go func(c2 C, n int) {
				defer wg.Done()
				client, err := p.Get(proto.NodeID(l.Addr().String()))
				defer func() { _ = client.Close() }()
				if err != nil {
					log.Errorf("get session failed: %s", err)
					return
				}
				c2.So(err, ShouldBeNil)
				err = client.Call("Test.IncCounter", &TestReq{Step: 1}, &TestRep{})
				c2.So(err, ShouldBeNil)
			}(c, b.N)
		}
		wg.Wait()
	})
}

func TestNewSessionPool(t *testing.T) {
	Convey("session pool", t, func(c C) {