Function getMemoryCapacityFromNodes

pkg/operator/operator/memory_capacity.go:32–74

(primaryInstances []string)

const _configKeyPrefix = "memory-capacity-"

func getMemoryCapacityFromNodes(primaryInstances []string) (map[string]*kresource.Quantity, error) {
	opts := kmeta.ListOptions{
		LabelSelector: klabels.SelectorFromSet(map[string]string{
			"workload": "true",
		}).String(),
	}
	nodes, err := config.K8s.ListNodes(&opts)
	if err != nil {
		return nil, err
	}

	minMemMap := map[string]*kresource.Quantity{}
	for _, primaryInstance := range primaryInstances {
		minMemMap[primaryInstance] = nil
	}

	for _, node := range nodes {
		isPrimaryInstance := false
		var primaryInstanceType string
		for k, v := range node.Labels {
			if k == "node.kubernetes.io/instance-type" && slices.HasString(primaryInstances, v) {
				isPrimaryInstance = true
				primaryInstanceType = v
				break
			}
		}
		if !isPrimaryInstance {
			continue
		}

		curMem := node.Status.Capacity.Memory()

		if curMem == nil || curMem.IsZero() {
			continue
		}

		if minMemMap[primaryInstanceType] == nil || minMemMap[primaryInstanceType].Cmp(*curMem) > 0 {
			minMemMap[primaryInstanceType] = curMem
		}
	}

	return minMemMap, nil
}

func getMemoryCapacityFromConfigMap() (map[string]*kresource.Quantity, error) {
	configMapData, _, err := config.K8s.GetConfigMapData(_memConfigMapName)