close the rcmgr on shutdown

filecoin-project · Jan 20, 2022 · 73ec10a · 73ec10a
1 parent 325a4c0
commit 73ec10a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/node/modules/lp2p/rcmgr.go b/node/modules/lp2p/rcmgr.go
@@ -1,6 +1,7 @@
 package lp2p
 
 import (
+	"context"
 	"errors"
 	"fmt"
 	"os"
@@ -56,6 +57,11 @@ func ResourceManager(lc fx.Lifecycle, repo repo.LockedRepo) (network.ResourceMan
 		return nil, fmt.Errorf("error creating resource manager: %w", err)
 	}
 
+	lc.Append(fx.Hook{
+		OnStop: func(_ context.Context) error {
+			return mgr.Close()
+		}})
+
 	return mgr, nil
 }