package google import ( "bytes" "fmt" "log" "time" "github.com/hashicorp/terraform/helper/resource" "google.golang.org/api/sqladmin/v1beta4" ) type SqlAdminOperationWaiter struct { Service *sqladmin.Service Op *sqladmin.Operation Project string } func (w *SqlAdminOperationWaiter) RefreshFunc() resource.StateRefreshFunc { return func() (interface{}, string, error) { var op *sqladmin.Operation var err error log.Printf("[DEBUG] self_link: %s", w.Op.SelfLink) op, err = w.Service.Operations.Get(w.Project, w.Op.Name).Do() if err != nil { return nil, "", err } log.Printf("[DEBUG] Got %q when asking for operation %q", op.Status, w.Op.Name) return op, op.Status, nil } } func (w *SqlAdminOperationWaiter) Conf() *resource.StateChangeConf { return &resource.StateChangeConf{ Pending: []string{"PENDING", "RUNNING"}, Target: "DONE", Refresh: w.RefreshFunc(), } } // SqlAdminOperationError wraps sqladmin.OperationError and implements the // error interface so it can be returned. type SqlAdminOperationError sqladmin.OperationErrors func (e SqlAdminOperationError) Error() string { var buf bytes.Buffer for _, err := range e.Errors { buf.WriteString(err.Message + "\n") } return buf.String() } func sqladminOperationWait(config *Config, op *sqladmin.Operation, activity string) error { w := &SqlAdminOperationWaiter{ Service: config.clientSqlAdmin, Op: op, Project: config.Project, } state := w.Conf() state.Timeout = 5 * time.Minute state.MinTimeout = 2 * time.Second opRaw, err := state.WaitForState() if err != nil { return fmt.Errorf("Error waiting for %s: %s", activity, err) } op = opRaw.(*sqladmin.Operation) if op.Error != nil { return SqlAdminOperationError(*op.Error) } return nil }