unistack-org/micro
4
0
Fork 2
Code Issues 18 Pull Requests Actions Packages Projects Releases 331 Wiki Activity

support streams pool for grpc (#1062)

Browse Source 
* Update grpc_pool.go

* Update options.go

* Update grpc.go

* Update grpc_pool_test.go

* streams pool for grpc

* use busy list to speed up allocate while pool is very busy

* fix idle bug
This commit is contained in:
jamsonzan
2019-12-27 20:25:58 +08:00
committed by Asim Aslam
parent a09b6729cc
commit 22aa7d14b3
4 changed files with 199 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
client/grpc/grpc_pool_test.go
View File

@@ -11,7 +11,7 @@ import (
pb "google.golang.org/grpc/examples/helloworld/helloworld"
)
func testPool(t *testing.T, size int, ttl time.Duration) {
func testPool(t *testing.T, size int, ttl time.Duration, idle int, ms int) {
// setup server
l, err := net.Listen("tcp", ":0")
if err != nil {
@@ -26,7 +26,7 @@ func testPool(t *testing.T, size int, ttl time.Duration) {
defer s.Stop()
// zero pool
p := newPool(size, ttl)
p := newPool(size, ttl, idle, ms)
for i := 0; i < 10; i++ {
// get a conn
@@ -50,7 +50,7 @@ func testPool(t *testing.T, size int, ttl time.Duration) {
p.release(l.Addr().String(), cc, nil)
p.Lock()
if i := len(p.conns[l.Addr().String()]); i > size {
if i := p.conns[l.Addr().String()].count; i > size {
p.Unlock()
t.Fatalf("pool size %d is greater than expected %d", i, size)
}
@@ -59,6 +59,6 @@ func testPool(t *testing.T, size int, ttl time.Duration) {
}
func TestGRPCPool(t *testing.T) {
testPool(t, 0, time.Minute)
testPool(t, 2, time.Minute)
testPool(t, 0, time.Minute, 10, 2)
testPool(t, 2, time.Minute, 10, 1)
}