installer/pkg/agent/waitfor.go

package agent

import (
	"context"
	"time"

	"github.com/pkg/errors"
	"github.com/sirupsen/logrus"
	"k8s.io/apimachinery/pkg/util/wait"
)

// WaitForBootstrapComplete Wait for the bootstrap process to complete on
// cluster installations triggered by the agent installer.
func WaitForBootstrapComplete(cluster *Cluster) error {
	timeout := 60 * time.Minute
	waitContext, cancel := context.WithTimeout(cluster.Ctx, timeout)
	defer cancel()

	var lastErrOnExit error
	var lastErrStr string
	wait.Until(func() {
		bootstrap, exitOnErr, err := cluster.IsBootstrapComplete()
		if bootstrap && err == nil {
			logrus.Info("cluster bootstrap is complete")
			cancel()
		}

		if err != nil {
			if exitOnErr {
				lastErrOnExit = err
				cancel()
			} else {
				if err.Error() != lastErrStr {
					logrus.Info(err)
					lastErrStr = err.Error()
				}
			}
		}
	}, 2*time.Second, waitContext.Done())

	waitErr := waitContext.Err()
	if waitErr != nil {
		if errors.Is(waitErr, context.Canceled) && lastErrOnExit != nil {
			return errors.Wrap(lastErrOnExit, "bootstrap process returned error")
		}
		if errors.Is(waitErr, context.DeadlineExceeded) {
			return errors.Wrap(waitErr, "bootstrap process timed out")
		}
	}

	return nil
}