feat : support kubectl rollout history for statefulset

NeverRaR · NeverRaR · commit d6103da630d7 · 2021-08-23T16:02:43.000+08:00
diff --git a/extended/src/main/java/io/kubernetes/client/extended/kubectl/KubectlRollout.java b/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,25 @@ 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)
+          PatchHelper.dryRunStrategyMergePatch(getGenericApi(), patch, namespace, name);
+    }
+
     private void parseHistory(List<V1ControllerRevision> historyList, PodTemplateParser parser)
         throws ApiException, KubectlException {
       Map<Long, V1ControllerRevision> historyInfo = new HashMap<>();
