feat(nodebuilder/p2p): If node is a bootstrapper, automatically initiate connection to other hardcoded bootstrappers for the given network (#4182)

Author: renaynay

nodebuilder/p2p/bootstrap.go

@@ -1,9 +1,12 @@
 package p2p
 
 import (
+	"context"
 	"os"
 	"strconv"
+	"sync"
 
+	"github.com/libp2p/go-libp2p/core/host"
 	"github.com/libp2p/go-libp2p/core/peer"
 	ma "github.com/multiformats/go-multiaddr"
 )
@@ -82,3 +85,37 @@ func parseAddrInfos(addrs []string) ([]peer.AddrInfo, error) {
 
 	return infos, nil
 }
+
+// connectToBootstrappers ensures that the bootstrapper node
+// initiates a connection to other hardcoded bootstrap peers
+// while the connectionManager hook adds them as mutual peers to prevent
+// trimming the connection. This will aid the network's connectivity.
+func connectToBootstrappers(ctx context.Context, h host.Host, network Network) error {
+	if !isBootstrapper() {
+		return nil
+	}
+
+	boots, err := BootstrappersFor(network)
+	if err != nil {
+		return err
+	}
+
+	wg := sync.WaitGroup{}
+	wg.Add(len(boots))
+	for _, b := range boots {
+		go func() {
+			defer wg.Done()
+
+			err := h.Connect(ctx, b)
+			if err != nil {
+				log.Errorw("bootstrap: failed to connect to bootstrapper", "id", b, "err", err)
+				return
+			}
+
+			log.Infow("bootstrap: successfully connected to bootstrapper", "id", b.String())
+		}()
+	}
+	wg.Wait()
+
+	return nil
+}

nodebuilder/p2p/module.go

@@ -42,6 +42,7 @@ func ConstructModule(tp node.Type, cfg *Config) fx.Option {
 			baseComponents,
 			fx.Provide(infiniteResources),
 			fx.Invoke(reachabilityCheck),
+			fx.Invoke(connectToBootstrappers),
 		)
 	case node.Light:
 		return fx.Module(