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

fix bug #239

Browse Source
This commit is contained in:
leon.peng
2018-03-21 03:17:38 +00:00
parent d486125d07
commit 9b99d50396
2 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
client/rpc_client.go
View File

@@ -13,12 +13,14 @@ import (
"github.com/micro/go-micro/metadata" "github.com/micro/go-micro/metadata"
"github.com/micro/go-micro/selector" "github.com/micro/go-micro/selector"
"github.com/micro/go-micro/transport" "github.com/micro/go-micro/transport"
"sync/atomic"
) )
type rpcClient struct { type rpcClient struct {
once sync.Once once sync.Once
opts Options opts Options
pool *pool pool *pool
seq uint64
} }
func newRpcClient(opt ...Option) Client { func newRpcClient(opt ...Option) Client {
@@ -28,6 +30,7 @@ func newRpcClient(opt ...Option) Client {
once: sync.Once{}, once: sync.Once{},
opts: opts, opts: opts,
pool: newPool(opts.PoolSize, opts.PoolTTL), pool: newPool(opts.PoolSize, opts.PoolTTL),
seq: 0,
} }
c := Client(rc) c := Client(rc)
@@ -84,11 +87,15 @@ func (r *rpcClient) call(ctx context.Context, address string, req Request, resp
r.pool.release(address, c, grr) r.pool.release(address, c, grr)
}() }()
seq := r.seq
atomic.AddUint64(&r.seq, 1)
stream := &rpcStream{ stream := &rpcStream{
context: ctx, context: ctx,
request: req, request: req,
closed: make(chan bool), closed: make(chan bool),
codec: newRpcPlusCodec(msg, c, cf), codec: newRpcPlusCodec(msg, c, cf),
seq: seq,
} }
defer stream.Close() defer stream.Close()

1
client/rpc_stream.go
View File

@@ -44,7 +44,6 @@ func (r *rpcStream) Send(msg interface{}) error {
} }
seq := r.seq seq := r.seq
r.seq++
req := request{ req := request{
Service: r.request.Service(), Service: r.request.Service(),