From 003fc667546e7484b17daa6cd66623b4246120ca Mon Sep 17 00:00:00 2001
From: mii443 <mii@sfc.wide.ad.jp>
Date: Thu, 17 Oct 2024 08:43:23 +0000
Subject: [PATCH] add add_procs

---
 src/cgroup/cgroup.rs | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/cgroup/cgroup.rs b/src/cgroup/cgroup.rs
index d5c22cd..2721705 100644
--- a/src/cgroup/cgroup.rs
+++ b/src/cgroup/cgroup.rs
@@ -145,6 +145,18 @@ impl CGroup {
         self.write_value("cgroup.max.descendants", max)
     }
 
+    pub fn add_procs(&self, procs: Vec<u32>) -> Result<(), std::io::Error> {
+        let to_write = procs
+            .iter()
+            .map(|proc| proc.to_string())
+            .collect::<Vec<String>>()
+            .join("\n");
+
+        self.write("cgroup.procs", &to_write)?;
+
+        Ok(())
+    }
+
     fn write_value<T>(&self, name: &str, value: T) -> Result<(), std::io::Error>
     where
         T: fmt::Display,