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

Fix invalid usage for sync.WaitGroup (#1752)

Browse Source 
* Custom private blocks

* Fix invalid usage for sync.WaitGroup

Co-authored-by: Asim Aslam <asim@aslam.me>
This commit is contained in:
Di Wu
2020-06-28 03:07:04 +08:00
committed by Dominic Wong
parent aef6878ee0
commit 3e6ac73cfe
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
server/rpc_server.go
View File

@@ -363,11 +363,12 @@ func (s *rpcServer) ServeConn(sock transport.Socket) {
r = rpcRouter{h: handler} r = rpcRouter{h: handler}
} }
// wait for two coroutines to exit
// serve the request and process the outbound messages
wg.Add(2)
// process the outbound messages from the socket // process the outbound messages from the socket
go func(id string, psock *socket.Socket) { go func(id string, psock *socket.Socket) {
// wait for processing to exit
wg.Add(1)
defer func() { defer func() {
// TODO: don't hack this but if its grpc just break out of the stream // TODO: don't hack this but if its grpc just break out of the stream
// We do this because the underlying connection is h2 and its a stream // We do this because the underlying connection is h2 and its a stream
@@ -405,9 +406,6 @@ func (s *rpcServer) ServeConn(sock transport.Socket) {
// serve the request in a go routine as this may be a stream // serve the request in a go routine as this may be a stream
go func(id string, psock *socket.Socket) { go func(id string, psock *socket.Socket) {
// add to the waitgroup
wg.Add(1)
defer func() { defer func() {
// release the socket // release the socket
pool.Release(psock) pool.Release(psock)