feat: Support az aks loadbalancer commands

Author: nilo19

src/aks-preview/HISTORY.rst

@@ -11,7 +11,11 @@ To release a new version, please select a new version number (usually plus 1 to
 
 Pending
 +++++++
+
+13.0.0b9
++++++++
 * Vendor new SDK and bump API version to 2024-10-02-preview.
+* Support `az aks loadbalancer show/list/add/update/delete/rebalance` commands.
 
 13.0.0b8
 +++++++

src/aks-preview/azext_aks_preview/_client_factory.py

@@ -151,3 +151,7 @@ def get_providers_client_factory(cli_ctx, subscription_id=None):
 
 def get_keyvault_client(cli_ctx, subscription_id=None):
     return get_mgmt_service_client(cli_ctx, ResourceType.MGMT_KEYVAULT, subscription_id=subscription_id).vaults
+
+
+def cf_load_balancers(cli_ctx, *_):
+    return get_container_service_client(cli_ctx).load_balancers

src/aks-preview/azext_aks_preview/_help.py

@@ -3260,3 +3260,119 @@
         type: string
         short-summary: Additional endpoint(s) to perform the connectivity check, separated by comma.
 """
+
+helps['aks loadbalancer'] = """
+    type: group
+    short-summary: Commands to manage load balancer configurations in a managed Kubernetes cluster.
+    long-summary: These commands enable the feature of multiple standard load balancers for Azure Kubernetes Service clusters.
+"""
+
+helps['aks loadbalancer add'] = """
+    type: command
+    short-summary: Add a load balancer configuration to a managed Kubernetes cluster.
+    parameters:
+        - name: --name -n
+          type: string
+          short-summary: Name of the load balancer configuration.
+          long-summary: Load balancer name used for identification. There must be a configuration named "kubernetes" in the cluster.
+        - name: --primary-agent-pool-name -p
+          type: string
+          short-summary: Name of the primary agent pool for this load balancer.
+          long-summary: Required field. A string value that must specify the ID of an existing agent pool. All nodes in the given pool will always be added to this load balancer.
+        - name: --allow-service-placement -a
+          type: bool
+          short-summary: Whether to automatically place services on the load balancer.
+          long-summary: If not supplied, the default value is true. If set to false manually, both the external and internal load balancer will not be selected for services unless they explicitly target it.
+        - name: --service-label-selector -l
+          type: string
+          short-summary: Label selector for services that can be placed on this load balancer.
+          long-summary: Only services that match this selector can be placed on this load balancer. Format as comma-separated key=value pairs or expressions like "key In value1,value2".
+        - name: --service-namespace-selector -s
+          type: string
+          short-summary: Namespace label selector for services that can be placed on this load balancer.
+          long-summary: Services created in namespaces that match the selector can be placed on this load balancer. Format as comma-separated key=value pairs.
+        - name: --node-selector -d
+          type: string
+          short-summary: Node label selector for nodes that can be members of this load balancer.
+          long-summary: Nodes that match this selector will be possible members of this load balancer. Format as comma-separated key=value pairs.
+    examples:
+        - name: Add a load balancer configuration with a specific primary agent pool
+          text: az aks loadbalancer add -g MyResourceGroup -n secondary --cluster-name MyManagedCluster --primary-agent-pool-name nodepool1
+        - name: Add a load balancer configuration with service label selector
+          text: az aks loadbalancer add -g MyResourceGroup -n app-lb --cluster-name MyManagedCluster --primary-agent-pool-name nodepool2 --service-label-selector app=frontend
+        - name: Add a load balancer configuration that doesn't automatically place services
+          text: az aks loadbalancer add -g MyResourceGroup -n restricted-lb --cluster-name MyManagedCluster --primary-agent-pool-name nodepool3 --allow-service-placement false
+"""
+
+helps['aks loadbalancer update'] = """
+    type: command
+    short-summary: Update a load balancer configuration in a managed Kubernetes cluster.
+    parameters:
+        - name: --name -n
+          type: string
+          short-summary: Name of the load balancer configuration to update.
+        - name: --primary-agent-pool-name -p
+          type: string
+          short-summary: Name of the primary agent pool for this load balancer.
+          long-summary: A string value that must specify the ID of an existing agent pool. All nodes in the given pool will always be added to this load balancer.
+        - name: --allow-service-placement -a
+          type: bool
+          short-summary: Whether to automatically place services on the load balancer.
+          long-summary: If set to false, both the external and internal load balancer will not be selected for services unless they explicitly target it.
+        - name: --service-label-selector -l
+          type: string
+          short-summary: Label selector for services that can be placed on this load balancer.
+          long-summary: Only services that match this selector can be placed on this load balancer. Format as comma-separated key=value pairs or expressions like "key In value1,value2".
+        - name: --service-namespace-selector -s
+          type: string
+          short-summary: Namespace label selector for services that can be placed on this load balancer.
+          long-summary: Services created in namespaces that match the selector can be placed on this load balancer. Format as comma-separated key=value pairs.
+        - name: --node-selector -d
+          type: string
+          short-summary: Node label selector for nodes that can be members of this load balancer.
+          long-summary: Nodes that match this selector will be possible members of this load balancer. Format as comma-separated key=value pairs.
+    examples:
+        - name: Update a load balancer configuration's primary agent pool
+          text: az aks loadbalancer update -g MyResourceGroup -n secondary --cluster-name MyManagedCluster --primary-agent-pool-name nodepool2
+        - name: Update a load balancer configuration to disable automatic service placement
+          text: az aks loadbalancer update -g MyResourceGroup -n app-lb --cluster-name MyManagedCluster --allow-service-placement false
+        - name: Update a load balancer configuration with new service selector
+          text: az aks loadbalancer update -g MyResourceGroup -n app-lb --cluster-name MyManagedCluster --service-label-selector tier=frontend,environment=production
+"""
+
+helps['aks loadbalancer delete'] = """
+    type: command
+    short-summary: Delete a load balancer configuration from a managed Kubernetes cluster.
+    parameters:
+        - name: --name -n
+          type: string
+          short-summary: Name of the load balancer configuration to delete.
+          long-summary: The "kubernetes" load balancer cannot be deleted as it's required for cluster operation.
+    examples:
+        - name: Delete a load balancer configuration
+          text: az aks loadbalancer delete -g MyResourceGroup -n secondary --cluster-name MyManagedCluster
+"""
+
+helps['aks loadbalancer list'] = """
+    type: command
+    short-summary: List all load balancer configurations in a managed Kubernetes cluster.
+    examples:
+        - name: List all load balancer configurations
+          text: az aks loadbalancer list -g MyResourceGroup --cluster-name MyManagedCluster
+        - name: List all load balancer configurations in table format
+          text: az aks loadbalancer list -g MyResourceGroup --cluster-name MyManagedCluster -o table
+"""
+
+helps['aks loadbalancer show'] = """
+    type: command
+    short-summary: Show details of a specific load balancer configuration in a managed Kubernetes cluster.
+    parameters:
+        - name: --name -n
+          type: string
+          short-summary: Name of the load balancer configuration to show.
+    examples:
+        - name: Show details of a specific load balancer configuration
+          text: az aks loadbalancer show -g MyResourceGroup -n secondary --cluster-name MyManagedCluster
+        - name: Show details of a load balancer configuration in table format
+          text: az aks loadbalancer show -g MyResourceGroup -n kubernetes --cluster-name MyManagedCluster -o table
+"""

src/aks-preview/azext_aks_preview/_params.py

@@ -2290,6 +2290,141 @@ def load_arguments(self, _):
                    help='Space-separated additional endpoint(s) to perform the connectivity check.',
                    validator=validate_custom_endpoints)
 
+    # AKS loadbalancer command parameter configuration
+    with self.argument_context("aks loadbalancer add") as c:
+        c.argument(
+            "name",
+            options_list=["--name", "-n"],
+            help="Name of the load balancer configuration. Required.",
+        )
+        c.argument(
+            "primary_agent_pool_name",
+            options_list=["--primary-agent-pool-name", "-p"],
+            help=(
+                "Name of the primary agent pool for this load balancer. "
+                "All nodes in this pool will be added to the load balancer. Required."
+            ),
+        )
+        c.argument(
+            "allow_service_placement",
+            options_list=["--allow-service-placement", "-a"],
+            arg_type=get_three_state_flag(),
+            help="Whether to automatically place services on the load balancer. Default is true.",
+        )
+        c.argument(
+            "service_label_selector",
+            options_list=["--service-label-selector", "-l"],
+            help=(
+                "Only services that match this selector can be placed on this load balancer. "
+                "Format: key1=value1,key2=value2 for simple selectors, "
+                "or key1 In val1 val2,key2 Exists for advanced expressions."
+            ),
+        )
+        c.argument(
+            "service_namespace_selector",
+            options_list=["--service-namespace-selector", "-s"],
+            help=(
+                "Services created in namespaces that match the selector can be placed on this load balancer. "
+                "Format: key1=value1,key2=value2 for simple selectors, "
+                "or key1 In val1 val2,key2 Exists for advanced expressions."
+            ),
+        )
+        c.argument(
+            "node_selector",
+            options_list=["--node-selector", "-d"],
+            help=(
+                "Nodes that match this selector will be possible members of this load balancer. "
+                "Format: key1=value1,key2=value2 for simple selectors, "
+                "or key1 In val1 val2,key2 Exists for advanced expressions."
+            ),
+        )
+
+    with self.argument_context("aks loadbalancer rebalance-nodes") as c:
+        c.argument(
+            "resource_group_name",
+            options_list=["--resource-group", "-g"],
+            help="Name of resource group.",
+            id_part="resource_group",
+            configured_default="aks",
+        )
+        c.argument(
+            "cluster_name",
+            options_list=["--name", "-n"],
+            help="Name of the managed cluster.",
+        )
+        c.argument(
+            "load_balancer_names",
+            options_list=["--load-balancer-names", "--lb-names"],
+            nargs="+",
+            help=(
+                "Space-separated list of load balancer names to rebalance. "
+                "If not specified, all load balancers will be rebalanced."
+            ),
+        )
+        c.argument(
+            "no_wait", help="Do not wait for the long-running operation to finish."
+        )
+
+    with self.argument_context("aks loadbalancer update") as c:
+        c.argument(
+            "name",
+            options_list=["--name", "-n"],
+            help="Name of the public load balancer. Required.",
+        )
+        c.argument(
+            "primary_agent_pool_name",
+            options_list=["--primary-agent-pool-name", "-p"],
+            help=(
+                "Name of the primary agent pool for this load balancer. "
+                "All nodes in this pool will be added to the load balancer."
+            ),
+        )
+        c.argument(
+            "allow_service_placement",
+            options_list=["--allow-service-placement", "-a"],
+            arg_type=get_three_state_flag(),
+            help="Whether to automatically place services on the load balancer. Default is true.",
+        )
+        c.argument(
+            "service_label_selector",
+            options_list=["--service-label-selector", "-l"],
+            help=(
+                "Only services that match this selector can be placed on this load balancer. "
+                "Format: key1=value1,key2=value2 for simple selectors, "
+                "or key1 In val1 val2,key2 Exists for advanced expressions."
+            ),
+        )
+        c.argument(
+            "service_namespace_selector",
+            options_list=["--service-namespace-selector", "-s"],
+            help=(
+                "Services created in namespaces that match the selector can be placed on this load balancer. "
+                "Format: key1=value1,key2=value2 for simple selectors, "
+                "or key1 In val1 val2,key2 Exists for advanced expressions."
+            ),
+        )
+        c.argument(
+            "node_selector",
+            options_list=["--node-selector", "-d"],
+            help=(
+                "Nodes that match this selector will be possible members of this load balancer. "
+                "Format: key1=value1,key2=value2 for simple selectors, "
+                "or key1 In val1 val2,key2 Exists for advanced expressions."
+            ),
+        )
+
+    # Define parameters for show and delete commands
+    for scope in [
+        "aks loadbalancer show",
+        "aks loadbalancer delete",
+    ]:
+        with self.argument_context(scope) as c:
+            c.argument(
+                "name",
+                options_list=["--name", "-n"],
+                help="Name of the load balancer configuration. Required.",
+            )
+
 
 def _get_default_install_location(exe_name):
     system = platform.system()

src/aks-preview/azext_aks_preview/commands.py

@@ -15,6 +15,7 @@
     cf_trustedaccess_role_binding,
     cf_machines,
     cf_operations,
+    cf_load_balancers,
 )
 from azext_aks_preview._format import (
     aks_addon_list_available_table_format,
@@ -196,6 +197,18 @@ def load_command_table(self, _):
         g.custom_command("update", "aks_maintenanceconfiguration_update")
         g.custom_command("delete", "aks_maintenanceconfiguration_delete")
 
+    # AKS loadbalancer commands
+    with self.command_group(
+        "aks loadbalancer",
+        client_factory=cf_load_balancers,
+    ) as g:
+        g.custom_command("list", "aks_loadbalancer_list")
+        g.custom_show_command("show", "aks_loadbalancer_show")
+        g.custom_command("add", "aks_loadbalancer_add")
+        g.custom_command("update", "aks_loadbalancer_update")
+        g.custom_command("delete", "aks_loadbalancer_delete")
+        g.custom_command("rebalance-nodes", "aks_loadbalancer_rebalance_nodes")
+
     # AKS addon commands
     with self.command_group(
         "aks addon", managed_clusters_sdk, client_factory=cf_managed_clusters