fix(discovery):Fix the protected data race

discovery/discovery.go

@@ -2,6 +2,7 @@ package discovery
 
 import (
 	"context"
+	"sync"
 	"sync/atomic"
 	"time"
 
@@ -17,6 +18,7 @@ type Discovery struct {
 	client    *http.Client
 	registry  *registry.Registry
 	nodes     atomic.Value
+	lock      sync.RWMutex
 }
 
 // New get a discovery.
@@ -38,5 +40,7 @@ func New(c *conf.Config) (d *Discovery, cancel context.CancelFunc) {
 func (d *Discovery) exitProtect() {
 	// exist protect mode after two renew cycle
 	time.Sleep(time.Second * 60)
+	d.lock.Lock()
 	d.protected = false
+	d.lock.Unlock()
 }

discovery/syncup.go

@@ -21,7 +21,10 @@ var (
 // if service in init protect mode,only support write,
 // read operator isn't supported.
 func (d *Discovery) Protected() bool {
-	return d.protected
+	d.lock.RLock()
+	protected := d.protected
+	d.lock.RUnlock()
+	return protected
 }
 
 // syncUp populates the registry information from a peer eureka node.
@@ -45,7 +48,9 @@ func (d *Discovery) syncUp() {
 			continue
 		}
 		// sync success from other node,exit protected mode
+		d.lock.Lock()
 		d.protected = false
+		d.lock.Unlock()
 		for _, is := range res.Data {
 			for _, i := range is {
 				_ = d.registry.Register(i, i.LatestTimestamp)