Fix cleanup leak in pipeline collectors

Signed-off-by: Zhang Wei <[email protected]>

pkg/dcgmexporter/pipeline.go

@@ -33,19 +33,25 @@ func NewMetricsPipeline(c *Config) (*MetricsPipeline, func(), error) {
 		return nil, func() {}, err
 	}
 
+	cleanups := []func(){}
 	gpuCollector, cleanup, err := NewDCGMCollector(counters, c, dcgm.FE_GPU)
 	if err != nil {
 		return nil, func() {}, err
 	}
+	cleanups = append(cleanups, cleanup)
 
 	switchCollector, cleanup, err := NewDCGMCollector(counters, c, dcgm.FE_SWITCH)
 	if err != nil {
 		logrus.Info("Not collecting switch metrics: ", err)
+	} else {
+		cleanups = append(cleanups, cleanup)
 	}
 
 	linkCollector, cleanup, err := NewDCGMCollector(counters, c, dcgm.FE_LINK)
 	if err != nil {
 		logrus.Info("Not collecting link metrics: ", err)
+	} else {
+		cleanups = append(cleanups, cleanup)
 	}
 
 	transformations := []Transform{}
@@ -71,7 +77,9 @@ func NewMetricsPipeline(c *Config) (*MetricsPipeline, func(), error) {
 			linkCollector:   linkCollector,
 			transformations: transformations,
 		}, func() {
-			cleanup()
+			for _, cleanup := range cleanups {
+				cleanup()
+			}
 		}, nil
 }