k3s/pkg/node/controller.go

111 lines
2.4 KiB
Go
Raw Normal View History

2019-04-16 23:42:23 +00:00
package node
import (
"context"
"strings"
"github.com/pkg/errors"
coreclient "github.com/rancher/wrangler-api/pkg/generated/controllers/core/v1"
2019-04-16 23:42:23 +00:00
"github.com/sirupsen/logrus"
core "k8s.io/api/core/v1"
)
2019-05-09 22:05:51 +00:00
func Register(ctx context.Context, configMap coreclient.ConfigMapController, nodes coreclient.NodeController) error {
2019-04-16 23:42:23 +00:00
h := &handler{
2019-05-09 22:05:51 +00:00
configCache: configMap.Cache(),
configClient: configMap,
2019-04-16 23:42:23 +00:00
}
2019-05-09 22:05:51 +00:00
nodes.OnChange(ctx, "node", h.onChange)
nodes.OnRemove(ctx, "node", h.onRemove)
2019-04-16 23:42:23 +00:00
return nil
}
type handler struct {
2019-05-09 22:05:51 +00:00
configCache coreclient.ConfigMapCache
2019-04-16 23:42:23 +00:00
configClient coreclient.ConfigMapClient
}
2019-05-09 22:05:51 +00:00
func (h *handler) onChange(key string, node *core.Node) (*core.Node, error) {
if node == nil {
return nil, nil
}
2019-04-16 23:42:23 +00:00
return h.updateHosts(node, false)
}
2019-05-09 22:05:51 +00:00
func (h *handler) onRemove(key string, node *core.Node) (*core.Node, error) {
2019-04-16 23:42:23 +00:00
return h.updateHosts(node, true)
}
2019-05-09 22:05:51 +00:00
func (h *handler) updateHosts(node *core.Node, removed bool) (*core.Node, error) {
2019-04-16 23:42:23 +00:00
var (
newHosts string
nodeAddress string
2019-04-22 23:13:16 +00:00
hostsMap map[string]string
2019-04-16 23:42:23 +00:00
)
2019-04-22 23:13:16 +00:00
hostsMap = make(map[string]string)
2019-04-16 23:42:23 +00:00
for _, address := range node.Status.Addresses {
if address.Type == "InternalIP" {
nodeAddress = address.Address
break
}
}
if nodeAddress == "" {
logrus.Errorf("No InternalIP found for node %s", node.Name)
return nil, nil
}
2019-08-06 17:12:40 +00:00
configMapCache, err := h.configCache.Get("kube-system", "coredns")
if err != nil || configMapCache == nil {
2019-04-16 23:42:23 +00:00
logrus.Warn(errors.Wrap(err, "Unable to fetch coredns config map"))
return nil, nil
}
2019-08-06 17:12:40 +00:00
configMap := configMapCache.DeepCopy()
2019-04-16 23:42:23 +00:00
hosts := configMap.Data["NodeHosts"]
for _, line := range strings.Split(hosts, "\n") {
if line == "" {
continue
}
fields := strings.Fields(line)
2019-04-22 23:13:16 +00:00
if len(fields) != 2 {
2019-04-16 23:42:23 +00:00
logrus.Warnf("Unknown format for hosts line [%s]", line)
continue
}
ip := fields[0]
host := fields[1]
2019-04-22 23:13:16 +00:00
if host == node.Name {
2019-04-16 23:42:23 +00:00
if removed {
continue
}
2019-04-22 23:13:16 +00:00
if ip == nodeAddress {
2019-04-16 23:42:23 +00:00
return nil, nil
}
}
2019-04-22 23:13:16 +00:00
hostsMap[host] = ip
2019-04-16 23:42:23 +00:00
}
if !removed {
2019-04-22 23:13:16 +00:00
hostsMap[node.Name] = nodeAddress
2019-04-16 23:42:23 +00:00
}
2019-04-22 23:13:16 +00:00
for host, ip := range hostsMap {
newHosts += ip + " " + host + "\n"
2019-04-16 23:42:23 +00:00
}
configMap.Data["NodeHosts"] = newHosts
if _, err := h.configClient.Update(configMap); err != nil {
return nil, err
}
var actionType string
if removed {
actionType = "Removed"
} else {
actionType = "Updated"
}
2019-04-22 23:13:16 +00:00
logrus.Infof("%s coredns node hosts entry [%s]", actionType, nodeAddress+" "+node.Name)
2019-04-16 23:42:23 +00:00
return nil, nil
}