feat(controller): add app runtime class

Author: duanhongyi

charts/controller/templates/_helpers.tpl

@@ -22,6 +22,10 @@ env:
 - name: "DRYCC_APP_STORAGE_CLASS"
   value: "{{ .Values.appStorageClass }}"
 {{- end }}
+{{- if (.Values.appRuntimeClass) }}
+- name: "DRYCC_APP_RUNTIME_CLASS"
+  value: "{{ .Values.appRuntimeClass }}"
+{{- end }}
 {{- if (.Values.appPodExecTimeout) }}
 - name: "DRYCC_APP_POD_EXEC_TIMEOUT"
   value: "{{ .Values.appPodExecTimeout }}"

charts/controller/values.yaml

@@ -16,8 +16,10 @@ registrationMode: "admin_only"
 # Option to disable ssl verification to connect to k8s api server
 k8sApiVerifyTls: "true"
 
-# Set storageClassName. It is used for application mount.
+# Set storageClassName. It is used for application.
 appStorageClass: "drycc-storage"
+# Set runtimeClassName. It is used for application.
+appRuntimeClass: ""
 # set timeout seconds. It is used for pod exec
 appPodExecTimeout: 3600
 # Set api replicas

rootfs/api/models/app.py

@@ -1169,10 +1169,12 @@ def _gather_app_settings(self, release, app_settings, process_type, replicas, vo
             'build_type': release.build.type,
             'annotations': json.loads(settings.KUBERNETES_POD_DEFAULT_ANNOTATIONS),
             'healthcheck': healthcheck,
+            'runtime_class_name': settings.DRYCC_APP_RUNTIME_CLASS,
             'lifecycle_post_start': config.lifecycle_post_start,
             'lifecycle_pre_stop': config.lifecycle_pre_stop,
             'routable': routable,
             'deploy_batches': batches,
+            'restart_policy': "Always",
             'deploy_timeout': deploy_timeout,
             'deployment_revision_history_limit': deployment_history,
             'release_summary': release.summary,

rootfs/api/settings/production.py

@@ -328,6 +328,8 @@
 
 DRYCC_APP_STORAGE_CLASS = os.environ.get('DRYCC_APP_STORAGE_CLASS', "")
 
+DRYCC_APP_RUNTIME_CLASS = os.environ.get('DRYCC_APP_RUNTIME_CLASS', "")
+
 DRYCC_APP_POD_EXEC_TIMEOUT = int(os.environ.get('DRYCC_APP_POD_EXEC_TIMEOUT', "3600"))
 
 DRYCC_DEFAULT_CONFIG_TAGS = os.environ.get('DRYCC_DEFAULT_CONFIG_TAGS', '')

rootfs/scheduler/resources/pod.py

@@ -155,9 +155,12 @@ def manifest(self, namespace, name, image, **kwargs):
         }
         # pod manifest spec
         spec = manifest['spec']
+        # pod runtimeClassName
+        if kwargs.get('runtime_class_name', ''):
+            spec['runtimeClassName'] = kwargs.get('runtime_class_name')
 
         # what should the pod do if it exits
-        spec['restartPolicy'] = kwargs.get('restartPolicy', 'Always')
+        spec['restartPolicy'] = kwargs.get('restart_policy', 'Always')
 
         # apply tags as needed to restrict pod to particular node(s)
         spec['nodeSelector'] = kwargs.get('tags', {})