If address list less than 1000 send getaddr req

If address list less than 1000 send getaddr request

Signed-off-by: Jin Qing <[email protected]>

net/message/verack.go

@@ -69,7 +69,9 @@ func (msg verACK) Handle(node Noder) error {
 	// Fixme, there is a race condition here,
 	// but it doesn't matter to access the invalid
 	// node which will trigger a warning
-	node.ReqNeighborList()
+	if node.LocalNode().NeedMoreAddresses() {
+		node.ReqNeighborList()
+	}
 	addr := node.GetAddr()
 	port := node.GetPort()
 	nodeAddr := addr + ":" + strconv.Itoa(int(port))

net/node/infoUpdate.go

@@ -136,7 +136,9 @@ func (node *node) ConnectSeeds() {
 			node.nbrNodes.Unlock()
 			if found {
 				if n.GetState() == ESTABLISH {
-					n.ReqNeighborList()
+					if node.LocalNode().NeedMoreAddresses() {
+						n.ReqNeighborList()
+					}
 				}
 			} else { //not found
 				go node.Connect(nodeAddr)

net/node/knowaddrlist.go

@@ -7,6 +7,12 @@ import (
 	"sync"
 )
 
+const (
+	// needAddressThreshold is the number of addresses under which the
+	// address manager will claim to need more addresses.
+	needAddressThreshold = 1000
+)
+
 type KnownAddress struct {
 	srcAddr NodeAddr
 }
@@ -33,6 +39,13 @@ func (ka *KnownAddress) GetID() uint64 {
 	return ka.srcAddr.ID
 }
 
+func (al *KnownAddressList) NeedMoreAddresses() bool {
+	al.Lock()
+	defer al.Unlock()
+
+	return al.addrCount < needAddressThreshold
+}
+
 func (al *KnownAddressList) AddressExisted(uid uint64) bool {
 	_, ok := al.List[uid]
 	return ok

net/protocol/protocol.go

@@ -148,6 +148,7 @@ type Noder interface {
 	GetDefaultMaxPeers() uint
 	GetMaxOutboundCnt() uint
 	GetGetAddrMax() uint
+	NeedMoreAddresses() bool
 }
 
 func (msg *NodeAddr) Deserialization(p []byte) error {