Enable adjusting container Linux memory policy.

Signed-off-by: Antti Kervinen <[email protected]>

Author: askervin

pkg/adaptation/api.go

@@ -88,6 +88,7 @@ type (
 	LinuxResources           = api.LinuxResources
 	LinuxCPU                 = api.LinuxCPU
 	LinuxMemory              = api.LinuxMemory
+	LinuxMemoryPolicy        = api.LinuxMemoryPolicy
 	LinuxDevice              = api.LinuxDevice
 	LinuxDeviceCgroup        = api.LinuxDeviceCgroup
 	LinuxIOPriority          = api.LinuxIOPriority

pkg/adaptation/result.go

@@ -248,6 +248,9 @@ func (r *result) adjust(rpl *ContainerAdjustment, plugin string) error {
 		if err := r.adjustLinuxScheduler(rpl.Linux.Scheduler, plugin); err != nil {
 			return err
 		}
+		if err := r.adjustMemoryPolicy(rpl.Linux.MemoryPolicy, plugin); err != nil {
+			return err
+		}
 	}
 	if err := r.adjustRlimits(rpl.Rlimits, plugin); err != nil {
 		return err
@@ -974,6 +977,22 @@ func (r *result) adjustLinuxScheduler(sch *LinuxScheduler, plugin string) error
 	return nil
 }
 
+func (r *result) adjustMemoryPolicy(memoryPolicy *LinuxMemoryPolicy, plugin string) error {
+	if memoryPolicy == nil {
+		return nil
+	}
+
+	id := r.request.create.Container.Id
+
+	if err := r.owners.ClaimMemoryPolicy(id, plugin); err != nil {
+		return err
+	}
+
+	r.reply.adjust.Linux.MemoryPolicy = memoryPolicy
+
+	return nil
+}
+
 func (r *result) adjustRlimits(rlimits []*POSIXRlimit, plugin string) error {
 	create, id, adjust := r.request.create, r.request.create.Container.Id, r.reply.adjust
 	for _, l := range rlimits {

pkg/api/adjustment.go

@@ -180,6 +180,14 @@ func (a *ContainerAdjustment) RemoveLinuxNetDevice(hostDev string) {
 	a.Linux.NetDevices[MarkForRemoval(hostDev)] = nil
 }
 
+// SetLinuxMemoryPolicy records setting the Linux memory policy for a container.
+func (a *ContainerAdjustment) SetLinuxMemoryPolicy(mode MpolMode, nodes string, flags ...MpolFlag) {
+	a.initLinuxMemoryPolicy()
+	a.Linux.MemoryPolicy.Mode = mode
+	a.Linux.MemoryPolicy.Nodes = nodes
+	a.Linux.MemoryPolicy.Flags = slices.Clone(flags)
+}
+
 // SetLinuxMemoryLimit records setting the memory limit for a container.
 func (a *ContainerAdjustment) SetLinuxMemoryLimit(value int64) {
 	a.initLinuxResourcesMemory()
@@ -378,6 +386,13 @@ func (a *ContainerAdjustment) initLinuxNamespaces() {
 	}
 }
 
+func (a *ContainerAdjustment) initLinuxMemoryPolicy() {
+	a.initLinux()
+	if a.Linux.MemoryPolicy == nil {
+		a.Linux.MemoryPolicy = &LinuxMemoryPolicy{}
+	}
+}
+
 func (a *ContainerAdjustment) initLinuxResources() {
 	a.initLinux()
 	if a.Linux.Resources == nil {