2017-10-30 23:41:37 +00:00
|
|
|
package google
|
|
|
|
|
|
|
|
import (
|
2019-01-25 21:03:43 +00:00
|
|
|
"fmt"
|
|
|
|
|
2017-10-30 23:41:37 +00:00
|
|
|
"google.golang.org/api/dataproc/v1"
|
|
|
|
)
|
|
|
|
|
|
|
|
type DataprocClusterOperationWaiter struct {
|
|
|
|
Service *dataproc.Service
|
2018-12-27 01:42:37 +00:00
|
|
|
CommonOperationWaiter
|
2017-10-30 23:41:37 +00:00
|
|
|
}
|
|
|
|
|
2018-12-27 01:42:37 +00:00
|
|
|
func (w *DataprocClusterOperationWaiter) 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.Projects.Regions.Operations.Get(w.Op.Name).Do()
|
2017-10-30 23:41:37 +00:00
|
|
|
}
|
|
|
|
|
2018-12-27 01:42:37 +00:00
|
|
|
func dataprocClusterOperationWait(config *Config, op *dataproc.Operation, activity string, timeoutMinutes int) error {
|
2017-10-30 23:41:37 +00:00
|
|
|
w := &DataprocClusterOperationWaiter{
|
|
|
|
Service: config.clientDataproc,
|
|
|
|
}
|
2018-12-27 01:42:37 +00:00
|
|
|
if err := w.SetOp(op); err != nil {
|
|
|
|
return err
|
2017-10-30 23:41:37 +00:00
|
|
|
}
|
2018-12-27 01:42:37 +00:00
|
|
|
return OperationWait(w, activity, timeoutMinutes)
|
2017-10-30 23:41:37 +00:00
|
|
|
}
|