cgroups: add ability to update CPUSet

add function for applying a CPUset change to a cgroup Signed-off-by: Eric Ernst <eric@amperecomputing.com>
kata-containers · Jul 16, 2020 · 093aaa8 · 093aaa8
1 parent 9fa2bf1
commit 093aaa8
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/virtcontainers/pkg/cgroups/manager.go b/virtcontainers/pkg/cgroups/manager.go
@@ -331,3 +331,16 @@ func (m *Manager) RemoveDevice(device string) error {
 	m.Unlock()
 	return fmt.Errorf("device %v not found in the cgroup", device)
 }
+
+func (m *Manager) SetCPUSet(cpuset string) error {
+	cgroups, err := m.GetCgroups()
+	if err != nil {
+		return err
+	}
+
+	m.Lock()
+	cgroups.CpusetCpus = cpuset
+	m.Unlock()
+
+	return m.Apply()
+}