waitForTask.go

func waitForTask(id string) (err error) {
	done := make(chan error)
	timer := time.NewTimer(0)
	timeout := time.After(10 * time.Seconds)

	go func() {
		for {
			select {
			case <-timeout:
				timer.Stop()
				done <- fmt.Errorf("Timeout")
				return
			case <-timer.C:
				taskDone, err := GetTaskStatus(id)
				if err != nil {
					done <- err
					return
				} else if taskDone {
					done <- nil
					return
				}
				timer = time.NewTimer(2 * time.Second)
			}
		}
	}()

	return <-done
}