App Engine Service Split Traffic resource

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

Author: khanali21

google/provider.go

@@ -437,9 +437,9 @@ func Provider() terraform.ResourceProvider {
 	return provider
 }
 
-// Generated resources: 80
+// Generated resources: 81
 // Generated IAM resources: 33
-// Total generated resources: 113
+// Total generated resources: 114
 func ResourceMap() map[string]*schema.Resource {
 	resourceMap, _ := ResourceMapWithErrors()
 	return resourceMap
@@ -453,6 +453,7 @@ func ResourceMapWithErrors() (map[string]*schema.Resource, error) {
 			"google_access_context_manager_service_perimeter":  resourceAccessContextManagerServicePerimeter(),
 			"google_app_engine_domain_mapping":                 resourceAppEngineDomainMapping(),
 			"google_app_engine_firewall_rule":                  resourceAppEngineFirewallRule(),
+			"google_app_engine_service_split_traffic":          resourceAppEngineServiceSplitTraffic(),
 			"google_app_engine_standard_app_version":           resourceAppEngineStandardAppVersion(),
 			"google_bigquery_dataset":                          resourceBigQueryDataset(),
 			"google_bigquery_data_transfer_config":             resourceBigqueryDataTransferConfig(),

google/resource_app_engine_service_split_traffic.go

@@ -0,0 +1,352 @@
+// ----------------------------------------------------------------------------
+//
+//     ***     AUTO GENERATED CODE    ***    AUTO GENERATED CODE     ***
+//
+// ----------------------------------------------------------------------------
+//
+//     This file is automatically generated by Magic Modules and manual
+//     changes will be clobbered when the file is regenerated.
+//
+//     Please read more about how to change this file in
+//     .github/CONTRIBUTING.md.
+//
+// ----------------------------------------------------------------------------
+
+package google
+
+import (
+	"fmt"
+	"log"
+	"reflect"
+	"strings"
+	"time"
+
+	"github.com/hashicorp/terraform-plugin-sdk/helper/schema"
+	"github.com/hashicorp/terraform-plugin-sdk/helper/validation"
+	"google.golang.org/api/appengine/v1"
+)
+
+func resourceAppEngineServiceSplitTraffic() *schema.Resource {
+	return &schema.Resource{
+		Create: resourceAppEngineServiceSplitTrafficCreate,
+		Read:   resourceAppEngineServiceSplitTrafficRead,
+		Update: resourceAppEngineServiceSplitTrafficUpdate,
+		Delete: resourceAppEngineServiceSplitTrafficDelete,
+
+		Importer: &schema.ResourceImporter{
+			State: resourceAppEngineServiceSplitTrafficImport,
+		},
+
+		Timeouts: &schema.ResourceTimeout{
+			Create: schema.DefaultTimeout(4 * time.Minute),
+			Update: schema.DefaultTimeout(4 * time.Minute),
+			Delete: schema.DefaultTimeout(4 * time.Minute),
+		},
+
+		Schema: map[string]*schema.Schema{
+			"split": {
+				Type:     schema.TypeList,
+				Required: true,
+				MaxItems: 1,
+				Elem: &schema.Resource{
+					Schema: map[string]*schema.Schema{
+						"allocations": {
+							Type:     schema.TypeMap,
+							Required: true,
+							Elem:     &schema.Schema{Type: schema.TypeString},
+						},
+						"shard_by": {
+							Type:         schema.TypeString,
+							Optional:     true,
+							ValidateFunc: validation.StringInSlice([]string{"UNSPECIFIED", "COOKIE", "IP", "RANDOM", ""}, false),
+						},
+					},
+				},
+			},
+			"migrate_traffic": {
+				Type:     schema.TypeBool,
+				Optional: true,
+			},
+			"name": {
+				Type:     schema.TypeString,
+				Optional: true,
+			},
+			"service_id": {
+				Type:     schema.TypeString,
+				Optional: true,
+			},
+			"project": {
+				Type:     schema.TypeString,
+				Optional: true,
+				Computed: true,
+				ForceNew: true,
+			},
+		},
+	}
+}
+
+func resourceAppEngineServiceSplitTrafficCreate(d *schema.ResourceData, meta interface{}) error {
+	config := meta.(*Config)
+
+	obj := make(map[string]interface{})
+	nameProp, err := expandAppEngineServiceSplitTrafficName(d.Get("name"), d, config)
+	if err != nil {
+		return err
+	} else if v, ok := d.GetOkExists("name"); !isEmptyValue(reflect.ValueOf(nameProp)) && (ok || !reflect.DeepEqual(v, nameProp)) {
+		obj["name"] = nameProp
+	}
+	idProp, err := expandAppEngineServiceSplitTrafficServiceId(d.Get("service_id"), d, config)
+	if err != nil {
+		return err
+	} else if v, ok := d.GetOkExists("service_id"); !isEmptyValue(reflect.ValueOf(idProp)) && (ok || !reflect.DeepEqual(v, idProp)) {
+		obj["id"] = idProp
+	}
+	splitProp, err := expandAppEngineServiceSplitTrafficSplit(d.Get("split"), d, config)
+	if err != nil {
+		return err
+	} else if v, ok := d.GetOkExists("split"); !isEmptyValue(reflect.ValueOf(splitProp)) && (ok || !reflect.DeepEqual(v, splitProp)) {
+		obj["split"] = splitProp
+	}
+
+	lockName, err := replaceVars(d, config, "apps/{{project}}/services/{{service_id}}")
+	if err != nil {
+		return err
+	}
+	mutexKV.Lock(lockName)
+	defer mutexKV.Unlock(lockName)
+
+	url, err := replaceVars(d, config, "{{AppEngineBasePath}}apps/{{project}}/services/{{service_id}}?migrateTraffic={{migrate_traffic}}&updateMask=split")
+	if err != nil {
+		return err
+	}
+
+	log.Printf("[DEBUG] Creating new ServiceSplitTraffic: %#v", obj)
+	project, err := getProject(d, config)
+	if err != nil {
+		return err
+	}
+	res, err := sendRequestWithTimeout(config, "PATCH", project, url, obj, d.Timeout(schema.TimeoutCreate))
+	if err != nil {
+		return fmt.Errorf("Error creating ServiceSplitTraffic: %s", err)
+	}
+
+	// Store the ID now
+	id, err := replaceVars(d, config, "{{service_id}}")
+	if err != nil {
+		return fmt.Errorf("Error constructing id: %s", err)
+	}
+	d.SetId(id)
+
+	op := &appengine.Operation{}
+	err = Convert(res, op)
+	if err != nil {
+		return err
+	}
+
+	waitErr := appEngineOperationWaitTime(
+		config.clientAppEngine, op, project, "Creating ServiceSplitTraffic",
+		int(d.Timeout(schema.TimeoutCreate).Minutes()))
+
+	if waitErr != nil {
+		// The resource didn't actually create
+		d.SetId("")
+		return fmt.Errorf("Error waiting to create ServiceSplitTraffic: %s", waitErr)
+	}
+
+	log.Printf("[DEBUG] Finished creating ServiceSplitTraffic %q: %#v", d.Id(), res)
+
+	return resourceAppEngineServiceSplitTrafficRead(d, meta)
+}
+
+func resourceAppEngineServiceSplitTrafficRead(d *schema.ResourceData, meta interface{}) error {
+	config := meta.(*Config)
+
+	url, err := replaceVars(d, config, "{{AppEngineBasePath}}apps/{{project}}/services/{{name}}")
+	if err != nil {
+		return err
+	}
+
+	project, err := getProject(d, config)
+	if err != nil {
+		return err
+	}
+	res, err := sendRequest(config, "GET", project, url, nil)
+	if err != nil {
+		return handleNotFoundError(err, d, fmt.Sprintf("AppEngineServiceSplitTraffic %q", d.Id()))
+	}
+
+	if err := d.Set("project", project); err != nil {
+		return fmt.Errorf("Error reading ServiceSplitTraffic: %s", err)
+	}
+
+	if err := d.Set("name", flattenAppEngineServiceSplitTrafficName(res["name"], d)); err != nil {
+		return fmt.Errorf("Error reading ServiceSplitTraffic: %s", err)
+	}
+
+	return nil
+}
+
+func resourceAppEngineServiceSplitTrafficUpdate(d *schema.ResourceData, meta interface{}) error {
+	config := meta.(*Config)
+
+	project, err := getProject(d, config)
+	if err != nil {
+		return err
+	}
+
+	obj := make(map[string]interface{})
+	nameProp, err := expandAppEngineServiceSplitTrafficName(d.Get("name"), d, config)
+	if err != nil {
+		return err
+	} else if v, ok := d.GetOkExists("name"); !isEmptyValue(reflect.ValueOf(v)) && (ok || !reflect.DeepEqual(v, nameProp)) {
+		obj["name"] = nameProp
+	}
+	idProp, err := expandAppEngineServiceSplitTrafficServiceId(d.Get("service_id"), d, config)
+	if err != nil {
+		return err
+	} else if v, ok := d.GetOkExists("service_id"); !isEmptyValue(reflect.ValueOf(v)) && (ok || !reflect.DeepEqual(v, idProp)) {
+		obj["id"] = idProp
+	}
+	splitProp, err := expandAppEngineServiceSplitTrafficSplit(d.Get("split"), d, config)
+	if err != nil {
+		return err
+	} else if v, ok := d.GetOkExists("split"); !isEmptyValue(reflect.ValueOf(v)) && (ok || !reflect.DeepEqual(v, splitProp)) {
+		obj["split"] = splitProp
+	}
+
+	lockName, err := replaceVars(d, config, "apps/{{project}}/services/{{service_id}}")
+	if err != nil {
+		return err
+	}
+	mutexKV.Lock(lockName)
+	defer mutexKV.Unlock(lockName)
+
+	url, err := replaceVars(d, config, "{{AppEngineBasePath}}apps/{{project}}/services/{{service_id}}?migrateTraffic={{migrate_traffic}}")
+	if err != nil {
+		return err
+	}
+
+	log.Printf("[DEBUG] Updating ServiceSplitTraffic %q: %#v", d.Id(), obj)
+	updateMask := []string{}
+
+	if d.HasChange("name") {
+		updateMask = append(updateMask, "name")
+	}
+
+	if d.HasChange("service_id") {
+		updateMask = append(updateMask, "id")
+	}
+
+	if d.HasChange("split") {
+		updateMask = append(updateMask, "split")
+	}
+	// updateMask is a URL parameter but not present in the schema, so replaceVars
+	// won't set it
+	url, err = addQueryParams(url, map[string]string{"updateMask": strings.Join(updateMask, ",")})
+	if err != nil {
+		return err
+	}
+	res, err := sendRequestWithTimeout(config, "PATCH", project, url, obj, d.Timeout(schema.TimeoutUpdate))
+
+	if err != nil {
+		return fmt.Errorf("Error updating ServiceSplitTraffic %q: %s", d.Id(), err)
+	}
+
+	op := &appengine.Operation{}
+	err = Convert(res, op)
+	if err != nil {
+		return err
+	}
+
+	err = appEngineOperationWaitTime(
+		config.clientAppEngine, op, project, "Updating ServiceSplitTraffic",
+		int(d.Timeout(schema.TimeoutUpdate).Minutes()))
+
+	if err != nil {
+		return err
+	}
+
+	return resourceAppEngineServiceSplitTrafficRead(d, meta)
+}
+
+func resourceAppEngineServiceSplitTrafficDelete(d *schema.ResourceData, meta interface{}) error {
+	log.Printf("[WARNING] AppEngine ServiceSplitTraffic resources"+
+		" cannot be deleted from GCP. The resource %s will be removed from Terraform"+
+		" state, but will still be present on the server.", d.Id())
+	d.SetId("")
+
+	return nil
+}
+
+func resourceAppEngineServiceSplitTrafficImport(d *schema.ResourceData, meta interface{}) ([]*schema.ResourceData, error) {
+	config := meta.(*Config)
+	if err := parseImportId([]string{
+		"apps/(?P<project>[^/]+)/services/(?P<service_id>[^/]+)",
+		"(?P<project>[^/]+)/(?P<service_id>[^/]+)",
+		"(?P<service_id>[^/]+)",
+	}, d, config); err != nil {
+		return nil, err
+	}
+
+	// Replace import id for the resource id
+	id, err := replaceVars(d, config, "{{service_id}}")
+	if err != nil {
+		return nil, fmt.Errorf("Error constructing id: %s", err)
+	}
+	d.SetId(id)
+
+	return []*schema.ResourceData{d}, nil
+}
+
+func flattenAppEngineServiceSplitTrafficName(v interface{}, d *schema.ResourceData) interface{} {
+	return v
+}
+
+func expandAppEngineServiceSplitTrafficName(v interface{}, d TerraformResourceData, config *Config) (interface{}, error) {
+	return v, nil
+}
+
+func expandAppEngineServiceSplitTrafficServiceId(v interface{}, d TerraformResourceData, config *Config) (interface{}, error) {
+	return v, nil
+}
+
+func expandAppEngineServiceSplitTrafficSplit(v interface{}, d TerraformResourceData, config *Config) (interface{}, error) {
+	l := v.([]interface{})
+	if len(l) == 0 || l[0] == nil {
+		return nil, nil
+	}
+	raw := l[0]
+	original := raw.(map[string]interface{})
+	transformed := make(map[string]interface{})
+
+	transformedShardBy, err := expandAppEngineServiceSplitTrafficSplitShardBy(original["shard_by"], d, config)
+	if err != nil {
+		return nil, err
+	} else if val := reflect.ValueOf(transformedShardBy); val.IsValid() && !isEmptyValue(val) {
+		transformed["shardBy"] = transformedShardBy
+	}
+
+	transformedAllocations, err := expandAppEngineServiceSplitTrafficSplitAllocations(original["allocations"], d, config)
+	if err != nil {
+		return nil, err
+	} else if val := reflect.ValueOf(transformedAllocations); val.IsValid() && !isEmptyValue(val) {
+		transformed["allocations"] = transformedAllocations
+	}
+
+	return transformed, nil
+}
+
+func expandAppEngineServiceSplitTrafficSplitShardBy(v interface{}, d TerraformResourceData, config *Config) (interface{}, error) {
+	return v, nil
+}
+
+func expandAppEngineServiceSplitTrafficSplitAllocations(v interface{}, d TerraformResourceData, config *Config) (map[string]string, error) {
+	if v == nil {
+		return map[string]string{}, nil
+	}
+	m := make(map[string]string)
+	for k, val := range v.(map[string]interface{}) {
+		m[k] = val.(string)
+	}
+	return m, nil
+}