Merge pull request #1181 from brendandburns/cordon

Add support for kubectl cordon.

Author: k8s-ci-robot

examples/src/main/java/io/kubernetes/client/examples/KubectlExample.java

@@ -14,13 +14,15 @@
 
 import static io.kubernetes.client.extended.kubectl.Kubectl.apiResources;
 import static io.kubernetes.client.extended.kubectl.Kubectl.copy;
+import static io.kubernetes.client.extended.kubectl.Kubectl.cordon;
 import static io.kubernetes.client.extended.kubectl.Kubectl.exec;
 import static io.kubernetes.client.extended.kubectl.Kubectl.label;
 import static io.kubernetes.client.extended.kubectl.Kubectl.log;
 import static io.kubernetes.client.extended.kubectl.Kubectl.portforward;
 import static io.kubernetes.client.extended.kubectl.Kubectl.scale;
 import static io.kubernetes.client.extended.kubectl.Kubectl.taint;
 import static io.kubernetes.client.extended.kubectl.Kubectl.top;
+import static io.kubernetes.client.extended.kubectl.Kubectl.uncordon;
 import static io.kubernetes.client.extended.kubectl.Kubectl.version;
 import static io.kubernetes.client.extended.kubectl.KubectlTop.podMetricSum;
 
@@ -104,6 +106,16 @@ public static void main(String[] args)
     String name = null;
 
     switch (verb) {
+      case "cordon":
+        name = args[1];
+        cordon().apiClient(client).name(name).execute();
+        System.out.println("Node cordoned");
+        System.exit(0);
+      case "uncordon":
+        name = args[1];
+        uncordon().apiClient(client).name(name).execute();
+        System.out.println("Node uncordoned");
+        System.exit(0);
       case "top":
         String what = args[1];
         switch (what) {

extended/src/main/java/io/kubernetes/client/extended/kubectl/Kubectl.java

@@ -23,6 +23,16 @@
  */
 public class Kubectl {
 
+  /** Equivalent for `kubectl cordon` */
+  public static KubectlCordon cordon() {
+    return new KubectlCordon(true);
+  }
+
+  /** Equivalent for `kubectl uncordon` */
+  public static KubectlCordon uncordon() {
+    return new KubectlCordon(false);
+  }
+
   /**
    * Equivalent for `kubectl top`
    *

extended/src/main/java/io/kubernetes/client/extended/kubectl/KubectlCordon.java

@@ -0,0 +1,60 @@
+/*
+Copyright 2020 The Kubernetes Authors.
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+http://www.apache.org/licenses/LICENSE-2.0
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+package io.kubernetes.client.extended.kubectl;
+
+import io.kubernetes.client.custom.V1Patch;
+import io.kubernetes.client.extended.kubectl.exception.KubectlException;
+import io.kubernetes.client.openapi.ApiException;
+import io.kubernetes.client.openapi.apis.CoreV1Api;
+import io.kubernetes.client.openapi.models.V1Node;
+import io.kubernetes.client.util.PatchUtils;
+
+public class KubectlCordon extends Kubectl.ResourceAndContainerBuilder<V1Node, KubectlCordon>
+    implements Kubectl.Executable<V1Node> {
+
+  static final String CORDON_PATCH_STR =
+      "[{\"op\":\"replace\",\"path\":\"/spec/unschedulable\",\"value\":true}]";
+  static String UNCORDON_PATCH_STR =
+      "[{\"op\":\"replace\",\"path\":\"/spec/unschedulable\",\"value\":false}]";
+
+  private boolean cordon;
+
+  KubectlCordon(boolean performCordon) {
+    super(V1Node.class);
+    this.cordon = performCordon;
+  }
+
+  @Override
+  public V1Node execute() throws KubectlException {
+    String patch = this.cordon ? CORDON_PATCH_STR : UNCORDON_PATCH_STR;
+    CoreV1Api api = new CoreV1Api(apiClient);
+
+    try {
+      return PatchUtils.patch(
+          V1Node.class,
+          () ->
+              api.patchNodeCall(
+                  name,
+                  new V1Patch(patch),
+                  null,
+                  null,
+                  null, // field-manager is optional
+                  null,
+                  null),
+          V1Patch.PATCH_FORMAT_JSON_PATCH,
+          apiClient);
+    } catch (ApiException ex) {
+      throw new KubectlException(ex);
+    }
+  }
+}