feat: run.WithCPUSet and run.WithMems

Signed-off-by: thediveo <thediveo@gmx.eu>

Author: thediveo

run/doc.go

@@ -1,4 +1,11 @@
 /*
 Package run provides configuration options for running containers.
+
+In many cases, Docker's documentation on [docker container run] and the
+[containers/create API endpoint] can give useful hints as to what various API
+configuration settings might do.
+
+[docker container run]: https://docs.docker.com/reference/cli/docker/container/run/
+[containers/create API endpoint]: https://docs.docker.com/reference/api/engine/version/v1.39/#tag/Container/operation/ContainerCreate
 */
 package run

run/options.go

@@ -632,6 +632,37 @@ func WithAllPortsPublished() Opt {
 	}
 }
 
+// WithCPUSet configures the [set of CPUs] on which processes of the container
+// are allowed to execute. The list of CPUs is a comma-separated list of CPU
+// numbers and ranges of numbers, where the numbers are decimal numbers. For
+// instance, “1,3,5”, "0-42,666", et cetera.
+//
+// To avoid stuttering this option is simply named “WithCPUSet” instead of
+// “WithCpusetCPUs”, or similar awkward letter salads.
+//
+// [set of CPUs]: https://man7.org/linux/man-pages/man7/cpuset.7.html
+func WithCPUSet(cpulist string) Opt {
+	return func(o *Options) error {
+		o.Host.CpusetCpus = cpulist
+		return nil
+	}
+}
+
+// WithMems configures the [set of memory nodes] on which processes of the
+// container are allowed to allocate memory. The list of memory nodes is a
+// comma-separated list of memory node numbers and ranges of numbers, where the
+// numbers are decimal numbers. For instance, “1,3,5”, "0-42,666", et cetera.
+//
+// [set of memory nodes]: https://man7.org/linux/man-pages/man7/cpuset.7.html
+func WithMems(memlist string) Opt {
+	return func(o *Options) error {
+		o.Host.CpusetMems = memlist
+		return nil
+	}
+}
+
+// WithCustomInit instructs Docker to run an init inside the container that
+// forwards signals and reaps processes.
 func WithCustomInit() Opt {
 	return func(o *Options) error {
 		customInit := true

run/options_test.go

@@ -112,6 +112,8 @@ var _ = Describe("run (container) options", func() {
 			WithPublishedPort("[fe80::dead:beef]:2345:1234"),
 			WithPublishedPort("127.0.0.1:1234"),
 			WithPublishedPort("127.0.0.2:12345/udp"),
+			WithCPUSet("1,3,5,99"),
+			WithMems("6,66"),
 			WithCustomInit(),
 		)
 
@@ -167,6 +169,8 @@ var _ = Describe("run (container) options", func() {
 			nat.Port("12345/udp"),
 			ConsistOf(nat.PortBinding{HostIP: "127.0.0.2", HostPort: "0"})))
 
+		Expect(o.Host.CpusetCpus).To(Equal("1,3,5,99"))
+		Expect(o.Host.CpusetMems).To(Equal("6,66"))
 		Expect(o.Host.Init).To(gstruct.PointTo(BeTrue()))
 
 		o = opts(WithLabel("foo=bar"))