func updateCommand(cmd Command) {
log.Debugln("Updating ", cmd.Target)
switch cmd.Target {
case "all":
cluster := cmd.Object.(string)
updateAll(cluster)
case "agent":
cluster := cmd.Object.(string)
updateAgent(cluster)
case "services":
cluster := cmd.Object.(string)
updateServices(cluster)
case "policy":
cluster := cmd.Object.(string)
updatePolicy(cluster)
case "node-base-services":
data := cmd.Object.([]interface{})
upd := []string{}
for _, item := range data {
upd = append(upd, item.(string))
}
constraints := cfg.GetNodeConstraints()
constraints.BaseServices = upd
cfg.WriteNodeConstraints(cfg.GetNodeConfig().Remote, *constraints)
case "node-cpumin":
upd := cmd.Object.(float64)
constraints := cfg.GetNodeConstraints()
constraints.CpuMin = upd
cfg.WriteNodeConstraints(cfg.GetNodeConfig().Remote, *constraints)
case "node-cpumax":
upd := cmd.Object.(float64)
constraints := cfg.GetNodeConstraints()
constraints.CpuMax = upd
cfg.WriteNodeConstraints(cfg.GetNodeConfig().Remote, *constraints)
case "service-constraints":
name := cmd.Object.(string)
srv, _ := service.GetServiceByName(name)
upd := cfg.ReadService(srv.Remote)
srv.Constraints = upd.Constraints
default:
log.WithField("target", cmd.Target).Errorln("Unrecognized target for command update")
}
}
func updateNodeFolder(ttl int) error {
opt := discovery.Options{
"TTL": time.Second * time.Duration(ttl),
"Dir": true,
"PrevExist": true,
}
remote := cfg.GetNodeConfig().Remote
err := discovery.Set(remote, "", opt)
cfg.WriteNodeConfig(remote, cfg.GetNode().Configuration)
cfg.WriteNodeConstraints(remote, cfg.GetNode().Constraints)
cfg.WriteNodeResources(remote, cfg.GetNode().Resources)
cfg.WriteNodeActive(remote, cfg.GetNode().Active)
if err != nil {
log.WithField("err", err).Errorln("Error updating node folder")
return err
}
return nil
}
func createNodeFolder() error {
var err error
opt := discovery.Options{
"TTL": time.Second * time.Duration(c_TTL+1),
"Dir": true,
}
config := cfg.GetNodeConfig()
config.Cluster = myCluster.Name
remote := c_GRU_REMOTE + myCluster.Name + "/" + c_NODES_REMOTE + config.Name
config.Remote = remote
err = discovery.Set(remote, "", opt)
cfg.WriteNodeConfig(remote, cfg.GetNode().Configuration)
cfg.WriteNodeConstraints(remote, cfg.GetNode().Constraints)
cfg.WriteNodeResources(remote, cfg.GetNode().Resources)
cfg.WriteNodeActive(remote, cfg.GetNode().Active)
if err != nil {
log.WithField("err", err).Errorln("Error creating node folder")
return err
}
return nil
}
