feat: add resource manager for separate DHT libp2p host

ipfs · Dec 22, 2023 · 9ac2e2b · 9ac2e2b
1 parent 5067f27
commit 9ac2e2b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/setup.go b/setup.go
@@ -199,11 +199,18 @@ func Setup(ctx context.Context, cfg Config, key crypto.PrivKey, dnsCache *cached
 			if cfg.DHTSharedHost {
 				dhtHost = h
 			} else {
+				dhtLimiter := rcmgr.NewFixedLimiter(makeResourceManagerConfig(cfg.MaxMemory, cfg.MaxFD, cfg.ConnMgrHi))
+				dhtMgr, err := rcmgr.NewResourceManager(dhtLimiter)
+				if err != nil {
+					return nil, err
+				}
+
 				dhtHost, err = libp2p.New(
 					libp2p.NoListenAddrs,
 					libp2p.BandwidthReporter(bwc),
 					libp2p.DefaultTransports,
 					libp2p.DefaultMuxers,
+					libp2p.ResourceManager(dhtMgr),
 				)
 				if err != nil {
 					return nil, err