From cb12e80969a652570ec15f42eaaf47843032f759 Mon Sep 17 00:00:00 2001
From: Michael Crosby <michael@crosbymichael.com>
Date: Fri, 21 Mar 2014 14:53:47 +0000
Subject: [PATCH] Add cpuset.cpus to cgroups and native driver options
 Docker-DCO-1.1-Signed-off-by: Michael Crosby <michael@crosbymichael.com>
 (github: crosbymichael)

---
 cgroups/cgroups.go | 32 ++++++++++++++++++++++++++++----
 1 file changed, 28 insertions(+), 4 deletions(-)

diff --git a/cgroups/cgroups.go b/cgroups/cgroups.go
index b40e1a3..9d485d1 100644
--- a/cgroups/cgroups.go
+++ b/cgroups/cgroups.go
@@ -16,10 +16,11 @@ type Cgroup struct {
 	Name   string `json:"name,omitempty"`
 	Parent string `json:"parent,omitempty"`
 
-	DeviceAccess bool  `json:"device_access,omitempty"` // name of parent cgroup or slice
-	Memory       int64 `json:"memory,omitempty"`        // Memory limit (in bytes)
-	MemorySwap   int64 `json:"memory_swap,omitempty"`   // Total memory usage (memory + swap); set `-1' to disable swap
-	CpuShares    int64 `json:"cpu_shares,omitempty"`    // CPU shares (relative weight vs. other containers)
+	DeviceAccess bool   `json:"device_access,omitempty"` // name of parent cgroup or slice
+	Memory       int64  `json:"memory,omitempty"`        // Memory limit (in bytes)
+	MemorySwap   int64  `json:"memory_swap,omitempty"`   // Total memory usage (memory + swap); set `-1' to disable swap
+	CpuShares    int64  `json:"cpu_shares,omitempty"`    // CPU shares (relative weight vs. other containers)
+	CpusetCpus   string `json:"cpuset_cpus,omitempty"`   // CPU to use
 }
 
 // https://www.kernel.org/doc/Documentation/cgroups/cgroups.txt
@@ -98,6 +99,7 @@ func (c *Cgroup) Cleanup(root string) error {
 		get("memory"),
 		get("devices"),
 		get("cpu"),
+		get("cpuset"),
 	} {
 		os.RemoveAll(path)
 	}
@@ -150,6 +152,9 @@ func (c *Cgroup) Apply(pid int) error {
 	if err := c.setupCpu(cgroupRoot, pid); err != nil {
 		return err
 	}
+	if err := c.setupCpuset(cgroupRoot, pid); err != nil {
+		return err
+	}
 	return nil
 }
 
@@ -248,3 +253,22 @@ func (c *Cgroup) setupCpu(cgroupRoot string, pid int) (err error) {
 	}
 	return nil
 }
+
+func (c *Cgroup) setupCpuset(cgroupRoot string, pid int) (err error) {
+	if c.CpusetCpus != "" {
+		dir, err := c.Join(cgroupRoot, "cpuset", pid)
+		if err != nil {
+			return err
+		}
+		defer func() {
+			if err != nil {
+				os.RemoveAll(dir)
+			}
+		}()
+
+		if err := writeFile(dir, "cpuset.cpus", c.CpusetCpus); err != nil {
+			return err
+		}
+	}
+	return nil
+}