initialize the resource manager

Author: marten-seemann

core/node/groups.go

@@ -112,10 +112,10 @@ func LibP2P(bcfg *BuildCfg, cfg *config.Config) fx.Option {
 	}
 
 	// If `cfg.Swarm.DisableRelay` is set and `Network.RelayTransport` isn't, use the former.
-	enableRelayTransport := cfg.Swarm.Transports.Network.Relay.WithDefault(!cfg.Swarm.DisableRelay) //nolint
+	enableRelayTransport := cfg.Swarm.Transports.Network.Relay.WithDefault(!cfg.Swarm.DisableRelay) // nolint
 
 	// Warn about a deprecated option.
-	//nolint
+	// nolint
 	if cfg.Swarm.DisableRelay {
 		logger.Error("The 'Swarm.DisableRelay' config field is deprecated.")
 		if enableRelayTransport {
@@ -124,7 +124,7 @@ func LibP2P(bcfg *BuildCfg, cfg *config.Config) fx.Option {
 			logger.Error("Use the 'Swarm.Transports.Network.Relay' config field instead")
 		}
 	}
-	//nolint
+	// nolint
 	if cfg.Swarm.EnableAutoRelay {
 		logger.Error("The 'Swarm.EnableAutoRelay' config field is deprecated.")
 		if cfg.Swarm.RelayClient.Enabled == config.Default {
@@ -133,7 +133,7 @@ func LibP2P(bcfg *BuildCfg, cfg *config.Config) fx.Option {
 			logger.Error("'Swarm.EnableAutoRelay' has been overridden by 'Swarm.AutoRelay.Enabled'")
 		}
 	}
-	//nolint
+	// nolint
 	if cfg.Swarm.EnableRelayHop {
 		logger.Fatal("The `Swarm.EnableRelayHop` config field is ignored.\n" +
 			"Use `Swarm.RelayService` to configure the circuit v2 relay.\n" +
@@ -144,6 +144,7 @@ func LibP2P(bcfg *BuildCfg, cfg *config.Config) fx.Option {
 	opts := fx.Options(
 		BaseLibP2P,
 
+		fx.Provide(libp2p.ResourceManager()),
 		fx.Provide(libp2p.AddrFilters(cfg.Swarm.AddrFilters)),
 		fx.Provide(libp2p.AddrsFactory(cfg.Addresses.Announce, cfg.Addresses.AppendAnnounce, cfg.Addresses.NoAnnounce)),
 		fx.Provide(libp2p.SmuxTransport(cfg.Swarm.Transports)),

core/node/libp2p/rcmgr.go

@@ -0,0 +1,40 @@
+package libp2p
+
+import (
+	"errors"
+	"fmt"
+	"os"
+
+	"github.com/libp2p/go-libp2p"
+	rcmgr "github.com/libp2p/go-libp2p-resource-manager"
+)
+
+func ResourceManager() func() (Libp2pOpts, error) {
+	return func() (opts Libp2pOpts, err error) {
+		var limiter *rcmgr.BasicLimiter
+
+		limitsIn, err := os.Open("./limits.json")
+		switch {
+		case err == nil:
+			defer limitsIn.Close()
+			limiter, err = rcmgr.NewDefaultLimiterFromJSON(limitsIn)
+			if err != nil {
+				return opts, fmt.Errorf("error parsing limit file: %w", err)
+			}
+		case errors.Is(err, os.ErrNotExist):
+			limiter = rcmgr.NewDefaultLimiter()
+		default:
+			return opts, err
+		}
+
+		libp2p.SetDefaultServiceLimits(limiter)
+
+		// TODO: close the resource manager when the node is shut down
+		rcmgr, err := rcmgr.NewResourceManager(limiter)
+		if err != nil {
+			return opts, fmt.Errorf("error creating resource manager: %w", err)
+		}
+		opts.Opts = append(opts.Opts, libp2p.ResourceManager(rcmgr))
+		return opts, nil
+	}
+}

go.mod

@@ -81,6 +81,7 @@ require (
 	github.com/libp2p/go-libp2p-pubsub-router v0.5.0
 	github.com/libp2p/go-libp2p-quic-transport v0.16.1
 	github.com/libp2p/go-libp2p-record v0.1.3
+	github.com/libp2p/go-libp2p-resource-manager v0.1.3
 	github.com/libp2p/go-libp2p-routing-helpers v0.2.3
 	github.com/libp2p/go-libp2p-swarm v0.10.1
 	github.com/libp2p/go-libp2p-testing v0.7.0