Added WithRequireLeader() to etcd v3 watch

vulcand · Aug 23, 2018 · 8b4dad1 · 8b4dad1
1 parent 8d1f2fd
commit 8b4dad1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 26 deletions.
diff --git a/Gopkg.lock b/Gopkg.lock
diff --git a/engine/etcdng/v3/etcd.go b/engine/etcdng/v3/etcd.go
@@ -598,7 +598,8 @@ func (n *ng) Subscribe(changes chan interface{}, afterIdx uint64, cancelC chan s
 	defer watcher.Close()
 
 	log.Infof("begin watching: etcd revision %d", afterIdx)
-	watchChan := watcher.Watch(n.context, n.etcdKey, etcd.WithRev(int64(afterIdx)), etcd.WithPrefix())
+	watchChan := watcher.Watch(etcd.WithRequireLeader(n.context),
+		n.etcdKey, etcd.WithRev(int64(afterIdx)), etcd.WithPrefix())
 
 	for response := range watchChan {
 		if response.Canceled {