72 lines
1.4 KiB
Go
72 lines
1.4 KiB
Go
|
package agent
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
"gitlab.com/wpetit/goweb/logger"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Agent struct {
|
||
|
|
controllers []Controller
|
||
|
|
interval time.Duration
|
||
|
|
}
|
||
|
|
|
||
|
|
func (a *Agent) Run(ctx context.Context) error {
|
||
|
|
state := NewState()
|
||
|
|
|
||
|
|
logger.Info(ctx, "starting reconciliation ticker", logger.F("interval", a.interval))
|
||
|
|
|
||
|
|
ticker := time.NewTicker(a.interval)
|
||
|
|
defer ticker.Stop()
|
||
|
|
|
||
|
|
for {
|
||
|
|
select {
|
||
|
|
case <-ticker.C:
|
||
|
|
|
||
|
|
logger.Debug(ctx, "state before reconciliation", logger.F("state", state))
|
||
|
|
|
||
|
|
if err := a.Reconcile(ctx, state); err != nil {
|
||
|
|
logger.Error(ctx, "could not reconcile node with state", logger.E(errors.WithStack(err)))
|
||
|
|
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
logger.Debug(ctx, "state after reconciliation", logger.F("state", state))
|
||
|
|
|
||
|
|
case <-ctx.Done():
|
||
|
|
return errors.WithStack(ctx.Err())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (a *Agent) Reconcile(ctx context.Context, state *State) error {
|
||
|
|
for _, ctrl := range a.controllers {
|
||
|
|
ctrlCtx := logger.With(ctx, logger.F("controller", ctrl.Name()))
|
||
|
|
|
||
|
|
logger.Debug(
|
||
|
|
ctrlCtx, "executing controller",
|
||
|
|
logger.F("state", state),
|
||
|
|
)
|
||
|
|
|
||
|
|
if err := ctrl.Reconcile(ctrlCtx, state); err != nil {
|
||
|
|
return errors.WithStack(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func New(funcs ...OptionFunc) *Agent {
|
||
|
|
opt := defaultOption()
|
||
|
|
for _, fn := range funcs {
|
||
|
|
fn(opt)
|
||
|
|
}
|
||
|
|
|
||
|
|
return &Agent{
|
||
|
|
controllers: opt.Controllers,
|
||
|
|
interval: opt.Interval,
|
||
|
|
}
|
||
|
|
}
|