bonsai

Author: k1LoW

grouper/cgroup/cgroup.go

@@ -38,8 +38,8 @@ var DefaultSubsystems = []string{
 type Cgroup struct {
 	fsPath     string
 	subsystems []string
-	nRe        *regexp.Regexp
-	eRe        *regexp.Regexp
+	nRe        *regexp.Regexp // normalize regexp
+	eRe        *regexp.Regexp // exclude regexp
 }
 
 func (c *Cgroup) Name() string {
@@ -106,48 +106,49 @@ func (c *Cgroup) Collect(gprocs *grouped_proc.GroupedProcs, enabled map[metric.M
 					cPath = matches[1]
 				}
 			}
-			if cPath != "" {
-				f, err := os.Open(filepath.Clean(filepath.Join(path, "cgroup.procs")))
-				if err != nil {
+			if cPath == "" {
+				return nil
+			}
+			f, err := os.Open(filepath.Clean(filepath.Join(path, "cgroup.procs")))
+			if err != nil {
+				_ = f.Close()
+				return nil
+			}
+			var (
+				gproc *grouped_proc.GroupedProc
+				ok    bool
+			)
+			gproc, ok = gprocs.Load(cPath)
+			if !ok {
+				gproc = grouped_proc.NewGroupedProc(enabled)
+				gprocs.Store(cPath, gproc)
+			}
+			gproc.Exists = true
+			reader := bufio.NewReaderSize(f, 1028)
+			for {
+				line, _, err := reader.ReadLine()
+				if err == io.EOF {
+					break
+				} else if err != nil {
 					_ = f.Close()
-					return nil
+					return err
 				}
-				var (
-					gproc *grouped_proc.GroupedProc
-					ok    bool
-				)
-				gproc, ok = gprocs.Load(cPath)
-				if !ok {
-					gproc = grouped_proc.NewGroupedProc(enabled)
-					gprocs.Store(cPath, gproc)
+				pid, err := strconv.Atoi(string(line))
+				if err != nil {
+					_ = f.Close()
+					return err
 				}
-				gproc.Exists = true
-				reader := bufio.NewReaderSize(f, 1028)
-				for {
-					line, _, err := reader.ReadLine()
-					if err == io.EOF {
-						break
-					} else if err != nil {
-						_ = f.Close()
-						return err
-					}
-					pid, err := strconv.Atoi(string(line))
-					if err != nil {
-						_ = f.Close()
-						return err
-					}
-					err = sem.Acquire(ctx, gproc.RequiredWeight)
-					if err != nil {
-						_ = f.Close()
-						return err
-					}
-					wg.Add(1)
-					go func(wg *sync.WaitGroup, pid int, gproc *grouped_proc.GroupedProc) {
-						_ = gproc.AppendProcAndCollect(pid)
-						sem.Release(gproc.RequiredWeight)
-						wg.Done()
-					}(wg, pid, gproc)
+				err = sem.Acquire(ctx, gproc.RequiredWeight)
+				if err != nil {
+					_ = f.Close()
+					return err
 				}
+				wg.Add(1)
+				go func(wg *sync.WaitGroup, pid int, gproc *grouped_proc.GroupedProc) {
+					_ = gproc.AppendProcAndCollect(pid)
+					sem.Release(gproc.RequiredWeight)
+					wg.Done()
+				}(wg, pid, gproc)
 			}
 			return nil
 		})

grouper/proc_status_name/proc_status_name.go

@@ -14,8 +14,8 @@ import (
 )
 
 type ProcStatusName struct {
-	nRe            *regexp.Regexp
-	eRe            *regexp.Regexp
+	nRe            *regexp.Regexp // normalize regexp
+	eRe            *regexp.Regexp // exclude regexp
 	procMountPoint string
 }