2018-05-23 00:45:22 +00:00
|
|
|
package google
|
|
|
|
|
|
|
|
import (
|
2019-01-25 21:03:43 +00:00
|
|
|
"fmt"
|
|
|
|
|
2019-01-28 21:50:41 +00:00
|
|
|
"google.golang.org/api/serviceusage/v1"
|
2018-05-23 00:45:22 +00:00
|
|
|
)
|
|
|
|
|
2018-12-27 01:42:37 +00:00
|
|
|
type ServiceUsageOperationWaiter struct {
|
2019-01-28 21:50:41 +00:00
|
|
|
Service *serviceusage.Service
|
2018-12-27 01:42:37 +00:00
|
|
|
CommonOperationWaiter
|
2018-05-23 00:45:22 +00:00
|
|
|
}
|
|
|
|
|
2018-12-27 01:42:37 +00:00
|
|
|
func (w *ServiceUsageOperationWaiter) 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()
|
2018-05-23 00:45:22 +00:00
|
|
|
}
|
|
|
|
|
2018-12-27 01:42:37 +00:00
|
|
|
func serviceUsageOperationWait(config *Config, op *serviceusage.Operation, activity string) error {
|
2018-05-23 00:45:22 +00:00
|
|
|
return serviceUsageOperationWaitTime(config, op, activity, 10)
|
|
|
|
}
|
|
|
|
|
2018-12-27 01:42:37 +00:00
|
|
|
func serviceUsageOperationWaitTime(config *Config, op *serviceusage.Operation, activity string, timeoutMinutes int) error {
|
|
|
|
w := &ServiceUsageOperationWaiter{
|
2018-05-23 00:45:22 +00:00
|
|
|
Service: config.clientServiceUsage,
|
|
|
|
}
|
2018-12-27 01:42:37 +00:00
|
|
|
if err := w.SetOp(op); err != nil {
|
|
|
|
return err
|
2018-05-23 00:45:22 +00:00
|
|
|
}
|
2018-12-27 01:42:37 +00:00
|
|
|
return OperationWait(w, activity, timeoutMinutes)
|
2018-05-23 00:45:22 +00:00
|
|
|
}
|