feat(dms): add new resource to restart nodes

Author: Zippo-Wang

docs/resources/dms_rocketmq_instance_restart.md

@@ -0,0 +1,55 @@
+---
+subcategory: "Distributed Message Service (DMS)"
+layout: "huaweicloud"
+page_title: "HuaweiCloud: huaweicloud_dms_rocketmq_instance_restart"
+description: |-
+  Restart a DMS RocketMQ instance resource within HuaweiCloud.
+---
+
+# huaweicloud_dms_rocketmq_instance_restart
+
+Restart a DMS RocketMQ instance resource within HuaweiCloud.
+
+## Example Usage
+
+```hcl
+variable "instance_id" {}
+
+data "huaweicloud_dms_rocketmq_instance_nodes" "test" {
+  instance_id = var.instance_id
+}
+
+resource "huaweicloud_dms_rocketmq_instance_restart" "test" {
+  instance_id = var.instance_id
+
+  nodes = [
+    data.huaweicloud_dms_rocketmq_instance_nodes.test.nodes[0].id,
+    data.huaweicloud_dms_rocketmq_instance_nodes.test.nodes[1].id
+  ]
+}
+```
+
+## Argument Reference
+
+The following arguments are supported:
+
+* `region` - (Optional, String, ForceNew) Specifies the region in which to restart the RocketMQ instance.  
+  If omitted, the provider-level region will be used. Changing this parameter will create a new resource.
+
+* `instance_id` - (Required, String, NonUpdatable) Specifies the ID of the RocketMQ instance to be restarted.  
+
+* `nodes` - (Required, List, NonUpdatable) Specifies the list of node IDs to be restarted.  
+
+## Attribute Reference
+
+In addition to all arguments above, the following attributes are exported:
+
+* `id` - The resource ID.
+
+## Import
+
+The RocketMQ instance restart resource cannot be imported.
+
+## Timeouts
+
+This resource does not support timeout configuration.

huaweicloud/provider.go

@@ -2265,6 +2265,7 @@ func Provider() *schema.Provider {
 			"huaweicloud_dms_rabbitmq_user":                   rabbitmq.ResourceDmsRabbitmqUser(),
 
 			"huaweicloud_dms_rocketmq_instance":             rocketmq.ResourceDmsRocketMQInstance(),
+			"huaweicloud_dms_rocketmq_instance_restart":     rocketmq.ResourceDmsRocketMQInstanceRestart(),
 			"huaweicloud_dms_rocketmq_consumer_group":       rocketmq.ResourceDmsRocketMQConsumerGroup(),
 			"huaweicloud_dms_rocketmq_consumption_verify":   rocketmq.ResourceDmsRocketMQConsumptionVerify(),
 			"huaweicloud_dms_rocketmq_message_offset_reset": rocketmq.ResourceDmsRocketMQMessageOffsetReset(),

huaweicloud/services/acceptance/rocketmq/data_source_huaweicloud_dms_rocketmq_instance_restart_test.go

@@ -0,0 +1,50 @@
+package rocketmq
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+
+	"github.com/huaweicloud/terraform-provider-huaweicloud/huaweicloud/services/acceptance"
+)
+
+func TestAccDmsRocketMQInstanceRestart_basic(t *testing.T) {
+	var (
+		dataSourceName = "huaweicloud_dms_rocketmq_instance_restart.test"
+		instanceID     = acceptance.HW_DMS_ROCKETMQ_INSTANCE_ID
+	)
+
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck: func() {
+			acceptance.TestAccPreCheck(t)
+			acceptance.TestAccPreCheckDMSRocketMQInstanceID(t)
+		},
+		ProviderFactories: acceptance.TestAccProviderFactories,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccDmsRocketMQInstanceRestart_basic(instanceID),
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttr(dataSourceName, "instance_id", instanceID),
+					resource.TestCheckResourceAttr(dataSourceName, "nodes.#", "2"),
+					resource.TestCheckResourceAttrSet(dataSourceName, "nodes.0"),
+				),
+			},
+		},
+	})
+}
+
+func testAccDmsRocketMQInstanceRestart_basic(instanceID string) string {
+	return fmt.Sprintf(`
+%s
+
+resource "huaweicloud_dms_rocketmq_instance_restart" "test" {
+  instance_id = data.huaweicloud_dms_rocketmq_instance_nodes.test.instance_id
+
+  nodes = [
+    try(data.huaweicloud_dms_rocketmq_instance_nodes.test.nodes[0].id),
+    try(data.huaweicloud_dms_rocketmq_instance_nodes.test.nodes[1].id)
+  ]
+}
+`, testAccDmsRocketMQInstanceNodes_basic(instanceID))
+}

