2016-11-23 06:55:40 +00:00
|
|
|
package google
|
|
|
|
|
|
|
|
import (
|
2019-01-25 21:03:43 +00:00
|
|
|
"fmt"
|
|
|
|
|
2018-01-24 21:03:57 +00:00
|
|
|
"google.golang.org/api/googleapi"
|
2016-11-23 06:55:40 +00:00
|
|
|
"google.golang.org/api/servicemanagement/v1"
|
|
|
|
)
|
|
|
|
|
|
|
|
type ServiceManagementOperationWaiter struct {
|
|
|
|
Service *servicemanagement.APIService
|
2018-12-27 01:42:37 +00:00
|
|
|
CommonOperationWaiter
|
2016-11-23 06:55:40 +00:00
|
|
|
}
|
|
|
|
|
2018-12-27 01:42:37 +00:00
|
|
|
func (w *ServiceManagementOperationWaiter) QueryOp() (interface{}, error) {
|
2019-01-25 21:03:43 +00:00
|
|
|
if w == nil {
|
|
|
|
return nil, fmt.Errorf("Cannot query operation, it's unset or nil.")
|
|
|
|
}
|
2018-12-27 01:42:37 +00:00
|
|
|
return w.Service.Operations.Get(w.Op.Name).Do()
|
2016-11-23 06:55:40 +00:00
|
|
|
}
|
|
|
|
|
2018-01-24 21:03:57 +00:00
|
|
|
func serviceManagementOperationWait(config *Config, op *servicemanagement.Operation, activity string) (googleapi.RawMessage, error) {
|
2017-10-11 19:51:09 +00:00
|
|
|
return serviceManagementOperationWaitTime(config, op, activity, 10)
|
2016-11-23 06:55:40 +00:00
|
|
|
}
|
|
|
|
|
2018-12-27 01:42:37 +00:00
|
|
|
func serviceManagementOperationWaitTime(config *Config, op *servicemanagement.Operation, activity string, timeoutMinutes int) (googleapi.RawMessage, error) {
|
2016-11-23 06:55:40 +00:00
|
|
|
w := &ServiceManagementOperationWaiter{
|
|
|
|
Service: config.clientServiceMan,
|
|
|
|
}
|
|
|
|
|
2018-12-27 01:42:37 +00:00
|
|
|
if err := w.SetOp(op); err != nil {
|
|
|
|
return nil, err
|
2016-11-23 06:55:40 +00:00
|
|
|
}
|
|
|
|
|
2018-12-27 01:42:37 +00:00
|
|
|
if err := OperationWait(w, activity, timeoutMinutes); err != nil {
|
|
|
|
return nil, err
|
2016-11-23 06:55:40 +00:00
|
|
|
}
|
2018-12-27 01:42:37 +00:00
|
|
|
return w.Op.Response, nil
|
2016-11-23 06:55:40 +00:00
|
|
|
}
|