Configure libp2p gossipsub WithDirectPeers

xmtp · Aug 30, 2023 · a7335ac · a7335ac
1 parent b2e6089
commit a7335ac
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 6 deletions.
diff --git a/pkg/authn/authn.pb.go b/pkg/authn/authn.pb.go
diff --git a/pkg/server/server.go b/pkg/server/server.go
@@ -138,7 +138,23 @@ func New(ctx context.Context, log *zap.Logger, options Options) (*Server, error)
 
 	if !options.Relay.Disable {
 		var wakurelayopts []pubsub.Option
-		wakurelayopts = append(wakurelayopts, pubsub.WithPeerExchange(true))
+		directPeers := make([]peer.AddrInfo, 0, len(options.StaticNodes))
+		for _, staticNode := range options.StaticNodes {
+			ma, err := multiaddr.NewMultiaddr(staticNode)
+			if err != nil {
+				s.log.Error("building multiaddr from static node addr", zap.Error(err))
+			}
+			pi, err := peer.AddrInfoFromP2pAddr(ma)
+			if err != nil {
+				s.log.Error("getting peer addr info from static node addr", zap.Error(err))
+			}
+			if pi != nil {
+				s.log.Error("static node peer addr is nil", zap.String("peer", staticNode))
+			} else {
+				directPeers = append(directPeers, *pi)
+			}
+		}
+		wakurelayopts = append(wakurelayopts, pubsub.WithPeerExchange(true), pubsub.WithDirectPeers(directPeers))
 		nodeOpts = append(nodeOpts, node.WithWakuRelayAndMinPeers(options.Relay.MinRelayPeersToPublish, wakurelayopts...))
 	}