Mirroristas/etcd
2
0
Fork 0
mirror of https://github.com/etcd-io/etcd.git synced 2024-09-27 06:25:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

grpcproxy: use new lease interface

Browse Source
This commit is contained in:
Anthony Romano 2017-04-17 15:10:25 -07:00
parent 70f0bbe38c
commit b5384ac1c0
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
proxy/grpcproxy/lease.go
View File

@ -255,10 +255,7 @@ func (lps *leaseProxyStream) recvLoop() error {
func (lps *leaseProxyStream) keepAliveLoop(leaseID int64, neededResps *atomicCounter) error { func (lps *leaseProxyStream) keepAliveLoop(leaseID int64, neededResps *atomicCounter) error {
cctx, ccancel := context.WithCancel(lps.ctx) cctx, ccancel := context.WithCancel(lps.ctx)
defer ccancel() defer ccancel()
respc, err := lps.lessor.KeepAlive(cctx, clientv3.LeaseID(leaseID)) respc := lps.lessor.KeepAlive(cctx, clientv3.LeaseID(leaseID))
if err != nil {
return err
}
// ticker expires when loop hasn't received keepalive within TTL // ticker expires when loop hasn't received keepalive within TTL
var ticker <-chan time.Time var ticker <-chan time.Time
for { for {
@ -276,7 +273,7 @@ func (lps *leaseProxyStream) keepAliveLoop(leaseID int64, neededResps *atomicCou
lps.mu.Unlock() lps.mu.Unlock()
return nil return nil
case rp, ok := <-respc: case rp, ok := <-respc:
if !ok { if !ok || rp.Err != nil {
lps.mu.Lock() lps.mu.Lock()
delete(lps.keepAliveLeases, leaseID) delete(lps.keepAliveLeases, leaseID)
lps.mu.Unlock() lps.mu.Unlock()