Adds support for readiness probe

Author: geanpalacios

README.md

@@ -437,6 +437,14 @@ containerTemplate(name: 'busybox', image: 'busybox', command: 'sleep', args: '99
 ```
 See [Defining a liveness command](https://kubernetes.io/docs/tasks/configure-pod-container/configure-liveness-readiness-probes/#defining-a-liveness-command) for more details.
 
+### Readiness Probe Usage
+```groovy
+containerTemplate(name: 'busybox', image: 'busybox', command: 'sleep', args: '99d',
+                  readinessProbe: containerReadinessProbe(execArgs: 'some --command', initialDelaySeconds: 30, timeoutSeconds: 1, failureThreshold: 3, periodSeconds: 10, successThreshold: 1)
+)
+```
+See [Define readiness probes](https://kubernetes.io/docs/tasks/configure-pod-container/configure-liveness-readiness-startup-probes/#define-readiness-probes) for more details.
+
 # Inheritance
 
 ## Overview

src/main/java/org/csanchez/jenkins/plugins/kubernetes/ContainerReadinessProbe.java

@@ -0,0 +1,100 @@
+package org.csanchez.jenkins.plugins.kubernetes;
+
+import hudson.Extension;
+import hudson.model.AbstractDescribableImpl;
+import hudson.model.Descriptor;
+import org.jenkinsci.Symbol;
+import org.kohsuke.stapler.DataBoundConstructor;
+
+import java.io.Serializable;
+
+/**
+ * Created by geanpalacios on 15/02/23.
+ */
+public class ContainerReadinessProbe extends AbstractDescribableImpl<ContainerReadinessProbe> implements Serializable {
+        private String execArgs;
+        private int timeoutSeconds;
+        private int initialDelaySeconds;
+        private int failureThreshold;
+        private int periodSeconds;
+        private int successThreshold;
+
+    @DataBoundConstructor
+    public ContainerReadinessProbe(String execArgs, int timeoutSeconds, int initialDelaySeconds, int failureThreshold, int periodSeconds, int successThreshold) {
+        this.execArgs = execArgs;
+        this.timeoutSeconds = timeoutSeconds;
+        this.initialDelaySeconds = initialDelaySeconds;
+        this.failureThreshold = failureThreshold;
+        this.periodSeconds = periodSeconds;
+        this.successThreshold = successThreshold;
+    }
+
+    public String getExecArgs() {
+        return execArgs;
+    }
+
+    public void setExecArgs(String execArgs) {
+        this.execArgs = execArgs;
+    }
+
+    public int getTimeoutSeconds() {
+        return timeoutSeconds;
+    }
+
+    public void setTimeoutSeconds(int timeoutSeconds) {
+        this.timeoutSeconds = timeoutSeconds;
+    }
+
+    public int getInitialDelaySeconds() {
+        return initialDelaySeconds;
+    }
+
+    public void setInitialDelaySeconds(int initialDelaySeconds) {
+        this.initialDelaySeconds = initialDelaySeconds;
+    }
+
+    public int getFailureThreshold() {
+        return failureThreshold;
+    }
+
+    public void setFailureThreshold(int failureThreshold) {
+        this.failureThreshold = failureThreshold;
+    }
+
+    public int getPeriodSeconds() {
+        return periodSeconds;
+    }
+
+    public void setPeriodSeconds(int periodSeconds) {
+        this.periodSeconds = periodSeconds;
+    }
+
+    public int getSuccessThreshold() {
+        return successThreshold;
+    }
+
+    public void setSuccessThreshold(int successThreshold) {
+        this.successThreshold = successThreshold;
+    }
+
+    @Override
+    public String toString() {
+        return "ContainerReadinessProbe{" +
+                "execArgs='" + execArgs + '\'' +
+                ", timeoutSeconds=" + timeoutSeconds +
+                ", initialDelaySeconds=" + initialDelaySeconds +
+                ", failureThreshold=" + failureThreshold +
+                ", periodSeconds=" + periodSeconds +
+                ", successThreshold=" + successThreshold +
+                '}';
+    }
+
+    @Extension
+    @Symbol("containerReadinessProbe")
+    public static class DescriptorImpl extends Descriptor<ContainerReadinessProbe> {
+        @Override
+        public String getDisplayName() {
+            return "Container Readiness Probe";
+        }
+    }
+}

src/main/java/org/csanchez/jenkins/plugins/kubernetes/ContainerTemplate.java

@@ -69,6 +69,7 @@ public class ContainerTemplate extends AbstractDescribableImpl<ContainerTemplate
     private List<PortMapping> ports = new ArrayList<>();
 
     private ContainerLivenessProbe livenessProbe;
+    private ContainerReadinessProbe readinessProbe;
 
     @Deprecated
     public ContainerTemplate(String image) {
@@ -111,6 +112,7 @@ public ContainerTemplate(ContainerTemplate from) {
         this.setEnvVars(from.getEnvVars());
         this.setPorts(from.getPorts());
         this.setLivenessProbe(from.getLivenessProbe());
+        this.setReadinessProbe(from.getReadinessProbe());
     }
 
     public void setName(String name) {
@@ -229,6 +231,14 @@ public void setEnvVars(List<TemplateEnvVar> envVars) {
     public void setLivenessProbe(ContainerLivenessProbe livenessProbe) {
         this.livenessProbe = livenessProbe;
     }
+    
+    public ContainerReadinessProbe getReadinessProbe() { return readinessProbe; }
+
+    @DataBoundSetter
+    public void setReadinessProbe(ContainerReadinessProbe readinessProbe) {
+        this.readinessProbe = readinessProbe;
+    }
+
 
     public List<PortMapping> getPorts() {
         return ports != null ? ports : Collections.emptyList();
@@ -375,6 +385,7 @@ public String toString() {
                 (envVars == null || envVars.isEmpty() ? "" : ", envVars=" + envVars) +
                 (ports == null || ports.isEmpty() ? "" : ", ports=" + ports) +
                 (livenessProbe == null ? "" : ", livenessProbe=" + livenessProbe) +
+                (readinessProbe == null ? "" : ", readinessProbe=" + readinessProbe) +
                 '}';
     }
 
@@ -446,7 +457,10 @@ public boolean equals(Object o) {
         if (!Objects.equals(ports, that.ports)) {
             return false;
         }
-        return Objects.equals(livenessProbe, that.livenessProbe);
+        if (!Objects.equals(livenessProbe, that.livenessProbe)) {
+            return false;
+        }
+        return Objects.equals(readinessProbe, that.readinessProbe);
     }
 
     @Override
@@ -471,6 +485,7 @@ public int hashCode() {
         result = 31 * result + (envVars != null ? envVars.hashCode() : 0);
         result = 31 * result + (ports != null ? ports.hashCode() : 0);
         result = 31 * result + (livenessProbe != null ? livenessProbe.hashCode() : 0);
+        result = 31 * result + (readinessProbe != null ? readinessProbe.hashCode() : 0);
         return result;
     }
 

src/main/java/org/csanchez/jenkins/plugins/kubernetes/PodTemplateBuilder.java

@@ -487,6 +487,19 @@ private Container createContainer(ContainerTemplate containerTemplate, Collectio
                     .build();
         }
 
+        ContainerReadinessProbe crp = containerTemplate.getReadinessProbe();
+        Probe readinessProbe = null;
+        if (crp != null && parseReadinessProbe(crp.getExecArgs()) != null) {
+            readinessProbe = new ProbeBuilder()
+                    .withExec(new ExecAction(parseReadinessProbe(crp.getExecArgs())))
+                    .withInitialDelaySeconds(crp.getInitialDelaySeconds())
+                    .withTimeoutSeconds(crp.getTimeoutSeconds())
+                    .withFailureThreshold(crp.getFailureThreshold())
+                    .withPeriodSeconds(crp.getPeriodSeconds())
+                    .withSuccessThreshold(crp.getSuccessThreshold())
+                    .build();
+        }
+
         ContainerBuilder containerBuilder = new ContainerBuilder()
                 .withName(substituteEnv(containerTemplate.getName()))
                 .withImage(substituteEnv(containerTemplate.getImage()))
@@ -506,6 +519,7 @@ private Container createContainer(ContainerTemplate containerTemplate, Collectio
                 .withCommand(parseDockerCommand(containerTemplate.getCommand()))
                 .withArgs(arguments)
                 .withLivenessProbe(livenessProbe)
+                .withReadinessProbe(readinessProbe)
                 .withTty(containerTemplate.isTtyEnabled())
                 .withNewResources()
                 .withRequests(getResourcesMap(containerTemplate.getResourceRequestMemory(), containerTemplate.getResourceRequestCpu(),containerTemplate.getResourceRequestEphemeralStorage()))
@@ -571,6 +585,26 @@ static List<String> parseLivenessProbe(String livenessProbeExec) {
         return commands;
     }
 
+    /**
+     * Split a command in the parts that ReadinessProbe needs
+     *
+     * @param readinessProbeExec
+     * @return
+     */
+    @Restricted(NoExternalUse.class)
+    static List<String> parseReadinessProbe(String readinessProbeExec) {
+        if (StringUtils.isBlank(readinessProbeExec)) {
+            return null;
+        }
+        // handle quoted arguments
+        Matcher m = SPLIT_IN_SPACES.matcher(readinessProbeExec);
+        List<String> commands = new ArrayList<String>();
+        while (m.find()) {
+            commands.add(substituteEnv(m.group(1).replace("\"", "").replace("?:\\\"", "")));
+        }
+        return commands;
+    }
+
     private Map<String, Quantity> getResourcesMap(String memory, String cpu, String ephemeralStorage) {
         Map<String, Quantity> builder = new HashMap<>();
         String actualMemory = substituteEnv(memory);

src/main/java/org/csanchez/jenkins/plugins/kubernetes/PodTemplateUtils.java

@@ -109,6 +109,7 @@ public static ContainerTemplate combine(@CheckForNull ContainerTemplate parent,
                 .collect(Collectors.toMap(PortMapping::getName, Function.identity()));
         template.getPorts().stream().forEach(p -> ports.put(p.getName(), p));
         ContainerLivenessProbe livenessProbe = template.getLivenessProbe() != null ? template.getLivenessProbe() : parent.getLivenessProbe();
+        ContainerReadinessProbe readinessProbe = template.getReadinessProbe() != null ? template.getReadinessProbe() : parent.getReadinessProbe();
 
         ContainerTemplate combined = new ContainerTemplate(image);
         combined.setName(name);
@@ -131,6 +132,7 @@ public static ContainerTemplate combine(@CheckForNull ContainerTemplate parent,
         combined.setEnvVars(combineEnvVars(parent, template));
         combined.setPorts(new ArrayList<>(ports.values()));
         combined.setLivenessProbe(livenessProbe);
+        combined.setReadinessProbe(readinessProbe);
         return combined;
     }
 

src/main/resources/org/csanchez/jenkins/plugins/kubernetes/ContainerReadinessProbe/config.jelly

@@ -0,0 +1,30 @@
+<!--
+  Config page
+-->
+<?jelly escape-by-default='true'?>
+<j:jelly xmlns:j="jelly:core" xmlns:st="jelly:stapler" xmlns:d="jelly:define" xmlns:l="/lib/layout"
+         xmlns:t="/lib/hudson" xmlns:f="/lib/form">
+  <f:entry field="execArgs" title="${%Exec action}">
+    <f:textbox/>
+  </f:entry>
+
+  <f:entry field="initialDelaySeconds" title="${%Initial Delay Seconds}">
+    <f:textbox/>
+  </f:entry>
+
+  <f:entry field="timeoutSeconds" title="${%Timeout Seconds}">
+    <f:textbox/>
+  </f:entry>
+
+  <f:entry field="failureThreshold" title="${%Failure Threshold}">
+    <f:textbox/>
+  </f:entry>
+
+  <f:entry field="periodSeconds" title="${%Period Seconds}">
+    <f:textbox/>
+  </f:entry>
+
+  <f:entry field="successThreshold" title="${%Success Threshold}">
+    <f:textbox/>
+  </f:entry>
+</j:jelly>

src/main/resources/org/csanchez/jenkins/plugins/kubernetes/ContainerReadinessProbe/config_zh_CN.properties

@@ -0,0 +1,28 @@
+# The MIT License
+#
+# Copyright (c) 2018, Alauda
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+
+Exec\ action=
+Initial\ Delay\ Seconds=
+Timeout\ Seconds=
+Failure\ Threshold=
+Period\ Seconds=
+Success\ Threshold=

src/main/resources/org/csanchez/jenkins/plugins/kubernetes/ContainerReadinessProbe/help-execArgs.html

@@ -0,0 +1 @@
+Command executed by the readiness probe.  

src/main/resources/org/csanchez/jenkins/plugins/kubernetes/ContainerReadinessProbe/help-failureThreshold.html

@@ -0,0 +1,5 @@
+<p>
+    When a Pod starts and the probe fails, Kubernetes will try <em>failureThreshold</em> times before giving up.<br/>
+    Giving up in case of liveness probe means restarting the container.<br/>
+    In case of readiness probe the Pod will be marked Unready. Defaults to 3. Minimum value is 1.
+</p>

src/main/resources/org/csanchez/jenkins/plugins/kubernetes/ContainerReadinessProbe/help-initialDelaySeconds.html

@@ -0,0 +1 @@
+Number of seconds after the container has started before liveness or readiness probes are initiated. Defaults to 0 seconds. Minimum value is 0.