[sql] az sql midb update: Add update command

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

@@ -920,6 +920,14 @@
     text: az sql midb create -g mygroup --mi myinstance -n mymanageddb --collation Latin1_General_100_CS_AS_SC
 """
 
+helps['sql midb update'] = """
+type: command
+short-summary: Update a managed database.
+examples:
+  - name: Update a managed database with specified tags
+    text: az sql midb update -g mygroup --mi myinstance -n mymanageddb --tags tag1="value1"
+"""
+
 helps['sql midb delete'] = """
 type: command
 short-summary: Delete a managed database.

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

@@ -2099,13 +2099,24 @@ def _configure_security_policy_storage_params(arg_ctx):
         create_args_for_complex_type(
             c, 'parameters', ManagedDatabase, [
                 'collation',
+                'tags',
             ])
 
+        c.argument('tags', arg_type=tags_type)
+
         c.argument('collation',
                    required=False,
                    help='The collation of the Azure SQL Managed Database collation to use, '
                    'e.g.: SQL_Latin1_General_CP1_CI_AS or Latin1_General_100_CS_AS_SC')
 
+    with self.argument_context('sql midb update') as c:
+        create_args_for_complex_type(
+            c, 'parameters', ManagedDatabase, [
+                'tags',
+            ])
+
+        c.argument('tags', arg_type=tags_type)
+
     with self.argument_context('sql midb restore') as c:
         create_args_for_complex_type(
             c, 'parameters', ManagedDatabase, [

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

@@ -765,6 +765,10 @@ def load_command_table(self, _):
                             client_factory=get_sql_managed_databases_operations) as g:
 
         g.custom_command('create', 'managed_db_create', supports_no_wait=True)
+        g.generic_update_command('update',
+                                 setter_name='begin_create_or_update',
+                                 custom_func_name='managed_db_update',
+                                 supports_no_wait=True)
         g.custom_command('restore', 'managed_db_restore', supports_no_wait=True)
         g.show_command('show', 'get')
         g.command('list', 'list_by_instance')

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

@@ -4688,6 +4688,16 @@ def managed_db_create(
         parameters=kwargs)
 
 
+def managed_db_update(
+        instance,
+        tags=None):
+
+    if tags is not None:
+        instance.tags = tags
+
+    return instance
+
+
 def managed_db_restore(
         cmd,
         client,

src/azure-cli/azure/cli/command_modules/sql/tests/latest/test_sql_commands.py

@@ -5164,6 +5164,18 @@ def test_sql_managed_db_mgmt(self, mi, rg):
                  .format(resource_group_1, managed_instance_name_1),
                  checks=[JMESPathCheck('length(@)', 2)])
 
+        self.cmd('sql midb update -g {} --managed-instance {} -n {} --tags {}'
+                 .format(resource_group_1, managed_instance_name_1, database_name, "bar=foo"),
+                 checks=[JMESPathCheck('tags', "{'bar': 'foo'}")])
+
+        # test merge managed database tags
+        tag3 = "tagName3=tagValue3"
+        self.cmd('sql midb update -g {} --managed-instance {} -n {} --set tags.{}'
+                 .format(resource_group_1, managed_instance_name_1, database_name, tag3),
+                 checks=[
+                     JMESPathCheck('tags',
+                                   "{'bar': 'foo', 'tagName3': 'tagValue3'}")])
+
         # Show by group/managed_instance/database-name
         self.cmd('sql midb show -g {} --managed-instance {} -n {}'
                  .format(resource_group_1, managed_instance_name_1, database_name),