[SQL] az sql midb log-replay: Support for log replay service on managed database

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

@@ -895,6 +895,55 @@
 short-summary: Place the CLI in a waiting state until a condition of the managed database is met.
 """
 
+helps['sql midb log-replay'] = """
+type: group
+short-summary: SQL Managed Instance database Log Replay service commands.
+"""
+
+helps['sql midb log-replay start'] = """
+type: command
+short-summary: Start Log Replay service on specified database.
+examples:
+  - name: Start Log Replay service with auto complete option.
+    text: az sql midb log-replay start -g mygroup --mi myinstance -n mymanageddb -a --last-bn "backup.bak" --storage-uri "https://test.blob.core.windows.net/testing" --storage-sas "sv=2019-02-02&ss=b&srt=sco&sp=rl&se=2023-12-02T00:09:14Z&st=2019-11-25T16:09:14Z&spr=https&sig=92kAe4QYmXaht%2Fgjocqwerqwer41s%3D"
+  - name: Start Log Replay service without auto complete option.
+    text: az sql midb log-replay start -g mygroup --mi myinstance -n mymanageddb --storage-uri "https://test.blob.core.windows.net/testing" --storage-sas "sv=2019-02-02&ss=b&srt=sco&sp=rl&se=2023-12-02T00:09:14Z&st=2019-11-25T16:09:14Z&spr=https&sig=92kAe4QYmXaht%2Fgjocqwerqwer41s%3D"
+"""
+
+helps['sql midb log-replay complete'] = """
+type: command
+short-summary: Complete Log Replay service on specified database.
+examples:
+  - name: Complete log replay service.
+    text: az sql midb log-replay complete -g mygroup --mi myinstance -n mymanageddb --last-backup-name "backup.bak"
+"""
+
+helps['sql midb log-replay show'] = """
+type: command
+short-summary: Get status of Log Replay service.
+examples:
+  - name: Get status of the ongoing log replay service.
+    text: az sql midb log-replay show -g mygroup --mi myinstance -n mymanageddb
+"""
+
+helps['sql midb log-replay stop'] = """
+type: command
+short-summary: Stop Log Replay service.
+examples:
+  - name: Stop ongoing log replay service by deleting database.
+    text: az sql midb log-replay stop -g mygroup --mi myinstance -n mymanageddb
+"""
+
+helps['sql midb log-replay wait'] = """
+type: command
+short-summary: Place the CLI in a waiting state until a condition of the managed database is met.
+examples:
+  - name: Place the CLI in a waiting state until a condition of the managed database is met after starting log replay service.
+    text: az sql midb log-replay wait --exists --resource-group mygroup --managed-instance myinstance --name mymanageddb
+  - name: Place the CLI in a waiting state until a condition of the managed database is met after stopping log replay service.
+    text: az sql midb log-replay wait --deleted --resource-group mygroup --managed-instance myinstance --name mymanageddb
+"""
+
 helps['sql server'] = """
 type: group
 short-summary: Manage SQL servers.

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

@@ -1839,6 +1839,47 @@ def _configure_security_policy_storage_params(arg_ctx):
                    help='The resource id of the long term retention backup to be restored. '
                    'Use \'az sql midb ltr-backup show\' or \'az sql midb ltr-backup list\' for backup id.')
 
+    with self.argument_context('sql midb log-replay start') as c:
+        create_args_for_complex_type(
+            c, 'parameters', ManagedDatabase, [
+                'auto_complete',
+                'last_backup_name',
+                'storage_container_uri',
+                'storage_container_sas_token'
+            ])
+
+        c.argument('auto_complete',
+                   required=False,
+                   options_list=['--auto-complete', '-a'],
+                   action='store_true',
+                   help='The flag that in usage with last_backup_name automatically completes log replay servise.')
+
+        c.argument('last_backup_name',
+                   required=False,
+                   options_list=['--last-backup-name', '--last-bn'],
+                   help='The name of the last backup to restore.')
+
+        c.argument('storage_container_uri',
+                   required=True,
+                   options_list=['--storage-uri', '--su'],
+                   help='The URI of the storage container where backups are.')
+
+        c.argument('storage_container_sas_token',
+                   required=True,
+                   options_list=['--storage-sas', '--ss'],
+                   help='The authorization Sas token to access storage container where backups are.')
+
+    with self.argument_context('sql midb log-replay complete') as c:
+        create_args_for_complex_type(
+            c, 'parameters', ManagedDatabase, [
+                'last_backup_name'
+            ])
+
+        c.argument('last_backup_name',
+                   required=False,
+                   options_list=['--last-backup-name', '--last-bn'],
+                   help='The name of the last backup to restore.')
+
     ###############################################
     #                sql virtual cluster          #
     ###############################################

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

