fix: close cri (#76)

traas-stack · Oct 16, 2023 · b62baeb · b62baeb
1 parent 02e558f
commit b62baeb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pkg/bootstrap/bootstrap.go b/pkg/bootstrap/bootstrap.go
@@ -320,6 +320,7 @@ func (b *AgentBootstrap) setupDaemonAgent() error {
 	if err := b.setupCRI(); err != nil {
 		return err
 	}
+	b.AddStopComponents(ioc.Crii)
 
 	// system metrics
 	{
@@ -355,7 +356,7 @@ func (b *AgentBootstrap) setupDaemonAgent() error {
 	bsm := bistream.NewManager(ioc.RegistryService, bizbistream.GetBiStreamHandlerRegistry())
 
 	b.TM = manager.NewTransferManager(b.PM, b.LSM)
-	b.TM.AddStopComponents(b.httpServerComponent, ctm, bsm, b.AM)
+	b.TM.AddStopComponents(b.httpServerComponent, ctm, bsm, b.AM, ioc.Crii)
 	if err := b.TM.Transfer(); err != nil {
 		logger.Errorz("[transfer] error", zap.Error(err))
 	}