feat : support kubectl rollout history for statefulset

Author: NeverRaR

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

@@ -25,6 +25,7 @@
 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.openapi.models.V1StatefulSet;
 import io.kubernetes.client.util.labels.LabelSelector;
 import java.util.ArrayList;
 import java.util.HashMap;
@@ -77,6 +78,10 @@ public List<History> execute() throws KubectlException {
         } else if (apiTypeClass.equals(V1DaemonSet.class)) {
           V1DaemonSet daemonSet = api.readNamespacedDaemonSet(name, namespace, null, null, null);
           daemonSetViewHistory(daemonSet, api);
+        } else if (apiTypeClass.equals(V1StatefulSet.class)) {
+          V1StatefulSet statefulSet =
+              api.readNamespacedStatefulSet(name, namespace, null, null, null);
+          statefulSetViewHistory(statefulSet, api);
         } else {
           throw new KubectlException("Unsupported class for rollout history: " + apiTypeClass);
         }
@@ -171,6 +176,26 @@ private V1DaemonSet applyDaemonSetHistory(V1ControllerRevision history)
           PatchHelper.dryRunStrategyMergePatch(getGenericApi(), patch, namespace, name);
     }
 
+    private void statefulSetViewHistory(V1StatefulSet statefulSet, AppsV1Api api)
+        throws ApiException, KubectlException {
+      LabelSelector selector = LabelSelector.parse(statefulSet.getSpec().getSelector());
+      List<V1ControllerRevision> historyList = controlledHistory(api, statefulSet, selector);
+      parseHistory(
+          historyList,
+          history -> {
+            V1StatefulSet stsOfHistory = applyStatefulSetHistory(history);
+            return stsOfHistory.getSpec().getTemplate();
+          });
+    }
+
+    private V1StatefulSet applyStatefulSetHistory(V1ControllerRevision history)
+        throws KubectlException, ApiException {
+      String patch = apiClient.getJSON().serialize(history.getData());
+      return (V1StatefulSet)
+          io.kubernetes.client.extended.kubectl.util.patch.PatchHelper.dryRunStrategyMergePatch(
+              getGenericApi(), patch, namespace, name);
+    }
+
     private void parseHistory(List<V1ControllerRevision> historyList, PodTemplateParser parser)
         throws ApiException, KubectlException {
       Map<Long, V1ControllerRevision> historyInfo = new HashMap<>();

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();
+  }
+}