opensource/grpc-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

polish Balancer comments

Browse Source
This commit is contained in:
iamqizhao
2016-05-16 16:36:40 -07:00
parent 73ff8375a6
commit 5710bedd70
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
balancer.go
View File

@ -53,8 +53,9 @@ type Address struct {
// Balancer chooses network addresses for RPCs. // Balancer chooses network addresses for RPCs.
type Balancer interface { type Balancer interface {
// Up informs the balancer that gRPC has a connection to the server at // Up informs the balancer that gRPC has a connection to the server at
// addr. It returns down which will be called once the connection gets // addr. It returns down which is called once the connection to addr gets
// lost. Once down is called, addr may no longer be returned by Get. // lost or closed. Once down is called, addr may no longer be returned
// by Get.
Up(addr Address) (down func(error)) Up(addr Address) (down func(error))
// Get gets the address of a server for the rpc corresponding to ctx. // Get gets the address of a server for the rpc corresponding to ctx.
// It may block if there is no server available. It respects the // It may block if there is no server available. It respects the
@ -91,6 +92,7 @@ func (rr *roundRobin) Up(addr Address) func(error) {
} }
rr.addrs = append(rr.addrs, addr) rr.addrs = append(rr.addrs, addr)
if len(rr.addrs) == 1 { if len(rr.addrs) == 1 {
// addr is only one available. Notify the Get() callers who are blocking.
if rr.waitCh != nil { if rr.waitCh != nil {
close(rr.waitCh) close(rr.waitCh)
rr.waitCh = nil rr.waitCh = nil