Function UpdateMemoryCapacityConfigMap

pkg/operator/operator/memory_capacity.go:104–156

()

}

func UpdateMemoryCapacityConfigMap() (map[string]kresource.Quantity, error) {
	primaryInstances := []string{}

	minMemMap := map[string]kresource.Quantity{}
	for _, ng := range config.ClusterConfig.NodeGroups {
		instanceMetadata := aws.InstanceMetadatas[config.ClusterConfig.Region][ng.InstanceType]
		minMemMap[ng.InstanceType] = instanceMetadata.Memory
		primaryInstances = append(primaryInstances, ng.InstanceType)
	}

	nodeMemCapacityMap, err := getMemoryCapacityFromNodes(primaryInstances)
	if err != nil {
		return nil, err
	}

	previousMinMemMap, err := getMemoryCapacityFromConfigMap()
	if err != nil {
		return nil, err
	}

	configMapData := map[string]string{}
	for _, primaryInstance := range primaryInstances {
		minMem := minMemMap[primaryInstance]

		if nodeMemCapacityMap[primaryInstance] != nil && minMem.Cmp(*nodeMemCapacityMap[primaryInstance]) > 0 {
			minMem = *nodeMemCapacityMap[primaryInstance]
		}

		if previousMinMemMap[primaryInstance] != nil && minMem.Cmp(*previousMinMemMap[primaryInstance]) > 0 {
			minMem = *previousMinMemMap[primaryInstance]
		}

		if previousMinMemMap[primaryInstance] == nil || minMem.Cmp(*previousMinMemMap[primaryInstance]) < 0 {
			configMapData[_configKeyPrefix+primaryInstance] = minMem.String()
		} else {
			configMapData[_configKeyPrefix+primaryInstance] = previousMinMemMap[primaryInstance].String()
		}

		minMemMap[primaryInstance] = minMem
	}

	configMap := k8s.ConfigMap(&k8s.ConfigMapSpec{
		Name: _memConfigMapName,
		Data: configMapData,
	})

	_, err = config.K8s.ApplyConfigMap(configMap)
	if err != nil {
		return nil, err
	}

	return minMemMap, nil
}