mirror of
https://github.com/siderolabs/talos.git
synced 2025-08-20 22:21:13 +02:00
0081ac5fac
This extracts Docker Talos cluster provisioner as common code which might be shared between `osctl cluster` and integration-test. There should be almost no functional changes. As proof of concept, abstract cluster readiness checks were implemented based on provisioned cluster state. It implements same checks as `basic-integration.sh` in pure Go via Talos/K8s clients. `conditions` package was promoted from machined-internal to `internal/pkg` as it is used to run the checks. Signed-off-by: Andrey Smirnov <smirnov.andrey@gmail.com>
77 lines
1.7 KiB
Go
77 lines
1.7 KiB
Go
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
// Package check provides set of checks to verify cluster readiness.
|
|
package check
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/talos-systems/talos/internal/pkg/conditions"
|
|
"github.com/talos-systems/talos/internal/pkg/provision"
|
|
)
|
|
|
|
const updateInterval = 100 * time.Millisecond
|
|
|
|
// ClusterCheck implements a function which returns condition based on ClusterAccess.
|
|
type ClusterCheck func(provision.ClusterAccess) conditions.Condition
|
|
|
|
// Reporter presents wait progress.
|
|
//
|
|
// It is supposed that reporter drops duplicate messages.
|
|
type Reporter interface {
|
|
Update(check conditions.Condition)
|
|
}
|
|
|
|
// Wait run the checks against the cluster and waits for the full set to succeed.
|
|
//
|
|
// Context ctx might have a timeout set to limit overall wait time.
|
|
// Each check might define its own timeout.
|
|
func Wait(ctx context.Context, cluster provision.ClusterAccess, checks []ClusterCheck, reporter Reporter) error {
|
|
for _, check := range checks {
|
|
select {
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
default:
|
|
}
|
|
|
|
condition := check(cluster)
|
|
|
|
errCh := make(chan error, 1)
|
|
|
|
go func(condition conditions.Condition) {
|
|
errCh <- condition.Wait(ctx)
|
|
}(condition)
|
|
|
|
var err error
|
|
|
|
func() {
|
|
ticker := time.NewTicker(updateInterval)
|
|
defer ticker.Stop()
|
|
|
|
// report initial state
|
|
reporter.Update(condition)
|
|
|
|
// report last state
|
|
defer reporter.Update(condition)
|
|
|
|
for {
|
|
select {
|
|
case err = <-errCh:
|
|
return
|
|
case <-ticker.C:
|
|
reporter.Update(condition)
|
|
}
|
|
}
|
|
}()
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|