mirror of
https://github.com/siderolabs/talos.git
synced 2025-08-27 17:41:17 +02:00
d8249c8779
* refactor(init): Allow kubeadm init on controlplane This shifts the cluster formation from init(bootstrap) and join(control plane) to init(control plane). This makes use of the previously implemented initToken to provide a TTL for cluster initialization to take place and allows us to mostly treat all control plane nodes equal. This also sets up the path for us to handle master upgrades and not be concerned with odd behavior when upgrading the previously defined init node. To facilitate kubeadm init across all control plane nodes, we make use of the initToken to run `kubeadm init phase certs` command to generate any missing certificates once. All other control plane nodes will attempt to sync the necessary certs/files via all defined trustd endpoints and being the startup process. * feat(init): Add service runner context to PreFunc Signed-off-by: Brad Beam <brad.beam@talos-systems.com>
123 lines
2.6 KiB
Go
123 lines
2.6 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 system_test
|
|
|
|
import (
|
|
"context"
|
|
"sync/atomic"
|
|
"time"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/talos-systems/talos/internal/app/init/pkg/system/conditions"
|
|
"github.com/talos-systems/talos/internal/app/init/pkg/system/events"
|
|
"github.com/talos-systems/talos/internal/app/init/pkg/system/health"
|
|
"github.com/talos-systems/talos/internal/app/init/pkg/system/runner"
|
|
"github.com/talos-systems/talos/pkg/userdata"
|
|
)
|
|
|
|
type MockService struct {
|
|
name string
|
|
preError error
|
|
runnerError error
|
|
nilRunner bool
|
|
runner runner.Runner
|
|
condition conditions.Condition
|
|
postError error
|
|
dependencies []string
|
|
}
|
|
|
|
func (m *MockService) ID(*userdata.UserData) string {
|
|
if m.name != "" {
|
|
return m.name
|
|
}
|
|
return "MockRunner"
|
|
}
|
|
func (m *MockService) PreFunc(context.Context, *userdata.UserData) error {
|
|
return m.preError
|
|
}
|
|
|
|
func (m *MockService) Runner(*userdata.UserData) (runner.Runner, error) {
|
|
if m.runner != nil {
|
|
return m.runner, m.runnerError
|
|
}
|
|
|
|
if m.nilRunner {
|
|
return nil, nil
|
|
}
|
|
|
|
return &MockRunner{exitCh: make(chan error)}, m.runnerError
|
|
}
|
|
|
|
func (m *MockService) PostFunc(*userdata.UserData) error {
|
|
return m.postError
|
|
}
|
|
|
|
func (m *MockService) Condition(*userdata.UserData) conditions.Condition {
|
|
return m.condition
|
|
}
|
|
|
|
func (m *MockService) DependsOn(*userdata.UserData) []string {
|
|
return m.dependencies
|
|
}
|
|
|
|
type MockHealthcheckedService struct {
|
|
MockService
|
|
|
|
notHealthy uint32
|
|
}
|
|
|
|
func (m *MockHealthcheckedService) SetHealthy(healthy bool) {
|
|
if healthy {
|
|
atomic.StoreUint32(&m.notHealthy, 0)
|
|
} else {
|
|
atomic.StoreUint32(&m.notHealthy, 1)
|
|
}
|
|
}
|
|
|
|
func (m *MockHealthcheckedService) HealthFunc(*userdata.UserData) health.Check {
|
|
return func(context.Context) error {
|
|
if atomic.LoadUint32(&m.notHealthy) == 0 {
|
|
return nil
|
|
}
|
|
|
|
return errors.New("not healthy")
|
|
}
|
|
}
|
|
|
|
func (m *MockHealthcheckedService) HealthSettings(*userdata.UserData) *health.Settings {
|
|
return &health.Settings{
|
|
InitialDelay: time.Millisecond,
|
|
Timeout: time.Second,
|
|
Period: time.Millisecond,
|
|
}
|
|
}
|
|
|
|
type MockRunner struct {
|
|
exitCh chan error
|
|
}
|
|
|
|
func (m *MockRunner) Open(ctx context.Context) error {
|
|
return nil
|
|
}
|
|
|
|
func (m *MockRunner) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func (m *MockRunner) Run(eventSink events.Recorder) error {
|
|
eventSink(events.StateRunning, "Running")
|
|
return <-m.exitCh
|
|
}
|
|
|
|
func (m *MockRunner) Stop() error {
|
|
close(m.exitCh)
|
|
|
|
return nil
|
|
}
|
|
|
|
func (m *MockRunner) String() string {
|
|
return "MockRunner()"
|
|
}
|