From 29dee0db28cbe7503100df0f3b6947ebfeeee6dc Mon Sep 17 00:00:00 2001
From: cvictory <shenglicao2@gmail.com>
Date: Tue, 13 Oct 2020 22:38:03 +0800
Subject: [PATCH] enhance the refresh func

---
 registry/mock_registry.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/registry/mock_registry.go b/registry/mock_registry.go
index d84b52945..d10755396 100644
--- a/registry/mock_registry.go
+++ b/registry/mock_registry.go
@@ -18,6 +18,7 @@
 package registry
 
 import (
+	"fmt"
 	"time"
 )
 
@@ -113,7 +114,9 @@ func (r *MockRegistry) Subscribe(url *common.URL, notifyListener NotifyListener)
 			for {
 				select {
 				case e := <-r.allAddress:
-					notifyListener.NotifyAll(e)
+					notifyListener.NotifyAll(e, func() {
+						fmt.Print("notify all ok")
+					})
 					break
 				}
 			}
-- 
GitLab