Add priorityClassName to initializer/starter/runner pods

Author: vsoloviov

api/v1alpha1/testrun_types.go

@@ -52,6 +52,7 @@ type Pod struct {
 	InitContainers               []InitContainer                   `json:"initContainers,omitempty"`
 	Volumes                      []corev1.Volume                   `json:"volumes,omitempty"`
 	VolumeMounts                 []corev1.VolumeMount              `json:"volumeMounts,omitempty"`
+	PriorityClassName            string                            `json:"priorityClassName,omitempty"`
 }
 
 type InitContainer struct {

config/crd/bases/k6.io_testruns.yaml

@@ -892,6 +892,8 @@ spec:
                     additionalProperties:
                       type: string
                     type: object
+                  priorityClassName:
+                    type: string
                   readinessProbe:
                     properties:
                       exec:
@@ -2839,6 +2841,8 @@ spec:
                     additionalProperties:
                       type: string
                     type: object
+                  priorityClassName:
+                    type: string
                   readinessProbe:
                     properties:
                       exec:
@@ -4807,6 +4811,8 @@ spec:
                     additionalProperties:
                       type: string
                     type: object
+                  priorityClassName:
+                    type: string
                   readinessProbe:
                     properties:
                       exec:

docs/crd-generated.md

@@ -932,6 +932,13 @@ alive or ready to receive traffic.<br/>
           <br/>
         </td>
         <td>false</td>
+      </tr><tr>
+        <td><b>priorityClassName</b></td>
+        <td>string</td>
+        <td>
+          <br/>
+        </td>
+        <td>false</td>
       </tr><tr>
         <td><b><a href="#testrunspecinitializerreadinessprobe">readinessProbe</a></b></td>
         <td>object</td>
@@ -9483,6 +9490,13 @@ alive or ready to receive traffic.<br/>
           <br/>
         </td>
         <td>false</td>
+      </tr><tr>
+        <td><b>priorityClassName</b></td>
+        <td>string</td>
+        <td>
+          <br/>
+        </td>
+        <td>false</td>
       </tr><tr>
         <td><b><a href="#testrunspecrunnerreadinessprobe">readinessProbe</a></b></td>
         <td>object</td>
@@ -18056,6 +18070,13 @@ alive or ready to receive traffic.<br/>
           <br/>
         </td>
         <td>false</td>
+      </tr><tr>
+        <td><b>priorityClassName</b></td>
+        <td>string</td>
+        <td>
+          <br/>
+        </td>
+        <td>false</td>
       </tr><tr>
         <td><b><a href="#testrunspecstarterreadinessprobe">readinessProbe</a></b></td>
         <td>object</td>

pkg/resources/jobs/initializer.go

@@ -128,7 +128,8 @@ func NewInitializerJob(k6 *v1alpha1.TestRun, argLine string) (*batchv1.Job, erro
 							SecurityContext: &k6.GetSpec().Initializer.ContainerSecurityContext,
 						},
 					},
-					Volumes: volumes,
+					Volumes:           volumes,
+					PriorityClassName: k6.GetSpec().Initializer.PriorityClassName,
 				},
 			},
 		},

pkg/resources/jobs/runner.go

@@ -202,6 +202,7 @@ func NewRunnerJob(k6 *v1alpha1.TestRun, index int, tokenInfo *cloud.TokenInfo) (
 					}},
 					TerminationGracePeriodSeconds: &zero,
 					Volumes:                       volumes,
+					PriorityClassName:             k6.GetSpec().Runner.PriorityClassName,
 				},
 			},
 		},

pkg/resources/jobs/starter.go

@@ -96,6 +96,7 @@ func NewStarterJob(k6 *v1alpha1.TestRun, hostname []string) *batchv1.Job {
 							resourceRequirements,
 						),
 					},
+					PriorityClassName: k6.GetSpec().Starter.PriorityClassName,
 				},
 			},
 		},