{RDBMS} az postgres flexible-server virtual-endpoint create: Add the name check to see if virtual endpoint name is available

Author: Matthew Boentoro

src/azure-cli/azure/cli/command_modules/rdbms/flexible_server_custom_postgres.py

@@ -38,7 +38,8 @@
 from .flexible_server_virtual_network import prepare_private_network, prepare_private_dns_zone, prepare_public_network
 from .validators import pg_arguments_validator, validate_server_name, validate_and_format_restore_point_in_time, \
     validate_postgres_replica, validate_georestore_network, pg_byok_validator, validate_migration_runtime_server, \
-    validate_resource_group, check_resource_group, validate_citus_cluster, cluster_byok_validator, validate_backup_name
+    validate_resource_group, check_resource_group, validate_citus_cluster, cluster_byok_validator, validate_backup_name, \
+    validate_virtual_endpoint_name_availability
 
 logger = get_logger(__name__)
 DEFAULT_DB_NAME = 'flexibleserverdb'
@@ -1375,6 +1376,7 @@ def migration_check_name_availability(cmd, client, resource_group_name, server_n
 def virtual_endpoint_create_func(cmd, client, resource_group_name, server_name, virtual_endpoint_name, endpoint_type, members):
     validate_resource_group(resource_group_name)
     validate_citus_cluster(cmd, resource_group_name, server_name)
+    validate_virtual_endpoint_name_availability(cmd, virtual_endpoint_name)
 
     parameters = {
         'name': virtual_endpoint_name,

src/azure-cli/azure/cli/command_modules/rdbms/validators.py

@@ -19,7 +19,7 @@
 from azure.core.exceptions import HttpResponseError
 from azure.mgmt.rdbms.mysql_flexibleservers.operations._firewall_rules_operations import FirewallRulesOperations \
     as MySqlFirewallRulesOperations
-from ._client_factory import cf_mysql_flexible_servers, cf_postgres_flexible_servers
+from ._client_factory import cf_mysql_flexible_servers, cf_postgres_flexible_servers, cf_postgres_check_resource_availability
 from ._flexible_server_util import (get_mysql_versions, get_mysql_skus, get_mysql_storage_size,
                                     get_mysql_backup_retention, get_mysql_tiers, get_mysql_list_skus_info,
                                     get_postgres_skus, get_postgres_storage_sizes, get_postgres_tiers,
@@ -729,6 +729,12 @@ def validate_server_name(db_context, server_name, type_):
     if not result.name_available:
         raise ValidationError(result.message)
 
+def validate_virtual_endpoint_name_availability(cmd, virtual_endpoint_name):
+    client = cf_postgres_check_resource_availability(cmd.cli_ctx, '_')
+    result = client.execute(name_availability_request={'name': virtual_endpoint_name, 'type': 'Microsoft.DBforPostgreSQL/flexibleServers/virtualendpoints'})
+    if result and result.name_available is False:
+        raise ValidationError("Virtual endpoints base name is not available.")
+
 
 def validate_migration_runtime_server(cmd, migrationInstanceResourceId, target_resource_group_name, target_server_name):
     id_comps = parse_resource_id(migrationInstanceResourceId)