clientv3/leasing: fix racey waitSession

Signed-off-by: Gyuho Lee <gyuhox@gmail.com>
2017-12-21 15:57:38 -08:00 · 2017-12-21 15:57:38 -08:00 · 1880cf8da2
parent 3dd1c1b53c
commit 1880cf8da2
1 changed files with 4 additions and 1 deletions
--- a/clientv3/leasing/kv.go
+++ b/clientv3/leasing/kv.go
@ -445,8 +445,11 @@ func (lkv *leasingKV) revokeLeaseKvs(ctx context.Context, kvs []*mvccpb.KeyValue
 }

 func (lkv *leasingKV) waitSession(ctx context.Context) error {
+	lkv.leases.mu.RLock()
+	sessionc := lkv.sessionc
+	lkv.leases.mu.RUnlock()
 	select {
-	case <-lkv.sessionc:
+	case <-sessionc:
 		return nil
 	case <-lkv.ctx.Done():
 		return lkv.ctx.Err()