Expose a method to configure a command directly into a cgroup

This commit exposes a method from the cgroup manager to configure a
command directly into a cgroup. This is done by configuring the cgroup
in the SysProcAttr so the command gets spawned directly in the cgroup.
The functionality is only available with cgroups version 2 and kernel
version 5.7 or later.

1 files changed, 9 insertions, 0 deletions

diff --git a/internal/limiter/watchers/testhelper_test.go b/internal/limiter/watchers/testhelper_test.go
index fa9539a20..81fcef53c 100644
--- a/internal/limiter/watchers/testhelper_test.go
+++ b/internal/limiter/watchers/testhelper_test.go
@@ -1,6 +1,7 @@
 package watchers
 
 import (
+	"io"
 	"os/exec"
 	"testing"
 
@@ -32,3 +33,11 @@ func (m *testCgroupManager) Collect(ch chan<- prometheus.Metric) {}
 func (m *testCgroupManager) AddCommand(*exec.Cmd, ...cgroups.AddCommandOption) (string, error) {
 	return "", nil
 }
+
+func (m *testCgroupManager) SupportsCloneIntoCgroup() bool {
+	return true
+}
+
+func (m *testCgroupManager) CloneIntoCgroup(*exec.Cmd, ...cgroups.AddCommandOption) (string, io.Closer, error) {
+	return "", io.NopCloser(nil), nil
+}