fix: Initial support for pos priority classes

Author: Alan Christie

operator/handlers.py

@@ -30,6 +30,16 @@
 # The ingress class
 _DEFAULT_INGRESS_CLASS: str = "nginx"
 
+# Apply Pod Priority class?
+# Any value results in setting the Pod's Priority Class
+_APPLY_POD_PRIORITY_CLASS: Optional[str] = os.environ.get("JO_APPLY_POD_PRIORITY_CLASS")
+# If set and JO_APPLY_POD_PRIORITY_CLASS is set
+# this value will be used if now alternative is available.
+_DEFAULT_POD_PRIORITY_CLASS: str = os.environ.get(
+    "JO_DEFAULT_POD_PRIORITY_CLASS", "im-worker-medium"
+)
+
+
 # The cert-manager issuer,
 # expected if a INGRESS_TLS_SECRET is not defined.
 ingress_cert_issuer: Optional[str] = os.environ.get("INGRESS_CERT_ISSUER")
@@ -284,6 +294,12 @@ def create(spec: Dict[str, Any], name: str, namespace: str, **_: Any) -> Dict[st
         },
     }
 
+    # Insert a pod priority class?
+    if _APPLY_POD_PRIORITY_CLASS:
+        deployment_body["spec"]["template"]["spec"][
+            "priorityClassName"
+        ] = _DEFAULT_POD_PRIORITY_CLASS
+
     # Additional labels?
     # Provided by the DM as an array of strings of the form '<KEY>=<VALUE>'
     for label in material.get("labels", []):

roles/operator/defaults/main.yaml

@@ -38,3 +38,6 @@ jo_dmapi_namespace: data-manager-api
 # Jupyter will be run on nodes that have labels keys and values defined here...
 jo_pod_node_selector_key: informaticsmatters.com/purpose-worker
 jo_pod_node_selector_value: 'yes'
+
+# Apply Priority Class to Pods launched
+jo_apply_pod_priority_class: no

roles/operator/templates/deployment.yaml.j2

@@ -39,6 +39,10 @@ spec:
         imagePullPolicy: IfNotPresent
 {% endif %}
         env:
+{% if jo_apply_pod_priority_class %}
+        - name: JO_APPLY_POD_PRIORITY_CLASS
+          value: 'true'
+{% endif %}
         - name: JO_POD_NODE_SELECTOR_KEY
           value: '{{ jo_pod_node_selector_key }}'
         - name: JO_POD_NODE_SELECTOR_VALUE