mgmt aks, support beginCreateAgentPool (Azure#30646)

* draft, support beginCreateAgentPool

* update comments

* fix session records

* fix changelog

* add AgentPoolDataImpl

* make AgentPoolDataImpl package private, fix checkstyle

Co-authored-by: Weidong Xu <[email protected]>

Author: XiaofeiCao

sdk/resourcemanager/azure-resourcemanager-containerservice/CHANGELOG.md

@@ -2,6 +2,10 @@
 
 ## 2.18.0-beta.1 (Unreleased)
 
+### Features Added
+
+- Supported `beginCreateAgentPool` in `KubernetesCluster`.
+
 ### Other Changes
 
 - Deprecated method `KubernetesClusters.listKubernetesVersions`. Use `KubernetesClusters.listOrchestrators`.

sdk/resourcemanager/azure-resourcemanager-containerservice/src/main/java/com/azure/resourcemanager/containerservice/implementation/AgentPoolDataImpl.java

@@ -0,0 +1,21 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+package com.azure.resourcemanager.containerservice.implementation;
+
+import com.azure.resourcemanager.containerservice.fluent.models.AgentPoolInner;
+import com.azure.resourcemanager.containerservice.models.AgentPoolData;
+
+/**
+ * AgentPool implementation that exposes AgentPoolInner as constructor arguments.
+ */
+class AgentPoolDataImpl extends AgentPoolData {
+    /**
+     * Creates an instance of agent pool data.
+     *
+     * @param innerModel the inner model of agent pool.
+     */
+    AgentPoolDataImpl(AgentPoolInner innerModel) {
+        super(innerModel);
+    }
+}

sdk/resourcemanager/azure-resourcemanager-containerservice/src/main/java/com/azure/resourcemanager/containerservice/implementation/KubernetesClusterImpl.java

@@ -8,14 +8,18 @@
 import com.azure.core.http.rest.Response;
 import com.azure.core.http.rest.SimpleResponse;
 import com.azure.core.management.serializer.SerializerFactory;
+import com.azure.core.util.Context;
 import com.azure.core.util.CoreUtils;
 import com.azure.core.util.logging.ClientLogger;
 import com.azure.core.util.serializer.SerializerAdapter;
 import com.azure.core.util.serializer.SerializerEncoding;
 import com.azure.resourcemanager.containerservice.ContainerServiceManager;
+import com.azure.resourcemanager.containerservice.fluent.models.AgentPoolInner;
 import com.azure.resourcemanager.containerservice.fluent.models.ManagedClusterInner;
 import com.azure.resourcemanager.containerservice.fluent.models.PrivateEndpointConnectionInner;
 import com.azure.resourcemanager.containerservice.fluent.models.PrivateLinkResourceInner;
+import com.azure.resourcemanager.containerservice.models.AgentPool;
+import com.azure.resourcemanager.containerservice.models.AgentPoolData;
 import com.azure.resourcemanager.containerservice.models.ContainerServiceLinuxProfile;
 import com.azure.resourcemanager.containerservice.models.ContainerServiceNetworkProfile;
 import com.azure.resourcemanager.containerservice.models.ContainerServiceSshConfiguration;
@@ -39,6 +43,8 @@
 import com.azure.resourcemanager.resources.fluentcore.arm.models.PrivateEndpointConnectionProvisioningState;
 import com.azure.resourcemanager.resources.fluentcore.arm.models.PrivateLinkResource;
 import com.azure.resourcemanager.resources.fluentcore.arm.models.implementation.GroupableResourceImpl;
+import com.azure.resourcemanager.resources.fluentcore.model.Accepted;
+import com.azure.resourcemanager.resources.fluentcore.model.implementation.AcceptedImpl;
 import com.azure.resourcemanager.resources.fluentcore.utils.PagedConverter;
 import com.azure.resourcemanager.resources.fluentcore.utils.ResourceManagerUtils;
 import reactor.core.publisher.Mono;
@@ -297,6 +303,21 @@ public Mono<Void> stopAsync() {
         return manager().kubernetesClusters().stopAsync(this.resourceGroupName(), this.name());
     }
 
+    @Override
+    public Accepted<AgentPool> beginCreateAgentPool(String agentPoolName, AgentPoolData agentPool) {
+        return AcceptedImpl.newAccepted(
+            logger,
+            this.manager().serviceClient().getHttpPipeline(),
+            this.manager().serviceClient().getDefaultPollInterval(),
+            () -> this.manager().serviceClient().getAgentPools()
+                .createOrUpdateWithResponseAsync(resourceGroupName(), name(), agentPoolName, agentPool.innerModel())
+                .block(),
+            AgentPoolDataImpl::new,
+            AgentPoolInner.class,
+            null,
+            Context.NONE);
+    }
+
     @Override
     protected Mono<ManagedClusterInner> getInnerAsync() {
         return this

sdk/resourcemanager/azure-resourcemanager-containerservice/src/main/java/com/azure/resourcemanager/containerservice/models/AgentPool.java

@@ -0,0 +1,94 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+package com.azure.resourcemanager.containerservice.models;
+
+import com.azure.resourcemanager.resources.fluentcore.arm.models.HasName;
+
+import java.util.List;
+import java.util.Map;
+
+/**
+ * A client-side representation for an agent pool.
+ */
+public interface AgentPool extends HasName {
+
+    /** @return the provisioning state of the agent pool */
+    String provisioningState();
+
+    /** @return the number of agents (virtual machines) to host docker containers */
+    int count();
+
+    /** @return size of each agent virtual machine in the agent pool */
+    ContainerServiceVMSizeTypes vmSize();
+
+    /** @return OS disk size in GB set for each virtual machine in the agent pool */
+    int osDiskSizeInGB();
+
+    /** @return OS of each virtual machine in the agent pool */
+    OSType osType();
+
+    /** @return agent pool type */
+    AgentPoolType type();
+
+    /** @return agent pool mode */
+    AgentPoolMode mode();
+
+    /** @return the name of the subnet used by each virtual machine in the agent pool */
+    String subnetName();
+
+    /** @return the ID of the virtual network used by each virtual machine in the agent pool */
+    String networkId();
+
+    /** @return the list of availability zones */
+    List<String> availabilityZones();
+
+    /** @return the map of node labels */
+    Map<String, String> nodeLabels();
+
+    /** @return the list of node taints */
+    List<String> nodeTaints();
+
+    /** @return the power state, Running or Stopped */
+    PowerState powerState();
+
+    /** @return the number of agents (VMs) to host docker containers */
+    int nodeSize();
+
+    /** @return the maximum number of pods per node */
+    int maximumPodsPerNode();
+
+    /** @return whether auto-scaling is enabled */
+    boolean isAutoScalingEnabled();
+
+    /** @return the minimum number of nodes for auto-scaling */
+    int minimumNodeSize();
+
+    /** @return the maximum number of nodes for auto-scaling */
+    int maximumNodeSize();
+
+    /** @return the priority of each virtual machines in the agent pool */
+    ScaleSetPriority virtualMachinePriority();
+
+    /** @return the eviction policy of each virtual machines in the agent pool */
+    ScaleSetEvictionPolicy virtualMachineEvictionPolicy();
+
+    /** @return the maximum price of each spot virtual machines in the agent pool, -1 means pay-as-you-go prices */
+    Double virtualMachineMaximumPrice();
+
+    /**
+     * @return the OS disk type to be used for machines in the agent pool
+     */
+    OSDiskType osDiskType();
+
+    /**
+     * @return the disk type for the placement of emptyDir volumes, container runtime data root,
+     * and Kubelet ephemeral storage
+     */
+    KubeletDiskType kubeletDiskType();
+
+    /**
+     * @return the tags of the agents.
+     */
+    Map<String, String> tags();
+}