cache: enable setting scheduling params in cached containers

Signed-off-by: Antti Kervinen <antti.kervinen@intel.com>

Author: askervin

pkg/resmgr/cache/cache.go

@@ -265,6 +265,25 @@ type Container interface {
 	// SetMemorySwap sets the swap limit in bytes for the container.
 	SetMemorySwap(int64)
 
+	// SetSchedulingPolicy sets the scheduling policy for the container.
+	SetSchedulingPolicy(nri.LinuxSchedulerPolicy)
+	// SetSchedulingNice sets the nice value for the container.
+	SetSchedulingNice(int32)
+	// SetSchedulingPriority sets the real-time priority for the container.
+	SetSchedulingPriority(int32)
+	// SetSchedulingFlags sets the scheduling flags for the container.
+	SetSchedulingFlags([]nri.LinuxSchedulerFlag)
+	// SetSchedulingRuntime sets the runtime for the container in microseconds.
+	SetSchedulingRuntime(uint64)
+	// SetSchedulingDeadline sets the deadline for the container in microseconds.
+	SetSchedulingDeadline(uint64)
+	// SetSchedulingPeriod sets the scheduling period for the container in microseconds.
+	SetSchedulingPeriod(uint64)
+	// SetSchedulingIOClass sets the I/O priority class for the container.
+	SetSchedulingIOClass(nri.IOPrioClass)
+	// SetSchedulingIOPriority sets the I/O priority for the container.
+	SetSchedulingIOPriority(int32)
+
 	// GetCPUShares gets the CFS CPU shares of the container.
 	GetCPUShares() int64
 	// GetCPUQuota gets the CFS CPU quota of the container.
@@ -340,7 +359,9 @@ type container struct {
 	ResourceUpdates *v1.ResourceRequirements
 	request         interface{}
 
-	Resources *nri.LinuxResources
+	Resources       *nri.LinuxResources
+	LinuxScheduler  *nri.LinuxScheduler
+	LinuxIOPriority *nri.LinuxIOPriority
 
 	TopologyHints topology.Hints    // Set of topology hints for all containers within Pod
 	Tags          map[string]string // container tags (local dynamic labels)

pkg/resmgr/cache/container.go

@@ -601,6 +601,14 @@ func (c *container) GetLinuxResources() *nri.LinuxResources {
 	return c.Resources
 }
 
+func (c *container) GetLinuxScheduler() *nri.LinuxScheduler {
+	return c.LinuxScheduler
+}
+
+func (c *container) GetLinuxIOPriority() *nri.LinuxIOPriority {
+	return c.LinuxIOPriority
+}
+
 func (c *container) GetTopologyHints() topology.Hints {
 	return c.TopologyHints
 }
@@ -687,6 +695,18 @@ func (c *container) ensureLinuxResourcesMemory() {
 	}
 }
 
+func (c *container) ensureLinuxScheduler() {
+	if c.LinuxScheduler == nil {
+		c.LinuxScheduler = &nri.LinuxScheduler{}
+	}
+}
+
+func (c *container) ensureLinuxIOPriority() {
+	if c.LinuxIOPriority == nil {
+		c.LinuxIOPriority = &nri.LinuxIOPriority{}
+	}
+}
+
 func (c *container) SetCPUShares(value int64) {
 	switch req := c.getPendingRequest().(type) {
 	case *nri.ContainerAdjustment:
@@ -806,6 +826,93 @@ func (c *container) SetMemorySwap(value int64) {
 	c.Ctr.Linux.Resources.Memory.Swap = nri.Int64(value)
 }
 
+// Scheduling properties can be set only during container creation.
+// setSchedulingOnAdjustment is a helper to set scheduling parameters
+func (c *container) setSchedulingOnAdjustment(setter func(sch *nri.LinuxScheduler)) {
+	switch req := c.getPendingRequest().(type) {
+	case *nri.ContainerAdjustment:
+		c.ensureLinuxScheduler()
+		sch := c.GetLinuxScheduler()
+		setter(sch)
+		req.SetLinuxScheduler(sch)
+	default:
+		log.Error("%s: can't set scheduling parameter: incorrect pending request type %T",
+			c.PrettyName(), c.request)
+		return
+	}
+	c.markPending(NRI)
+}
+
+func (c *container) SetSchedulingPolicy(value nri.LinuxSchedulerPolicy) {
+	c.setSchedulingOnAdjustment(func(sch *nri.LinuxScheduler) {
+		sch.Policy = value
+	})
+}
+
+func (c *container) SetSchedulingNice(value int32) {
+	c.setSchedulingOnAdjustment(func(sch *nri.LinuxScheduler) {
+		sch.Nice = value
+	})
+}
+
+func (c *container) SetSchedulingPriority(value int32) {
+	c.setSchedulingOnAdjustment(func(sch *nri.LinuxScheduler) {
+		sch.Priority = value
+	})
+}
+
+func (c *container) SetSchedulingFlags(value []nri.LinuxSchedulerFlag) {
+	c.setSchedulingOnAdjustment(func(sch *nri.LinuxScheduler) {
+		sch.Flags = make([]nri.LinuxSchedulerFlag, len(value))
+		copy(sch.Flags, value)
+	})
+}
+
+func (c *container) SetSchedulingRuntime(value uint64) {
+	c.setSchedulingOnAdjustment(func(sch *nri.LinuxScheduler) {
+		sch.Runtime = value
+	})
+}
+
+func (c *container) SetSchedulingDeadline(value uint64) {
+	c.setSchedulingOnAdjustment(func(sch *nri.LinuxScheduler) {
+		sch.Deadline = value
+	})
+}
+
+func (c *container) SetSchedulingPeriod(value uint64) {
+	c.setSchedulingOnAdjustment(func(sch *nri.LinuxScheduler) {
+		sch.Period = value
+	})
+}
+
+func (c *container) setIOPriorityOnAdjustment(setter func(iop *nri.LinuxIOPriority)) {
+	switch req := c.getPendingRequest().(type) {
+	case *nri.ContainerAdjustment:
+		c.ensureLinuxIOPriority()
+		iop := c.GetLinuxIOPriority()
+		setter(iop)
+		req.SetLinuxIOPriority(iop)
+	default:
+		log.Error("%s: can't set IO priority parameter: incorrect pending request type %T",
+			c.PrettyName(), c.request)
+		return
+	}
+	c.markPending(NRI)
+}
+
+func (c *container) SetSchedulingIOClass(value nri.IOPrioClass) {
+	c.setIOPriorityOnAdjustment(func(iop *nri.LinuxIOPriority) {
+		iop.Class = value
+	})
+}
+
+func (c *container) SetSchedulingIOPriority(value int32) {
+	c.setIOPriorityOnAdjustment(func(iop *nri.LinuxIOPriority) {
+		iop.Priority = value
+	})
+}
+
 func (c *container) GetCPUShares() int64 {
 	return int64(c.Ctr.GetLinux().GetResources().GetCpu().GetShares().GetValue())
 }