huaweicloud/services/rocketmq/resource_huaweicloud_dms_rocketmq_instance_restart.go

@@ -0,0 +1,123 @@
+package rocketmq
+
+import (
+	"context"
+	"strings"
+
+	"github.com/hashicorp/go-uuid"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/customdiff"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/validation"
+
+	"github.com/chnsz/golangsdk"
+
+	"github.com/huaweicloud/terraform-provider-huaweicloud/huaweicloud/config"
+	"github.com/huaweicloud/terraform-provider-huaweicloud/huaweicloud/utils"
+)
+
+var instanceRestartNonUpdatableParams = []string{"instance_id", "nodes"}
+
+// @API RocketMQ POST /v2/{project_id}/{engine}/instances/{instance_id}/restart
+func ResourceDmsRocketMQInstanceRestart() *schema.Resource {
+	return &schema.Resource{
+		CreateContext: resourceDmsRocketMQInstanceRestartCreate,
+		ReadContext:   resourceDmsRocketMQInstanceRestartRead,
+		DeleteContext: resourceDmsRocketMQInstanceRestartDelete,
+
+		CustomizeDiff: customdiff.All(
+			config.FlexibleForceNew(instanceRestartNonUpdatableParams),
+		),
+
+		Schema: map[string]*schema.Schema{
+			"region": {
+				Type:        schema.TypeString,
+				Optional:    true,
+				ForceNew:    true,
+				Computed:    true,
+				Description: `The region where the RocketMQ instance is located.`,
+			},
+			"instance_id": {
+				Type:        schema.TypeString,
+				Required:    true,
+				Description: `The ID of the RocketMQ instance to be restarted.`,
+			},
+			"nodes": {
+				Type:        schema.TypeList,
+				Required:    true,
+				Elem:        &schema.Schema{Type: schema.TypeString},
+				Description: `The list of node names to be restarted.`,
+			},
+
+			// Internal
+			"enable_force_new": {
+				Type:         schema.TypeString,
+				Optional:     true,
+				ValidateFunc: validation.StringInSlice([]string{"true", "false"}, false),
+				Description:  utils.SchemaDesc("", utils.SchemaDescInput{Internal: true}),
+			},
+		},
+	}
+}
+
+func resourceDmsRocketMQInstanceRestartCreate(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
+	cfg := meta.(*config.Config)
+	region := cfg.GetRegion(d)
+	client, err := cfg.NewServiceClient("dmsv2", region)
+	if err != nil {
+		return diag.Errorf("error creating DMS client: %s", err)
+	}
+
+	instanceId := d.Get("instance_id").(string)
+
+	httpUrl := "v2/{project_id}/rocketmq/instances/{instance_id}/restart"
+	httpUrl = strings.ReplaceAll(httpUrl, "{project_id}", client.ProjectID)
+	httpUrl = strings.ReplaceAll(httpUrl, "{instance_id}", instanceId)
+	restartPath := client.Endpoint + httpUrl
+
+	opt := golangsdk.RequestOpts{
+		KeepResponseBody: true,
+		MoreHeaders: map[string]string{
+			"Content-Type": "application/json",
+		},
+	}
+
+	// Add nodes parameter to request body if specified
+	if nodesRaw := d.Get("nodes").([]interface{}); len(nodesRaw) > 0 {
+		nodes := make([]string, len(nodesRaw))
+		for i, node := range nodesRaw {
+			nodes[i] = node.(string)
+		}
+		opt.JSONBody = map[string]interface{}{
+			"nodes": nodes,
+		}
+	}
+
+	_, err = client.Request("POST", restartPath, &opt)
+	if err != nil {
+		return diag.Errorf("error restarting RocketMQ instance: %s", err)
+	}
+
+	randUUID, err := uuid.GenerateUUID()
+	if err != nil {
+		return diag.Errorf("unable to generate resource ID: %s", err)
+	}
+	d.SetId(randUUID)
+
+	return resourceDmsRocketMQInstanceRestartRead(ctx, d, meta)
+}
+
+func resourceDmsRocketMQInstanceRestartRead(_ context.Context, _ *schema.ResourceData, _ interface{}) diag.Diagnostics {
+	return nil
+}
+
+func resourceDmsRocketMQInstanceRestartDelete(_ context.Context, _ *schema.ResourceData, _ interface{}) diag.Diagnostics {
+	errorMsg := `This resource is a one-time action resource used to restart RocketMQ instance. Deleting this resource will
+not clear the restart operation record, but will only remove the resource information from the tfstate file.`
+	return diag.Diagnostics{
+		diag.Diagnostic{
+			Severity: diag.Warning,
+			Summary:  errorMsg,
+		},
+	}
+}