[SQL] az sql midb log-replay start: Add --storage-identity parameter

src/azure-cli/azure/cli/command_modules/sql/_params.py

@@ -2131,6 +2131,10 @@ def _configure_security_policy_storage_params(arg_ctx):
     ###############################################
     #                sql managed db               #
     ###############################################
+    class ContainerIdentityType(Enum):
+        managed_identity = "ManagedIdentity"
+        sas = "SharedAccessSignature"
+
     with self.argument_context('sql midb') as c:
         c.argument('managed_instance_name',
                    arg_type=managed_instance_param_type,
@@ -2340,7 +2344,8 @@ def _configure_security_policy_storage_params(arg_ctx):
                 'auto_complete',
                 'last_backup_name',
                 'storage_container_uri',
-                'storage_container_sas_token'
+                'storage_container_sas_token',
+                'storage_container_identity'
             ])
 
         c.argument('auto_complete',
@@ -2364,6 +2369,12 @@ def _configure_security_policy_storage_params(arg_ctx):
                    options_list=['--storage-sas', '--ss'],
                    help='The authorization Sas token to access storage container where backups are.')
 
+        c.argument('storage_container_identity',
+                   arg_type=get_enum_type(ContainerIdentityType),
+                   required=False,
+                   options_list=['--storage-identity', '--si'],
+                   help='The storage container identity to use.')
+
     with self.argument_context('sql midb log-replay complete') as c:
         create_args_for_complex_type(
             c, 'parameters', ManagedDatabase, [

src/azure-cli/azure/cli/command_modules/sql/custom.py

@@ -5165,6 +5165,7 @@ def managed_db_log_replay_start(
         last_backup_name,
         storage_container_uri,
         storage_container_sas_token,
+        storage_container_identity,
         **kwargs):
     '''
     Start a log replay restore.
@@ -5186,6 +5187,7 @@ def managed_db_log_replay_start(
 
     kwargs['storageContainerUri'] = storage_container_uri
     kwargs['storageContainerSasToken'] = storage_container_sas_token
+    kwargs['storage_container_identity'] = storage_container_identity
 
     # Create
     return client.begin_create_or_update(