fix : let scale support StatefulSet

NeverRaR · NeverRaR · commit 84cc9ccd0f19 · 2021-07-30T19:02:17.000+08:00
diff --git a/extended/src/main/java/io/kubernetes/client/extended/kubectl/KubectlScale.java b/extended/src/main/java/io/kubernetes/client/extended/kubectl/KubectlScale.java
@@ -19,6 +19,7 @@
 import io.kubernetes.client.openapi.apis.AppsV1Api;
 import io.kubernetes.client.openapi.models.V1Deployment;
 import io.kubernetes.client.openapi.models.V1ReplicaSet;
+import io.kubernetes.client.openapi.models.V1StatefulSet;
 import io.kubernetes.client.util.PatchUtils;
 
 public class KubectlScale<ApiType extends KubernetesObject>
@@ -68,6 +69,14 @@ public ApiType execute() throws KubectlException {
                     name, namespace, new V1Patch(jsonPatchStr), null, null, null, null, null),
             V1Patch.PATCH_FORMAT_JSON_PATCH,
             this.apiClient);
+      } else if (apiTypeClass.equals(V1StatefulSet.class)) {
+        return PatchUtils.patch(
+            apiTypeClass,
+            () ->
+                api.patchNamespacedStatefulSetCall(
+                    name, namespace, new V1Patch(jsonPatchStr), null, null, null, null, null),
+            V1Patch.PATCH_FORMAT_JSON_PATCH,
+            this.apiClient);
       } else {
         throw new KubectlException("Unsupported class for scale: " + apiTypeClass);
       }
