添加 Close() 方法停止轮询

client.go

@@ -75,13 +75,15 @@ type Client interface {
 	RemoveChangeListener(listener storage.ChangeListener)
 	GetChangeListeners() *list.List
 	UseEventDispatch()
+	Close()
 }
 
 // internalClient apollo 客户端实例
 type internalClient struct {
 	initAppConfigFunc func() (*config.AppConfig, error)
 	appConfig         *config.AppConfig
 	cache             *storage.Cache
+	configComponent   *notify.ConfigComponent
 }
 
 func (c *internalClient) getAppConfig() config.AppConfig {
@@ -135,6 +137,7 @@ func StartWithConfig(loadAppConfig func() (*config.AppConfig, error)) (Client, e
 	configComponent.SetAppConfig(c.getAppConfig)
 	configComponent.SetCache(c.cache)
 	go component.StartRefreshConfig(configComponent)
+	c.configComponent = configComponent
 
 	log.Info("agollo start finished ! ")
 
@@ -260,3 +263,8 @@ func (c *internalClient) GetChangeListeners() *list.List {
 func (c *internalClient) UseEventDispatch() {
 	c.AddChangeListener(storage.UseEventDispatch())
 }
+
+// Close 停止轮询
+func (c *internalClient) Close() {
+	c.configComponent.Stop()
+}

component/notify/componet_notify.go

@@ -34,6 +34,7 @@ const (
 type ConfigComponent struct {
 	appConfigFunc func() config.AppConfig
 	cache         *storage.Cache
+	stopCh        chan interface{}
 }
 
 // SetAppConfig nolint
@@ -48,9 +49,14 @@ func (c *ConfigComponent) SetCache(cache *storage.Cache) {
 
 //Start 启动配置组件定时器
 func (c *ConfigComponent) Start() {
+	if c.stopCh == nil {
+		c.stopCh = make(chan interface{})
+	}
+
 	t2 := time.NewTimer(longPollInterval)
 	instance := remote.CreateAsyncApolloConfig()
 	//long poll for sync
+loop:
 	for {
 		select {
 		case <-t2.C:
@@ -59,6 +65,15 @@ func (c *ConfigComponent) Start() {
 				c.cache.UpdateApolloConfig(apolloConfig, c.appConfigFunc)
 			}
 			t2.Reset(longPollInterval)
+		case <-c.stopCh:
+			break loop
 		}
 	}
 }
+
+// Stop 停止配置组件定时器
+func (c *ConfigComponent) Stop() {
+	if c.stopCh != nil {
+		close(c.stopCh)
+	}
+}