fix: netsize warning (#1077)

Author: guillaumemichel

lookup.go

@@ -7,6 +7,7 @@ import (
 
 	"github.com/libp2p/go-libp2p-kad-dht/internal"
 	"github.com/libp2p/go-libp2p-kad-dht/internal/metrics"
+	"github.com/libp2p/go-libp2p-kad-dht/netsize"
 	"github.com/libp2p/go-libp2p-kad-dht/qpeerset"
 	kb "github.com/libp2p/go-libp2p-kbucket"
 	"github.com/libp2p/go-libp2p/core/peer"
@@ -39,7 +40,11 @@ func (dht *IpfsDHT) GetClosestPeers(ctx context.Context, key string) ([]peer.ID,
 
 	// tracking lookup results for network size estimator
 	if err = dht.nsEstimator.Track(key, lookupRes.closest); err != nil {
-		logger.Warnf("network size estimator track peers: %s", err)
+		if err != netsize.ErrWrongNumOfPeers || dht.routingTable.Size() > dht.bucketSize {
+			// Don't warn if we have a wrong number of peers and the routing table is
+			// small because the network may simply not have enough peers.
+			logger.Warnf("network size estimator track peers: %s", err)
+		}
 	}
 
 	if ns, err := dht.nsEstimator.NetworkSize(); err == nil {

lookup_optim.go

@@ -169,7 +169,11 @@ func (dht *IpfsDHT) optimisticProvide(outerCtx context.Context, keyMH multihash.
 
 	// tracking lookup results for network size estimator as "completed" is true
 	if err = dht.nsEstimator.Track(key, lookupRes.closest); err != nil {
-		logger.Warnf("network size estimator track peers: %s", err)
+		if err != netsize.ErrWrongNumOfPeers || dht.routingTable.Size() > dht.bucketSize {
+			// Don't warn if we have a wrong number of peers and the routing table is
+			// small because the network may simply not have enough peers.
+			logger.Warnf("network size estimator track peers: %s", err)
+		}
 	}
 
 	if ns, err := dht.nsEstimator.NetworkSize(); err == nil {