added attribute terminationGracePeriodSeconds (#36)

theplant · Jul 6, 2021 · 82324bb · 82324bb
1 parent b69d30d
commit 82324bb
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/jsonnetlib/config.jsonnet b/jsonnetlib/config.jsonnet
@@ -19,6 +19,7 @@ assert std.length(v) > 0 : 'version is empty';
   successfulJobsHistoryLimit: 1,
   projectRoot: '/go/src/github.com',
   podSpec: {},
+  terminationGracePeriodSeconds: 30,
 
   // config helper func
   local root = self,

diff --git a/jsonnetlib/k8s.jsonnet b/jsonnetlib/k8s.jsonnet
@@ -314,6 +314,7 @@ cfg {
     env=[],
     container={},
     volumes=[],
+    terminationGracePeriodSeconds=root.terminationGracePeriodSeconds,
     minReplicas=2,
     maxReplicas=3,
     targetCPUUtilizationPercentage=75,
@@ -341,6 +342,7 @@ cfg {
         env=env,
         container=container,
         volumes=volumes,
+        terminationGracePeriodSeconds=terminationGracePeriodSeconds,
         podSpec=podSpec,
       ),
       root.svc(namespace, name, port, targetPort),
@@ -451,6 +453,7 @@ cfg {
     maxSurge=root.maxSurge,
     container={},
     volumes=[],
+    terminationGracePeriodSeconds=root.terminationGracePeriodSeconds,
     podSpec=root.podSpec,
   ):: {
     local labels = { app: name },
@@ -480,6 +483,10 @@ cfg {
       {
         volumes: volumes,
       } else {},
+    local graceShutdown = if std.length(std.toString(terminationGracePeriodSeconds)) > 0 then
+      {
+        terminationGracePeriodSeconds: terminationGracePeriodSeconds,
+      } else {},
     local colocatedSelector = {
       labelSelector: {
         matchExpressions: [{
@@ -553,7 +560,7 @@ cfg {
               },
             } + probe + configmapref(configmap) + envRef(envmap, env) + container,
           ],
-        } + affinity + vols + imagePullSecretsRef(imagePullSecrets) + podSpec,
+        } + affinity + vols + imagePullSecretsRef(imagePullSecrets) + podSpec + graceShutdown,
       },
     },
   },