-
Notifications
You must be signed in to change notification settings - Fork 262
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(vertexai): Add IAM resources for Vertex AI Endpoint (#9454) (#6657)
* feat(vertexai): iam resources for vertex ai prediction * fix: update the import format * refactor: add two spaces for docs * refactor: remove the suffix 'generated' from the hand-written tests [upstream:6b15a46443be95916fd32b81671f0bea3b5cb0cf] Signed-off-by: Modular Magician <[email protected]>
- Loading branch information
1 parent
8f18236
commit 3d43ea6
Showing
6 changed files
with
832 additions
and
2 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,3 @@ | ||
| ```release-note:new-resource | ||
| vertexai: Added IAM resources for google_vertex_ai_endpoint (beta) | ||
| ``` |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
245 changes: 245 additions & 0 deletions
245
google-beta/services/vertexai/iam_vertex_ai_endpoint.go
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,245 @@ | ||
| // Copyright (c) HashiCorp, Inc. | ||
| // SPDX-License-Identifier: MPL-2.0 | ||
|
|
||
| // ---------------------------------------------------------------------------- | ||
| // | ||
| // *** AUTO GENERATED CODE *** Type: MMv1 *** | ||
| // | ||
| // ---------------------------------------------------------------------------- | ||
| // | ||
| // 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 vertexai | ||
|
|
||
| import ( | ||
| "fmt" | ||
|
|
||
| "github.com/hashicorp/errwrap" | ||
| "github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema" | ||
| "google.golang.org/api/cloudresourcemanager/v1" | ||
|
|
||
| "github.com/hashicorp/terraform-provider-google-beta/google-beta/tpgiamresource" | ||
| "github.com/hashicorp/terraform-provider-google-beta/google-beta/tpgresource" | ||
| transport_tpg "github.com/hashicorp/terraform-provider-google-beta/google-beta/transport" | ||
| ) | ||
|
|
||
| var VertexAIEndpointIamSchema = map[string]*schema.Schema{ | ||
| "project": { | ||
| Type: schema.TypeString, | ||
| Computed: true, | ||
| Optional: true, | ||
| ForceNew: true, | ||
| }, | ||
| "location": { | ||
| Type: schema.TypeString, | ||
| Computed: true, | ||
| Optional: true, | ||
| ForceNew: true, | ||
| }, | ||
| "endpoint": { | ||
| Type: schema.TypeString, | ||
| Required: true, | ||
| ForceNew: true, | ||
| DiffSuppressFunc: tpgresource.CompareSelfLinkOrResourceName, | ||
| }, | ||
| } | ||
|
|
||
| type VertexAIEndpointIamUpdater struct { | ||
| project string | ||
| location string | ||
| endpoint string | ||
| d tpgresource.TerraformResourceData | ||
| Config *transport_tpg.Config | ||
| } | ||
|
|
||
| func VertexAIEndpointIamUpdaterProducer(d tpgresource.TerraformResourceData, config *transport_tpg.Config) (tpgiamresource.ResourceIamUpdater, error) { | ||
| values := make(map[string]string) | ||
|
|
||
| project, _ := tpgresource.GetProject(d, config) | ||
| if project != "" { | ||
| if err := d.Set("project", project); err != nil { | ||
| return nil, fmt.Errorf("Error setting project: %s", err) | ||
| } | ||
| } | ||
| values["project"] = project | ||
| location, _ := tpgresource.GetLocation(d, config) | ||
| if location != "" { | ||
| if err := d.Set("location", location); err != nil { | ||
| return nil, fmt.Errorf("Error setting location: %s", err) | ||
| } | ||
| } | ||
| values["location"] = location | ||
| if v, ok := d.GetOk("endpoint"); ok { | ||
| values["endpoint"] = v.(string) | ||
| } | ||
|
|
||
| // We may have gotten either a long or short name, so attempt to parse long name if possible | ||
| m, err := tpgresource.GetImportIdQualifiers([]string{"projects/(?P<project>[^/]+)/locations/(?P<location>[^/]+)/endpoints/(?P<endpoint>[^/]+)", "(?P<project>[^/]+)/(?P<location>[^/]+)/(?P<endpoint>[^/]+)", "(?P<location>[^/]+)/(?P<endpoint>[^/]+)", "(?P<endpoint>[^/]+)"}, d, config, d.Get("endpoint").(string)) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| for k, v := range m { | ||
| values[k] = v | ||
| } | ||
|
|
||
| u := &VertexAIEndpointIamUpdater{ | ||
| project: values["project"], | ||
| location: values["location"], | ||
| endpoint: values["endpoint"], | ||
| d: d, | ||
| Config: config, | ||
| } | ||
|
|
||
| if err := d.Set("project", u.project); err != nil { | ||
| return nil, fmt.Errorf("Error setting project: %s", err) | ||
| } | ||
| if err := d.Set("location", u.location); err != nil { | ||
| return nil, fmt.Errorf("Error setting location: %s", err) | ||
| } | ||
| if err := d.Set("endpoint", u.GetResourceId()); err != nil { | ||
| return nil, fmt.Errorf("Error setting endpoint: %s", err) | ||
| } | ||
|
|
||
| return u, nil | ||
| } | ||
|
|
||
| func VertexAIEndpointIdParseFunc(d *schema.ResourceData, config *transport_tpg.Config) error { | ||
| values := make(map[string]string) | ||
|
|
||
| project, _ := tpgresource.GetProject(d, config) | ||
| if project != "" { | ||
| values["project"] = project | ||
| } | ||
|
|
||
| location, _ := tpgresource.GetLocation(d, config) | ||
| if location != "" { | ||
| values["location"] = location | ||
| } | ||
|
|
||
| m, err := tpgresource.GetImportIdQualifiers([]string{"projects/(?P<project>[^/]+)/locations/(?P<location>[^/]+)/endpoints/(?P<endpoint>[^/]+)", "(?P<project>[^/]+)/(?P<location>[^/]+)/(?P<endpoint>[^/]+)", "(?P<location>[^/]+)/(?P<endpoint>[^/]+)", "(?P<endpoint>[^/]+)"}, d, config, d.Id()) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| for k, v := range m { | ||
| values[k] = v | ||
| } | ||
|
|
||
| u := &VertexAIEndpointIamUpdater{ | ||
| project: values["project"], | ||
| location: values["location"], | ||
| endpoint: values["endpoint"], | ||
| d: d, | ||
| Config: config, | ||
| } | ||
| if err := d.Set("endpoint", u.GetResourceId()); err != nil { | ||
| return fmt.Errorf("Error setting endpoint: %s", err) | ||
| } | ||
| d.SetId(u.GetResourceId()) | ||
| return nil | ||
| } | ||
|
|
||
| func (u *VertexAIEndpointIamUpdater) GetResourceIamPolicy() (*cloudresourcemanager.Policy, error) { | ||
| url, err := u.qualifyEndpointUrl("getIamPolicy") | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| project, err := tpgresource.GetProject(u.d, u.Config) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| var obj map[string]interface{} | ||
|
|
||
| userAgent, err := tpgresource.GenerateUserAgentString(u.d, u.Config.UserAgent) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| policy, err := transport_tpg.SendRequest(transport_tpg.SendRequestOptions{ | ||
| Config: u.Config, | ||
| Method: "POST", | ||
| Project: project, | ||
| RawURL: url, | ||
| UserAgent: userAgent, | ||
| Body: obj, | ||
| }) | ||
| if err != nil { | ||
| return nil, errwrap.Wrapf(fmt.Sprintf("Error retrieving IAM policy for %s: {{err}}", u.DescribeResource()), err) | ||
| } | ||
|
|
||
| out := &cloudresourcemanager.Policy{} | ||
| err = tpgresource.Convert(policy, out) | ||
| if err != nil { | ||
| return nil, errwrap.Wrapf("Cannot convert a policy to a resource manager policy: {{err}}", err) | ||
| } | ||
|
|
||
| return out, nil | ||
| } | ||
|
|
||
| func (u *VertexAIEndpointIamUpdater) SetResourceIamPolicy(policy *cloudresourcemanager.Policy) error { | ||
| json, err := tpgresource.ConvertToMap(policy) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| obj := make(map[string]interface{}) | ||
| obj["policy"] = json | ||
|
|
||
| url, err := u.qualifyEndpointUrl("setIamPolicy") | ||
| if err != nil { | ||
| return err | ||
| } | ||
| project, err := tpgresource.GetProject(u.d, u.Config) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| userAgent, err := tpgresource.GenerateUserAgentString(u.d, u.Config.UserAgent) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| _, err = transport_tpg.SendRequest(transport_tpg.SendRequestOptions{ | ||
| Config: u.Config, | ||
| Method: "POST", | ||
| Project: project, | ||
| RawURL: url, | ||
| UserAgent: userAgent, | ||
| Body: obj, | ||
| Timeout: u.d.Timeout(schema.TimeoutCreate), | ||
| }) | ||
| if err != nil { | ||
| return errwrap.Wrapf(fmt.Sprintf("Error setting IAM policy for %s: {{err}}", u.DescribeResource()), err) | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func (u *VertexAIEndpointIamUpdater) qualifyEndpointUrl(methodIdentifier string) (string, error) { | ||
| urlTemplate := fmt.Sprintf("{{VertexAIBasePath}}%s:%s", fmt.Sprintf("projects/%s/locations/%s/endpoints/%s", u.project, u.location, u.endpoint), methodIdentifier) | ||
| url, err := tpgresource.ReplaceVars(u.d, u.Config, urlTemplate) | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
| return url, nil | ||
| } | ||
|
|
||
| func (u *VertexAIEndpointIamUpdater) GetResourceId() string { | ||
| return fmt.Sprintf("projects/%s/locations/%s/endpoints/%s", u.project, u.location, u.endpoint) | ||
| } | ||
|
|
||
| func (u *VertexAIEndpointIamUpdater) GetMutexKey() string { | ||
| return fmt.Sprintf("iam-vertexai-endpoint-%s", u.GetResourceId()) | ||
| } | ||
|
|
||
| func (u *VertexAIEndpointIamUpdater) DescribeResource() string { | ||
| return fmt.Sprintf("vertexai endpoint %q", u.GetResourceId()) | ||
| } |
Oops, something went wrong.