modify async response with buffered channel

baidu-golang · Aug 12, 2021 · 0a81ec1 · 0a81ec1
1 parent 7e6983c
commit 0a81ec1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/client.go b/client.go
@@ -309,7 +309,7 @@ func (c *RpcClient) SendRpcRequestWithTimeout(timeout time.Duration, rpcInvocati
 
 	var rsp *RpcDataPackage
 	if c.asyncMode {
-		ch := make(chan *RpcDataPackage)
+		ch := make(chan *RpcDataPackage, 1)
 		c.requestCallState[correlationId] = ch
 
 		if timeout > 0 {
@@ -321,7 +321,7 @@ func (c *RpcClient) SendRpcRequestWithTimeout(timeout time.Duration, rpcInvocati
 
 	} else {
 		if timeout > 0 {
-			ch := make(chan *RpcDataPackage)
+			ch := make(chan *RpcDataPackage, 1)
 			go c.asyncRequest(timeout, rpcDataPackage, ch)
 			defer close(ch)
 			// wait for message