mirror of
https://github.com/siderolabs/talos.git
synced 2025-12-09 11:31:56 +01:00
cedabeddf7
- Replace unsafe resource interface calls with type-safe versions. - Remove unused parameter names. - Minor changes. Signed-off-by: Dmitriy Matrenichev <dmitry.matrenichev@siderolabs.com>
103 lines
2.6 KiB
Go
103 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 v1alpha1
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
|
|
"github.com/cosi-project/runtime/pkg/controller"
|
|
"github.com/cosi-project/runtime/pkg/safe"
|
|
"github.com/cosi-project/runtime/pkg/state"
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/siderolabs/talos/internal/app/machined/pkg/runtime"
|
|
"github.com/siderolabs/talos/pkg/machinery/api/machine"
|
|
"github.com/siderolabs/talos/pkg/machinery/resources/v1alpha1"
|
|
)
|
|
|
|
// ServiceController manages v1alpha1.Service based on services subsystem state.
|
|
type ServiceController struct {
|
|
V1Alpha1Events runtime.Watcher
|
|
}
|
|
|
|
// Name implements controller.Controller interface.
|
|
func (ctrl *ServiceController) Name() string {
|
|
return "v1alpha1.ServiceController"
|
|
}
|
|
|
|
// Inputs implements controller.Controller interface.
|
|
func (ctrl *ServiceController) Inputs() []controller.Input {
|
|
return nil
|
|
}
|
|
|
|
// Outputs implements controller.Controller interface.
|
|
func (ctrl *ServiceController) Outputs() []controller.Output {
|
|
return []controller.Output{
|
|
{
|
|
Type: v1alpha1.ServiceType,
|
|
Kind: controller.OutputExclusive,
|
|
},
|
|
}
|
|
}
|
|
|
|
// Run implements controller.Controller interface.
|
|
//
|
|
//nolint:gocyclo
|
|
func (ctrl *ServiceController) Run(ctx context.Context, r controller.Runtime, logger *zap.Logger) error {
|
|
var wg sync.WaitGroup
|
|
|
|
wg.Add(1)
|
|
|
|
if err := ctrl.V1Alpha1Events.Watch(func(eventCh <-chan runtime.EventInfo) {
|
|
defer wg.Done()
|
|
|
|
for {
|
|
var (
|
|
event runtime.EventInfo
|
|
ok bool
|
|
)
|
|
|
|
select {
|
|
case <-ctx.Done():
|
|
return
|
|
case event, ok = <-eventCh:
|
|
if !ok {
|
|
return
|
|
}
|
|
}
|
|
|
|
if msg, ok := event.Payload.(*machine.ServiceStateEvent); ok {
|
|
service := v1alpha1.NewService(msg.Service)
|
|
|
|
switch msg.Action { //nolint:exhaustive
|
|
case machine.ServiceStateEvent_RUNNING:
|
|
if err := safe.WriterModify(ctx, r, service, func(svc *v1alpha1.Service) error {
|
|
*svc.TypedSpec() = v1alpha1.ServiceSpec{
|
|
Running: true,
|
|
Healthy: msg.GetHealth().GetHealthy(),
|
|
Unknown: msg.GetHealth().GetUnknown(),
|
|
}
|
|
|
|
return nil
|
|
}); err != nil {
|
|
logger.Info("failed creating service resource", zap.String("id", service.Metadata().ID()), zap.Error(err))
|
|
}
|
|
default:
|
|
if err := r.Destroy(ctx, service.Metadata()); err != nil && !state.IsNotFoundError(err) {
|
|
logger.Info("failed destroying service resource", zap.String("id", service.Metadata().ID()), zap.Error(err))
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}, runtime.WithTailEvents(-1)); err != nil {
|
|
return err
|
|
}
|
|
|
|
wg.Wait()
|
|
|
|
return nil
|
|
}
|