| // Copyright (c) HashiCorp, Inc. |
| // SPDX-License-Identifier: MPL-2.0 |
| package cloudfunctions |
| |
| import ( |
| "fmt" |
| "time" |
| |
| "github.com/hashicorp/terraform-provider-google-beta/google-beta/tpgresource" |
| transport_tpg "github.com/hashicorp/terraform-provider-google-beta/google-beta/transport" |
| "google.golang.org/api/cloudfunctions/v1" |
| ) |
| |
| type CloudFunctionsOperationWaiter struct { |
| Service *cloudfunctions.Service |
| tpgresource.CommonOperationWaiter |
| } |
| |
| func (w *CloudFunctionsOperationWaiter) QueryOp() (interface{}, error) { |
| if w == nil { |
| return nil, fmt.Errorf("Cannot query operation, it's unset or nil.") |
| } |
| return w.Service.Operations.Get(w.Op.Name).Do() |
| } |
| |
| func CloudFunctionsOperationWait(config *transport_tpg.Config, op *cloudfunctions.Operation, activity, userAgent string, timeout time.Duration) error { |
| w := &CloudFunctionsOperationWaiter{ |
| Service: config.NewCloudFunctionsClient(userAgent), |
| } |
| if err := w.SetOp(op); err != nil { |
| return err |
| } |
| return tpgresource.OperationWait(w, activity, timeout, config.PollInterval) |
| } |
| |
| func IsCloudFunctionsSourceCodeError(err error) (bool, string) { |
| if operr, ok := err.(*tpgresource.CommonOpError); ok { |
| if operr.Code == 3 && operr.Message == "Failed to retrieve function source code" { |
| return true, fmt.Sprintf("Retry on Function failing to pull code from GCS") |
| } |
| } |
| return false, "" |
| } |
