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
}
Exemple #2
0
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
}