feat : support kubectl rollout history for daemonset

Author: NeverRaR

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

@@ -15,12 +15,17 @@
 import io.kubernetes.client.common.KubernetesObject;
 import io.kubernetes.client.extended.kubectl.exception.KubectlException;
 import io.kubernetes.client.extended.kubectl.util.deployment.DeploymentHelper;
+import io.kubernetes.client.extended.kubectl.util.patch.PatchHelper;
 import io.kubernetes.client.openapi.ApiException;
 import io.kubernetes.client.openapi.apis.AppsV1Api;
+import io.kubernetes.client.openapi.models.V1ControllerRevision;
+import io.kubernetes.client.openapi.models.V1ControllerRevisionList;
+import io.kubernetes.client.openapi.models.V1DaemonSet;
 import io.kubernetes.client.openapi.models.V1Deployment;
 import io.kubernetes.client.openapi.models.V1ObjectMeta;
 import io.kubernetes.client.openapi.models.V1PodTemplateSpec;
 import io.kubernetes.client.openapi.models.V1ReplicaSet;
+import io.kubernetes.client.util.labels.LabelSelector;
 import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
@@ -30,6 +35,11 @@ public class KubectlRolloutHistory<ApiType extends KubernetesObject>
     extends Kubectl.ResourceBuilder<ApiType, KubectlRolloutHistory<ApiType>>
     implements Kubectl.Executable<ApiType> {
 
+  @FunctionalInterface
+  public interface PodTemplateParser {
+    V1PodTemplateSpec parse(V1ControllerRevision history) throws ApiException, KubectlException;
+  }
+
   public static final String CHANGE_CAUSE_ANNOTATION = "kubernetes.io/change-cause";
 
   public V1PodTemplateSpec getTemplate() {
@@ -73,11 +83,16 @@ public String toString() {
   public ApiType execute() throws KubectlException {
     validate();
     AppsV1Api api = new AppsV1Api(this.apiClient);
+    refreshDiscovery();
     try {
       if (apiTypeClass.equals(V1Deployment.class)) {
         V1Deployment deployment = api.readNamespacedDeployment(name, namespace, null, null, null);
         deploymentViewHistory(deployment, api);
         return (ApiType) deployment;
+      } else if (apiTypeClass.equals(V1DaemonSet.class)) {
+        V1DaemonSet daemonSet = api.readNamespacedDaemonSet(name, namespace, null, null, null);
+        daemonSetViewHistory(daemonSet, api);
+        return (ApiType) daemonSet;
       } else {
         throw new KubectlException("Unsupported class for rollout history: " + apiTypeClass);
       }
@@ -154,6 +169,78 @@ private void deploymentViewHistory(V1Deployment deployment, AppsV1Api api) throw
     }
   }
 
+  private void daemonSetViewHistory(V1DaemonSet daemonSet, AppsV1Api api)
+      throws ApiException, KubectlException {
+    LabelSelector selector = LabelSelector.parse(daemonSet.getSpec().getSelector());
+    List<V1ControllerRevision> historyList = controlledHistory(api, daemonSet, selector);
+    parseHistory(
+        historyList,
+        history -> {
+          V1DaemonSet dsOfHistory = applyDaemonSetHistory(history);
+          return dsOfHistory.getSpec().getTemplate();
+        });
+  }
+
+  private V1DaemonSet applyDaemonSetHistory(V1ControllerRevision history)
+      throws KubectlException, ApiException {
+    //todo use server-side-apply
+    String patch = apiClient.getJSON().serialize(history.getData());
+    return (V1DaemonSet)
+        PatchHelper.dryRunStrategyMergePatch(getGenericApi(), patch, namespace, name);
+  }
+
+  private void parseHistory(List<V1ControllerRevision> historyList, PodTemplateParser parser)
+      throws ApiException, KubectlException {
+    Map<Long, V1ControllerRevision> historyInfo = new HashMap<>();
+    for (V1ControllerRevision history : historyList) {
+      historyInfo.put(history.getRevision(), history);
+    }
+
+    if (revision > 0) {
+      V1ControllerRevision history = historyInfo.get(revision);
+      if (history == null) {
+        throw new ApiException("unable to find the specified revision " + revision);
+      }
+      template = parser.parse(history);
+      return;
+    }
+
+    List<Long> revisions = new ArrayList<>(historyInfo.keySet());
+    revisions.sort(Long::compareTo);
+    for (Long revision : revisions) {
+      String changeCause =
+          historyInfo.get(revision).getMetadata().getAnnotations().get(CHANGE_CAUSE_ANNOTATION);
+      if (changeCause == null || changeCause.isEmpty()) {
+        changeCause = "<none>";
+      }
+      histories.add(new History(revision, changeCause));
+    }
+  }
+
+  // controlledHistories returns all ControllerRevisions in namespace that selected by selector and
+  // owned by accessor
+  private List<V1ControllerRevision> controlledHistory(
+      AppsV1Api api, KubernetesObject accessor, LabelSelector selector) throws ApiException {
+    List<V1ControllerRevision> result = new ArrayList<>();
+    V1ControllerRevisionList historyList =
+        api.listNamespacedControllerRevision(
+            namespace, null, null, null, null, selector.toString(), null, null, null, null, null);
+    for (V1ControllerRevision history : historyList.getItems()) {
+      if (isControlledBy(history, accessor)) {
+        result.add(history);
+      }
+    }
+    return result;
+  }
+
+  private boolean isControlledBy(KubernetesObject obj, KubernetesObject owner) {
+    return obj.getMetadata().getOwnerReferences().stream()
+        .filter(r -> r.getController() != null && r.getController())
+        .findAny()
+        .map(v1OwnerReference -> v1OwnerReference.getUid().equals(owner.getMetadata().getUid()))
+        .orElse(false);
+  }
+
   // getChangeCause returns the change-cause annotation of the input object
   private static String getChangeCause(V1ObjectMeta meta) {
     return meta.getAnnotations().get(CHANGE_CAUSE_ANNOTATION);

extended/src/main/java/io/kubernetes/client/extended/kubectl/util/patch/PatchHelper.java

@@ -0,0 +1,42 @@
+/*
+Copyright 2021 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.util.patch;
+
+import io.kubernetes.client.common.KubernetesListObject;
+import io.kubernetes.client.common.KubernetesObject;
+import io.kubernetes.client.custom.V1Patch;
+import io.kubernetes.client.openapi.ApiException;
+import io.kubernetes.client.util.generic.GenericKubernetesApi;
+import io.kubernetes.client.util.generic.options.PatchOptions;
+
+public class PatchHelper {
+  public static <ApiType extends KubernetesObject, ApiListType extends KubernetesListObject>
+      ApiType dryRunStrategyMergePatch(
+          GenericKubernetesApi<ApiType, ApiListType> genericApi,
+          String patch,
+          String namespace,
+          String name)
+          throws ApiException {
+    PatchOptions options = new PatchOptions();
+    options.setDryRun("All");
+    return genericApi
+        .patch(
+            namespace,
+            name,
+            V1Patch.PATCH_FORMAT_STRATEGIC_MERGE_PATCH,
+            new V1Patch(patch),
+            options)
+        .throwsApiException()
+        .getObject();
+  }
+}