mirror of
https://github.com/kairos-io/kairos-agent.git
synced 2025-09-25 13:33:41 +00:00
adce182ea2
Now there is a `install` section in the config that has the fields that previously where in `c3os` but were actually only used during install phase. Also the k3s and c3os config were moved to the provider instead that in the global config.
60 lines
1.5 KiB
Go
60 lines
1.5 KiB
Go
package role
|
|
|
|
import (
|
|
"github.com/c3os-io/c3os/pkg/config"
|
|
|
|
providerConfig "github.com/c3os-io/c3os/internal/provider/config"
|
|
utils "github.com/mudler/edgevpn/pkg/utils"
|
|
|
|
service "github.com/mudler/edgevpn/api/client/service"
|
|
)
|
|
|
|
func contains(slice []string, elem string) bool {
|
|
for _, s := range slice {
|
|
if elem == s {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
func Auto(cc *config.Config, pconfig *providerConfig.Config) Role {
|
|
return func(c *service.RoleConfig) error {
|
|
advertizing, _ := c.Client.AdvertizingNodes()
|
|
actives, _ := c.Client.ActiveNodes()
|
|
|
|
c.Logger.Info("Active nodes:", actives)
|
|
c.Logger.Info("Advertizing nodes:", advertizing)
|
|
|
|
if len(advertizing) < 2 {
|
|
c.Logger.Info("Not enough nodes")
|
|
return nil
|
|
}
|
|
|
|
// first get available nodes
|
|
nodes := advertizing
|
|
shouldBeLeader := utils.Leader(advertizing)
|
|
|
|
lead, _ := c.Client.Get("auto", "leader")
|
|
|
|
// From now on, only the leader keeps processing
|
|
// TODO: Make this more reliable with consensus
|
|
if shouldBeLeader != c.UUID && lead != c.UUID {
|
|
c.Logger.Infof("<%s> not a leader, leader is '%s', sleeping", c.UUID, shouldBeLeader)
|
|
return nil
|
|
}
|
|
|
|
if shouldBeLeader == c.UUID && (lead == "" || !contains(nodes, lead)) {
|
|
c.Client.Set("auto", "leader", c.UUID)
|
|
c.Logger.Info("Announcing ourselves as leader, backing off")
|
|
return nil
|
|
}
|
|
|
|
if lead != c.UUID {
|
|
c.Logger.Info("Backing off, as we are not currently flagged as leader")
|
|
return nil
|
|
}
|
|
|
|
return scheduleRoles(nodes, c, cc, pconfig)
|
|
}
|
|
}
|