[api] support pod priority classes

m3db · Apr 8, 2019 · 5f95784 · 5f95784
1 parent 72037d8
commit 5f95784
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 1 deletion.
diff --git a/docs/api.md b/docs/api.md
@@ -52,6 +52,7 @@ ClusterSpec defines the desired state for a M3 cluster to be converge to.
 | securityContext | SecurityContext allows the user to specify a container-level security context. | *corev1.SecurityContext | false |
 | labels | Labels sets the base labels that will be applied to resources created by the cluster. // TODO(schallert): design doc on labeling scheme. | map[string]string | false |
 | tolerations | Tolerations sets the tolerations that will be applied to all M3DB pods. | []corev1.Toleration | false |
+| priorityClassName | PriorityClassName sets the priority class for all M3DB pods. | string | false |
 
 [Back to TOC](#table-of-contents)
 

diff --git a/pkg/apis/m3dboperator/v1alpha1/cluster.go b/pkg/apis/m3dboperator/v1alpha1/cluster.go
@@ -227,6 +227,10 @@ type ClusterSpec struct {
 	// Tolerations sets the tolerations that will be applied to all M3DB pods.
 	// +optional
 	Tolerations []corev1.Toleration `json:"tolerations,omitempty"`
+
+	// PriorityClassName sets the priority class for all M3DB pods.
+	// +optional
+	PriorityClassName string `json:"priorityClassName,omitempty"`
 }
 
 // IsolationGroup defines the name of zone as well attributes for the zone configuration

diff --git a/pkg/k8sops/fixtures/testM3DBCluster.yaml b/pkg/k8sops/fixtures/testM3DBCluster.yaml
@@ -29,6 +29,7 @@ spec:
     fsGroup: 10
   securityContext:
     runAsUser: 20
+  priorityClassName: m3db-priority
   etcdEndpoints:
     - ep0
     - ep1

diff --git a/pkg/k8sops/generators_test.go b/pkg/k8sops/generators_test.go
@@ -132,6 +132,7 @@ func TestGenerateStatefulSet(t *testing.T) {
 					Labels: labels,
 				},
 				Spec: v1.PodSpec{
+					PriorityClassName: "m3db-priority",
 					SecurityContext: &v1.PodSecurityContext{
 						FSGroup: pointer.Int64Ptr(10),
 					},

diff --git a/pkg/k8sops/statefulset.go b/pkg/k8sops/statefulset.go
@@ -252,7 +252,8 @@ func NewBaseStatefulSet(ssName, isolationGroup string, cluster *myspec.M3DBClust
 					Labels: objLabels,
 				},
 				Spec: v1.PodSpec{
-					SecurityContext: cluster.Spec.PodSecurityContext,
+					PriorityClassName: cluster.Spec.PriorityClassName,
+					SecurityContext:   cluster.Spec.PodSecurityContext,
 					Containers: []v1.Container{
 						{
 							Name:            ssName,