func populatedCache() *gocache.Cache {
var metrics []map[string]int
var cache *gocache.Cache
var resp *Response
metrics = append(metrics, map[string]int{"okay": 1})
cache = gocache.New(time.Minute, 20*time.Second)
resp = &Response{
Name: "check_test",
Status: 0,
Stdout: []string{"Mocked"},
Metrics: metrics,
}
cache.Set("check_test", resp, gocache.DefaultExpiration)
return cache
}
func sync(kClient *kubernetesclient.Client, metadataClient *metadata.Client, c *cache.Cache) error {
hosts, err := metadataClient.GetHosts()
if err != nil {
log.Errorf("Error reading host list from metadata service: [%v], retrying", err)
return err
}
for _, host := range hosts {
nodeInt, ok := c.Get(host.Hostname)
if !ok {
temp_node, err := getKubeNode(kClient, host.Hostname)
if err != nil {
log.Errorf("Error getting node: [%s] by name from kubernetes, err: [%v]", host.Hostname, err)
// This node might not have been added to kuberentes cluster yet, so skip it
continue
}
c.Set(host.Hostname, temp_node, 0)
nodeInt = temp_node
}
node := nodeInt.(*model.Node)
if node.Metadata.Annotations == nil {
node.Metadata.Annotations = make(map[string]interface{})
}
rancherLabelsMetadataStore := node.Metadata.Annotations
changed := false
//check for new/updated labels
for k, v1 := range host.Labels {
if !isValidLabelValue(v1) {
continue
}
if changed {
break
}
v2, ok := node.Metadata.Labels[k]
if !ok {
// This label doesn't exist
changed = true
} else {
v2String, ok := v2.(string)
if !ok {
changed = true
} else if v1 != v2String {
changed = true
}
}
}
for k := range node.Metadata.Labels {
if changed {
break
}
if _, ok := rancherLabelsMetadataStore[toKMetaLabel(k)]; !ok {
// This is not a rancher managed label
continue
}
if _, ok := host.Labels[k]; !ok {
changed = true
}
}
retryCount := 0
maxRetryCount := 3
for changed {
node, err := getKubeNode(kClient, host.Hostname)
if err != nil {
log.Errorf("Error getting node: [%s] by name from kubernetes: [%v]", host.Hostname, err)
continue
}
c.Set(host.Hostname, node, 0)
if node.Metadata.Annotations == nil {
node.Metadata.Annotations = make(map[string]interface{})
}
rancherLabelsMetadataStore := node.Metadata.Annotations
for k, v1 := range host.Labels {
if !isValidLabelValue(v1) {
log.Infof("skipping invalid label %s=%s", k, v1)
continue
}
node.Metadata.Labels[k] = v1
rancherLabelsMetadataStore[toKMetaLabel(k)] = ""
}
for k := range node.Metadata.Labels {
if _, ok := rancherLabelsMetadataStore[toKMetaLabel(k)]; !ok {
// This is not a rancher managed label
continue
}
if _, ok := host.Labels[k]; !ok {
delete(node.Metadata.Labels, k)
delete(rancherLabelsMetadataStore, toKMetaLabel(k))
}
}
_, err = kClient.Node.ReplaceNode(node)
if err != nil {
log.Errorf("Error updating node [%s] with new host labels, err :[%v]", host.Hostname, err)
if retryCount < maxRetryCount {
retryCount = retryCount + 1
continue
}
}
changed = false
}
}
return nil
}