mirror of
https://github.com/siderolabs/talos.git
synced 2025-08-29 10:31:12 +02:00
a0188aff73
This PR introduces dependencies between the services. Now each service has two virtual events associated with it: 'up' (running and healthy) and 'down' (finished or failed). These events are used to establish correct order via conditions abstraction. Service image unpacking was moved into 'pre' stage simplifying `init/main.go`, service images are now closer to the code which runs the service itself. Step 'pre' now runs after 'wait' step, and service dependencies are now mixed into other conditions of 'wait' step on startup. Signed-off-by: Andrey Smirnov <smirnov.andrey@gmail.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(*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()"
|
|
}
|