Crash fix for localmachine collector

lkarlslund · Apr 21, 2022 · 20cc4d0 · 20cc4d0
1 parent 60ded9f
commit 20cc4d0
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/modules/integrations/localmachine/collect/main.go b/modules/integrations/localmachine/collect/main.go
@@ -270,9 +270,11 @@ func Collect(outputpath string) error {
 	var scheduledtasksinfo taskmaster.RegisteredTaskCollection
 	ts, err := taskmaster.Connect()
 	if err == nil {
-		scheduledtasksinfo, _ = ts.GetRegisteredTasks()
-		defer scheduledtasksinfo.Release()
-		defer ts.Disconnect()
+		scheduledtasksinfo, err = ts.GetRegisteredTasks()
+		if err == nil {
+			scheduledtasksinfo.Release()
+		}
+		ts.Disconnect()
 	}
 
 	// GATHER INTERESTING STUFF FROM EVENT LOG
@@ -754,5 +756,6 @@ func Collect(outputpath string) error {
 		return fmt.Errorf("Problem writing to file %v: %v", outputfile, err)
 	}
 	log.Info().Msgf("Information collected to file %v", outputfile)
+
 	return nil
 }