Merge pull request #243 from yaoyinnan/242/cgroupv2-addThread

dmcgowan · web-flow · commit 38b9b0028b44 · 2022-09-20T12:26:23.000-07:00
Support cgroup.AddThread in cgroupv2 manager
diff --git a/v2/manager.go b/v2/manager.go
@@ -336,6 +336,14 @@ func (c *Manager) AddProc(pid uint64) error {
 	return writeValues(c.path, []Value{v})
 }
 
+func (c *Manager) AddThread(tid uint64) error {
+	v := Value{
+		filename: cgroupThreads,
+		value:    tid,
+	}
+	return writeValues(c.path, []Value{v})
+}
+
 func (c *Manager) Delete() error {
 	return remove(c.path)
 }
diff --git a/v2/utils.go b/v2/utils.go
@@ -37,6 +37,7 @@ import (
 
 const (
 	cgroupProcs    = "cgroup.procs"
+	cgroupThreads  = "cgroup.threads"
 	defaultDirPerm = 0755
 )
 

Original file line number	Diff line number	Diff line change
`@@ -336,6 +336,14 @@ func (c *Manager) AddProc(pid uint64) error {`
`336`	`336`	`return writeValues(c.path, []Value{v})`
`337`	`337`	`}`
`338`	`338`
	`339`	`+func (c *Manager) AddThread(tid uint64) error {`
	`340`	`+ v := Value{`
	`341`	`+ filename: cgroupThreads,`
	`342`	`+ value: tid,`
	`343`	`+ }`
	`344`	`+ return writeValues(c.path, []Value{v})`
	`345`	`+}`
	`346`	`+`
`339`	`347`	`func (c *Manager) Delete() error {`
`340`	`348`	`return remove(c.path)`
`341`	`349`	`}`
Original file line number	Diff line number	Diff line change
`@@ -37,6 +37,7 @@ import (`
`37`	`37`
`38`	`38`	`const (`
`39`	`39`	`cgroupProcs = "cgroup.procs"`
	`40`	`+ cgroupThreads = "cgroup.threads"`
`40`	`41`	`defaultDirPerm = 0755`
`41`	`42`	`)`
`42`	`43`