mirror of
https://github.com/kairos-io/kairos-agent.git
synced 2025-08-28 04:43:44 +00:00
63cd28d1cb
* 🎨 Split off cli into separate binaries This commit splits off the cli into 3 binaries: - agent - cli - provider The provider now is a separate component that can be tested by itself and have its own lifecycle. This paves the way to a ligher c3os variant, HA support and other features that can be provided on runtime. This is working, but still there are low hanging fruit to care about. Fixes #14 * 🤖 Add provider bin to releases * ⚙️ Handle signals * ⚙️ Reduce buildsize footprint * 🎨 Scan for providers also in /system/providers * 🤖 Run goreleaser * 🎨 Refactoring
74 lines
1.8 KiB
Go
74 lines
1.8 KiB
Go
package role
|
|
|
|
import (
|
|
"math/rand"
|
|
"time"
|
|
|
|
"github.com/c3os-io/c3os/pkg/config"
|
|
service "github.com/mudler/edgevpn/api/client/service"
|
|
)
|
|
|
|
// scheduleRoles assigns roles to nodes. Meant to be called only by leaders
|
|
// TODO: HA-Auto
|
|
func scheduleRoles(nodes []string, c *service.RoleConfig, cc *config.Config) error {
|
|
rand.Seed(time.Now().Unix())
|
|
|
|
// Assign roles to nodes
|
|
currentRoles := map[string]string{}
|
|
|
|
existsMaster := false
|
|
unassignedNodes := []string{}
|
|
for _, a := range nodes {
|
|
role, _ := c.Client.Get("role", a)
|
|
currentRoles[a] = role
|
|
if role == "master" {
|
|
existsMaster = true
|
|
} else if role == "" {
|
|
unassignedNodes = append(unassignedNodes, a)
|
|
}
|
|
}
|
|
|
|
c.Logger.Infof("I'm the leader. My UUID is: %s.\n Current assigned roles: %+v", c.UUID, currentRoles)
|
|
c.Logger.Infof("Master already present: %t", existsMaster)
|
|
c.Logger.Infof("Unassigned nodes: %+v", unassignedNodes)
|
|
|
|
if !existsMaster && len(unassignedNodes) > 0 {
|
|
var selected string
|
|
toSelect := unassignedNodes
|
|
|
|
// Avoid to schedule to ourselves if we have a static role
|
|
if cc.C3OS.Role != "" {
|
|
toSelect = []string{}
|
|
for _, u := range unassignedNodes {
|
|
if u != c.UUID {
|
|
toSelect = append(toSelect, u)
|
|
}
|
|
}
|
|
}
|
|
|
|
// select one node without roles to become master
|
|
if len(toSelect) == 1 {
|
|
selected = toSelect[0]
|
|
} else {
|
|
selected = toSelect[rand.Intn(len(toSelect)-1)]
|
|
}
|
|
|
|
c.Client.Set("role", selected, "master")
|
|
c.Logger.Info("-> Set master to", selected)
|
|
currentRoles[selected] = "master"
|
|
// Return here, so next time we get called
|
|
// makes sure master is set.
|
|
return nil
|
|
}
|
|
|
|
// cycle all empty roles and assign worker roles
|
|
for _, uuid := range unassignedNodes {
|
|
c.Client.Set("role", uuid, "worker")
|
|
c.Logger.Info("-> Set worker to", uuid)
|
|
}
|
|
|
|
c.Logger.Info("Done scheduling")
|
|
|
|
return nil
|
|
}
|