feat: add Threads for cgroupv2

Rouzip · Rouzip · commit 2585544c0b84 · 2023-11-30T18:02:05.000+08:00
Signed-off-by: Rouzip &lt;1226015390@qq.com&gt;
diff --git a/cgroup2/manager.go b/cgroup2/manager.go
@@ -504,6 +504,32 @@ func (c *Manager) Procs(recursive bool) ([]uint64, error) {
 	return processes, err
 }
 
+func (c *Manager) Threads(recursive bool) ([]uint64, error) {
+	var threads []uint64
+	err := filepath.Walk(c.path, func(p string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+		if !recursive && info.IsDir() {
+			if p == c.path {
+				return nil
+			}
+			return filepath.SkipDir
+		}
+		_, name := filepath.Split(p)
+		if name != cgroupThreads {
+			return nil
+		}
+		procs, err := parseCgroupProcsFile(p)
+		if err != nil {
+			return err
+		}
+		threads = append(threads, procs...)
+		return nil
+	})
+	return threads, err
+}
+
 func (c *Manager) MoveTo(destination *Manager) error {
 	processes, err := c.Procs(true)
 	if err != nil {
diff --git a/cgroup2/manager_test.go b/cgroup2/manager_test.go
@@ -211,6 +211,9 @@ func TestKill(t *testing.T) {
 	pids, err := manager.Procs(true)
 	require.NoError(t, err)
 	require.Len(t, pids, numProcs, "pid count unexpected")
+	threads, err := manager.Threads(true)
+	require.NoError(t, err)
+	require.Len(t, threads, numProcs, "pid count unexpected")
 
 	// Now run kill, and check that nothing is running after.
 	err = manager.Kill()
