Feature Gap: Implement resourcePolicies.workloadPolicy for MIG (#14255) (#10265)

[upstream:587caf7f4a7983a91bad804d20f828948b092f72]

Signed-off-by: Modular Magician <[email protected]>

Author: modular-magician

.changelog/14255.txt

@@ -0,0 +1,3 @@
+```release-note:enhancement
+compute: added `resource_policies.0.workload_policy` to `google_compute_instance_group_manager` resource (ga).
+```

google-beta/services/compute/resource_compute_instance_group_manager.go

@@ -17,6 +17,7 @@
 package compute
 
 import (
+	"context"
 	"fmt"
 	"log"
 	"strings"
@@ -50,6 +51,7 @@ func ResourceComputeInstanceGroupManager() *schema.Resource {
 		CustomizeDiff: customdiff.All(
 			tpgresource.DefaultProviderProject,
 			tpgresource.DefaultProviderZone,
+			customdiff.ForceNewIfChange("resource_policies.0.workload_policy", ForceNewResourcePoliciesWorkloadPolicyIfNewIsEmpty),
 		),
 		Schema: map[string]*schema.Schema{
 			"base_instance_name": {
@@ -585,11 +587,31 @@ func ResourceComputeInstanceGroupManager() *schema.Resource {
 					},
 				},
 			},
+			"resource_policies": {
+				Type:        schema.TypeList,
+				Optional:    true,
+				Description: `Resource policies for this managed instance group.`,
+				MaxItems:    1,
+				Elem: &schema.Resource{
+					Schema: map[string]*schema.Schema{
+						"workload_policy": {
+							Type:             schema.TypeString,
+							Optional:         true,
+							DiffSuppressFunc: tpgresource.CompareSelfLinkRelativePaths,
+							Description:      `The URL of the workload policy that is specified for this managed instance group. It can be a full or partial URL.`,
+						},
+					},
+				},
+			},
 		},
 		UseJSONNumber: true,
 	}
 }
 
+func ForceNewResourcePoliciesWorkloadPolicyIfNewIsEmpty(_ context.Context, old, new, _ interface{}) bool {
+	return (old.(string) != "") && (new.(string) == "")
+}
+
 func parseUniqueId(s string) (string, string) {
 	splits := strings.SplitN(s, "?uniqueId=", 2)
 	if len(splits) == 2 {
@@ -678,6 +700,7 @@ func resourceComputeInstanceGroupManagerCreate(d *schema.ResourceData, meta inte
 		InstanceLifecyclePolicy:     expandInstanceLifecyclePolicy(d.Get("instance_lifecycle_policy").([]interface{})),
 		AllInstancesConfig:          expandAllInstancesConfig(nil, d.Get("all_instances_config").([]interface{})),
 		StatefulPolicy:              expandStatefulPolicy(d),
+		ResourcePolicies:            expandResourcePolicies(d.Get("resource_policies").([]interface{})),
 		Params:                      expandInstanceGroupManagerParams(d),
 
 		// Force send TargetSize to allow a value of 0.
@@ -923,6 +946,9 @@ func resourceComputeInstanceGroupManagerRead(d *schema.ResourceData, meta interf
 	if err = d.Set("status", flattenStatus(manager.Status)); err != nil {
 		return fmt.Errorf("Error setting status in state: %s", err.Error())
 	}
+	if err = d.Set("resource_policies", flattenResourcePolicies(manager.ResourcePolicies)); err != nil {
+		return fmt.Errorf("Error setting resource_policies in state: %s", err.Error())
+	}
 
 	// If unset in state set to default value
 	if d.Get("wait_for_instances_status").(string) == "" {
@@ -1027,6 +1053,11 @@ func resourceComputeInstanceGroupManagerUpdate(d *schema.ResourceData, meta inte
 		change = true
 	}
 
+	if d.HasChange("resource_policies") {
+		updatedManager.ResourcePolicies = expandResourcePolicies(d.Get("resource_policies").([]interface{}))
+		change = true
+	}
+
 	if change {
 		op, err := config.NewComputeClient(userAgent).InstanceGroupManagers.Patch(project, zone, d.Get("name").(string), updatedManager).Do()
 		if err != nil {
@@ -1293,6 +1324,20 @@ func expandVersions(configured []interface{}) []*compute.InstanceGroupManagerVer
 	return versions
 }
 
+func expandResourcePolicies(configured []interface{}) *compute.InstanceGroupManagerResourcePolicies {
+	resourcePolicies := &compute.InstanceGroupManagerResourcePolicies{}
+
+	if len(configured) > 0 {
+		data := configured[0].(map[string]interface{})
+		resourcePolicies.WorkloadPolicy = data["workload_policy"].(string)
+		resourcePolicies.ForceSendFields = []string{"WorkloadPolicy"}
+	} else {
+		resourcePolicies.NullFields = []string{"WorkloadPolicy"}
+	}
+
+	return resourcePolicies
+}
+
 func expandFixedOrPercent(configured []interface{}) *compute.FixedOrPercent {
 	fixedOrPercent := &compute.FixedOrPercent{}
 
@@ -1624,6 +1669,17 @@ func flattenStatusAllInstancesConfig(allInstancesConfig *compute.InstanceGroupMa
 	return results
 }
 
+func flattenResourcePolicies(resourcePolicies *compute.InstanceGroupManagerResourcePolicies) []map[string]interface{} {
+	results := []map[string]interface{}{}
+	if resourcePolicies != nil {
+		data := map[string]interface{}{
+			"workload_policy": resourcePolicies.WorkloadPolicy,
+		}
+		results = append(results, data)
+	}
+	return results
+}
+
 func resourceInstanceGroupManagerStateImporter(d *schema.ResourceData, meta interface{}) ([]*schema.ResourceData, error) {
 	if err := d.Set("wait_for_instances", false); err != nil {
 		return nil, fmt.Errorf("Error setting wait_for_instances: %s", err)