unistack-org/micro
4
0
Fork 3
Code Issues 17 Pull Requests 1 Actions Packages Projects Releases 258 Wiki Activity

We don't need sync.Once, we can just select

Browse Source
This commit is contained in:
Asim 2016-05-11 00:01:32 +01:00
parent a0a4ab4c15
commit 2254578760
2 changed files with 9 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
client/rpc_client.go
View File

@ -75,11 +75,9 @@ func (r *rpcClient) call(ctx context.Context, address string, req Request, resp
return errors.InternalServerError("go.micro.client", fmt.Sprintf("Error sending request: %v", err))
}
var once sync.Once
stream := &rpcStream{
context: ctx,
request: req,
once: once,
closed: make(chan bool),
codec: newRpcPlusCodec(msg, c, cf),
}
@ -139,11 +137,9 @@ func (r *rpcClient) stream(ctx context.Context, address string, req Request, opt
return nil, errors.InternalServerError("go.micro.client", fmt.Sprintf("Error sending request: %v", err))
}
var once sync.Once
stream := &rpcStream{
context: ctx,
request: req,
once: once,
closed: make(chan bool),
codec: newRpcPlusCodec(msg, c, cf),
}

14
client/rpc_stream.go
View File

@ -12,7 +12,6 @@ import (
type rpcStream struct {
sync.RWMutex
seq uint64
once sync.Once
closed chan bool
err error
request Request
@ -22,14 +21,12 @@ type rpcStream struct {
func (r *rpcStream) isClosed() bool {
select {
case _, ok := <-r.closed:
if !ok {
case <-r.closed:
return true
}
default:
}
return false
}
}
func (r *rpcStream) Context() context.Context {
return r.context
@ -112,8 +109,11 @@ func (r *rpcStream) Error() error {
}
func (r *rpcStream) Close() error {
r.once.Do(func() {
select {
case <-r.closed:
return nil
default:
close(r.closed)
})
return r.codec.Close()
}
}