Add watch reset

deepfabric · Jun 5, 2020 · 0c0aa24 · 0c0aa24
1 parent b96ed25
commit 0c0aa24
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/watcher_client.go b/watcher_client.go
@@ -64,6 +64,14 @@ func (w *Watcher) Stop() {
 	close(w.eventC)
 }
 
+// Reset reset and refresh
+func (w *Watcher) Reset() {
+	w.Lock()
+	defer w.Unlock()
+
+	w.conn.Close()
+}
+
 func (w *Watcher) resetConn(flag int) {
 	for {
 		addr := w.next()
@@ -112,6 +120,12 @@ func (w *Watcher) resetConnWithProphet(flag int) {
 }
 
 func (w *Watcher) watchDog(flag int) {
+	defer func() {
+		if r := recover(); r != nil {
+			go w.watchDog(flag)
+		}
+	}()
+
 	for {
 		select {
 		case <-w.stopC: