api,adaptation,generate: allow setting scheduler attributes.

Allow setting/resetting container linux scheduler attributes.

Signed-off-by: Krisztian Litkey <[email protected]>

Author: klihub

README.md

@@ -205,6 +205,7 @@ The following pieces of container metadata are available to plugins in NRI:
         - cpuset memory
       - Block I/O class
       - RDT class
+    - scheduling policy parameters
   - container (init) process ID
   - container (init process) exit status
   - timestamp of container creation
@@ -250,6 +251,7 @@ container parameters:
         - cpuset memory
       - Block I/O class
       - RDT class
+    - scheduling policy parameters
 
 ### Container Updates
 

pkg/adaptation/adaptation_suite_test.go

@@ -514,6 +514,20 @@ var _ = Describe("Plugin container creation adjustments", func() {
 				},
 			)
 
+		case "linux scheduler":
+			a.SetLinuxScheduler(&api.LinuxScheduler{
+				Policy:   api.LinuxSchedulerPolicy_SCHED_FIFO,
+				Priority: 10,
+				Flags: []api.LinuxSchedulerFlag{
+					api.LinuxSchedulerFlag_SCHED_FLAG_RESET_ON_FORK,
+				},
+			})
+
+		case "clear linux scheduler":
+			a.SetLinuxScheduler(&api.LinuxScheduler{
+				Policy: api.LinuxSchedulerPolicy_SCHED_NONE,
+			})
+
 		case "resources/cpu":
 			a.SetLinuxCPUShares(123)
 			a.SetLinuxCPUQuota(456)
@@ -700,6 +714,28 @@ var _ = Describe("Plugin container creation adjustments", func() {
 					},
 				},
 			),
+			Entry("adjust linux scheduler", "linux scheduler",
+				&api.ContainerAdjustment{
+					Linux: &api.LinuxContainerAdjustment{
+						Scheduler: &api.LinuxScheduler{
+							Policy:   api.LinuxSchedulerPolicy_SCHED_FIFO,
+							Priority: 10,
+							Flags: []api.LinuxSchedulerFlag{
+								api.LinuxSchedulerFlag_SCHED_FLAG_RESET_ON_FORK,
+							},
+						},
+					},
+				},
+			),
+			Entry("clear linux scheduler", "clear linux scheduler",
+				&api.ContainerAdjustment{
+					Linux: &api.LinuxContainerAdjustment{
+						Scheduler: &api.LinuxScheduler{
+							Policy: api.LinuxSchedulerPolicy_SCHED_NONE,
+						},
+					},
+				},
+			),
 			Entry("adjust CPU resources", "resources/cpu",
 				&api.ContainerAdjustment{
 					Linux: &api.LinuxContainerAdjustment{
@@ -921,6 +957,7 @@ var _ = Describe("Plugin container creation adjustments", func() {
 				},
 			),
 			Entry("adjust resources", "resources/classes", false, true, nil),
+			Entry("adjust linux scheduler (conflicts)", "linux scheduler", false, true, nil),
 		)
 	})
 

pkg/adaptation/api.go

@@ -89,6 +89,9 @@ type (
 	LinuxMemory              = api.LinuxMemory
 	LinuxDevice              = api.LinuxDevice
 	LinuxDeviceCgroup        = api.LinuxDeviceCgroup
+	LinuxScheduler           = api.LinuxScheduler
+	LinuxSchedulerPolicy     = api.LinuxSchedulerPolicy
+	LinuxSchedulerFlag       = api.LinuxSchedulerFlag
 	CDIDevice                = api.CDIDevice
 	HugepageLimit            = api.HugepageLimit
 	Hooks                    = api.Hooks

pkg/adaptation/result.go

@@ -221,6 +221,9 @@ func (r *result) adjust(rpl *ContainerAdjustment, plugin string) error {
 		if err := r.adjustOomScoreAdj(rpl.Linux.OomScoreAdj, plugin); err != nil {
 			return err
 		}
+		if err := r.adjustLinuxScheduler(rpl.Linux.Scheduler, plugin); err != nil {
+			return err
+		}
 	}
 	if err := r.adjustRlimits(rpl.Rlimits, plugin); err != nil {
 		return err
@@ -773,6 +776,23 @@ func (r *result) adjustOomScoreAdj(OomScoreAdj *OptionalInt, plugin string) erro
 	return nil
 }
 
+func (r *result) adjustLinuxScheduler(sch *LinuxScheduler, plugin string) error {
+	if sch == nil {
+		return nil
+	}
+
+	create, id := r.request.create, r.request.create.Container.Id
+
+	if err := r.owners.ClaimLinuxScheduler(id, plugin); err != nil {
+		return err
+	}
+
+	create.Container.Linux.Scheduler = sch
+	r.reply.adjust.Linux.Scheduler = sch
+
+	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

@@ -283,6 +283,12 @@ func (a *ContainerAdjustment) SetLinuxOomScoreAdj(value *int) {
 	a.Linux.OomScoreAdj = Int(value) // using Int(value) from ./options.go to optionally allocate a pointer to normalized copy of value
 }
 
+// SetLinuxScheduler records setting the Linux scheduler attributes for a container.
+func (a *ContainerAdjustment) SetLinuxScheduler(sch *LinuxScheduler) {
+	a.initLinux()
+	a.Linux.Scheduler = sch
+}
+
 //
 // Initializing a container adjustment and container update.
 //