@@ -194,3 +194,7 @@ def get_sql_database_long_term_retention_policies_operations(cli_ctx, _):
 
 def get_sql_database_long_term_retention_backups_operations(cli_ctx, _):
     return get_sql_management_client(cli_ctx).long_term_retention_backups
+
+
+def get_sql_managed_database_restore_details_operations(cli_ctx, _):
+    return get_sql_management_client(cli_ctx).managed_database_restore_details

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

@@ -42,6 +42,7 @@
     get_sql_firewall_rules_operations,
     get_sql_instance_pools_operations,
     get_sql_managed_databases_operations,
+    get_sql_managed_database_restore_details_operations,
     get_sql_managed_backup_short_term_retention_policies_operations,
     get_sql_managed_database_long_term_retention_policies_operations,
     get_sql_managed_database_long_term_retention_backups_operations,
@@ -703,6 +704,24 @@ def load_command_table(self, _):
             is_preview=True)
         g.wait_command('wait')
 
+    with self.command_group('sql midb log-replay',
+                            managed_databases_operations,
+                            client_factory=get_sql_managed_databases_operations) as g:
+        g.custom_command('start', 'managed_db_log_replay_start', supports_no_wait=True)
+        g.command('stop', 'delete', confirmation=True, supports_no_wait=True)
+        g.command('complete', 'complete_restore')
+        g.wait_command('wait')
+
+    managed_database_restore_details_operations = CliCommandType(
+        operations_tmpl='azure.mgmt.sql.operations#ManagedDatabaseRestoreDetailsOperations.{}',
+        client_factory=get_sql_managed_database_restore_details_operations)
+
+    with self.command_group('sql midb log-replay',
+                            managed_database_restore_details_operations,
+                            client_factory=get_sql_managed_database_restore_details_operations) as g:
+
+        g.show_command('show', 'get')
+
     ###############################################
     #                sql virtual cluster         #
     ###############################################

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

@@ -3359,6 +3359,43 @@ def restore_long_term_retention_mi_backup(
         resource_group_name=target_resource_group_name,
         parameters=kwargs)
 
+
+def managed_db_log_replay_start(
+        cmd,
+        client,
+        database_name,
+        managed_instance_name,
+        resource_group_name,
+        auto_complete,
+        last_backup_name,
+        storage_container_uri,
+        storage_container_sas_token,
+        **kwargs):
+
+    # Determine managed instance location
+    kwargs['location'] = _get_managed_instance_location(
+        cmd.cli_ctx,
+        managed_instance_name=managed_instance_name,
+        resource_group_name=resource_group_name)
+
+    kwargs['create_mode'] = CreateMode.restore_external_backup.value
+
+    if auto_complete and not last_backup_name:
+        raise CLIError('Please specify a last backup name when using auto complete flag.')
+
+    kwargs['auto_complete'] = auto_complete
+    kwargs['last_backup_name'] = last_backup_name
+
+    kwargs['storageContainerUri'] = storage_container_uri
+    kwargs['storageContainerSasToken'] = storage_container_sas_token
+
+    # Create
+    return client.create_or_update(
+        database_name=database_name,
+        managed_instance_name=managed_instance_name,
+        resource_group_name=resource_group_name,
+        parameters=kwargs)
+
 ###############################################
 #              sql failover-group             #
 ###############################################