discover_standby_ng.go

/*
Copyright 2021 Flant JSC

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package hooks

import (
	"fmt"
	"regexp"
	"strconv"

	"github.com/flant/addon-operator/pkg/module_manager/go_hook"
	"github.com/flant/addon-operator/sdk"
	"github.com/flant/shell-operator/pkg/kube/object_patch"
	"github.com/flant/shell-operator/pkg/kube_events_manager/types"
	v1 "k8s.io/api/core/v1"
	"k8s.io/apimachinery/pkg/api/resource"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
	"k8s.io/apimachinery/pkg/util/intstr"

	ngv1 "github.com/deckhouse/deckhouse/modules/040-node-manager/hooks/internal/v1"
)

type StandbyNodeGroupInfo struct {
	Name                    string
	MaxPerZone              int
	ZonesCount              int
	Standby                 *intstr.IntOrString
	StandbyNotHeldResources ngv1.Resources
	Taints                  []v1.Taint
}

func standbyNodeGroupFilter(obj *unstructured.Unstructured) (go_hook.FilterResult, error) {
	nodeGroup := new(ngv1.NodeGroup)
	err := sdk.FromUnstructured(obj, nodeGroup)
	if err != nil {
		return nil, err
	}

	maxPerZone := 1
	if nodeGroup.Spec.CloudInstances.MaxPerZone != nil {
		maxPerZone = int(*nodeGroup.Spec.CloudInstances.MaxPerZone)
	}

	var zonesCount int
	if nodeGroup.Spec.CloudInstances.Zones != nil {
		zonesCount = len(nodeGroup.Spec.CloudInstances.Zones)
	}

	taints := nodeGroup.Spec.NodeTemplate.Taints
	if len(taints) == 0 {
		taints = make([]v1.Taint, 0)
	}

	return StandbyNodeGroupInfo{
		Name:                    nodeGroup.GetName(),
		MaxPerZone:              maxPerZone,
		ZonesCount:              zonesCount,
		Standby:                 nodeGroup.Spec.CloudInstances.Standby,
		StandbyNotHeldResources: nodeGroup.Spec.CloudInstances.StandbyHolder.NotHeldResources,
		Taints:                  taints,
	}, nil
}

type StandbyNodeInfo struct {
	Group             string
	AllocatableCPU    *resource.Quantity
	AllocatableMemory *resource.Quantity
	IsReady           bool
	IsUnschedulable   bool
}

func standbyNodeFilter(obj *unstructured.Unstructured) (go_hook.FilterResult, error) {
	node := new(v1.Node)
	err := sdk.FromUnstructured(obj, node)
	if err != nil {
		return nil, err
	}

	isReady := false
	for _, condition := range node.Status.Conditions {
		if condition.Type == v1.NodeReady && condition.Status == v1.ConditionTrue {
			isReady = true
			break
		}
	}

	// .status.allocatable represents all available resources, not the only remaining.
	return StandbyNodeInfo{
		Group:             node.GetLabels()["node.deckhouse.io/group"],
		AllocatableCPU:    node.Status.Allocatable.Cpu(),
		AllocatableMemory: node.Status.Allocatable.Memory(),
		IsReady:           isReady,
		IsUnschedulable:   node.Spec.Unschedulable,
	}, nil
}

type StandbyPodInfo struct {
	Group   string
	IsReady bool
}

func standbyPodFilter(obj *unstructured.Unstructured) (go_hook.FilterResult, error) {
	pod := new(v1.Pod)
	err := sdk.FromUnstructured(obj, pod)
	if err != nil {
		return nil, err
	}

	isReady := false
	for _, condition := range pod.Status.Conditions {
		if condition.Type == v1.PodReady && condition.Status == v1.ConditionTrue {
			isReady = true
			break
		}
	}

	return StandbyPodInfo{
		Group:   pod.GetLabels()["ng"],
		IsReady: isReady,
	}, nil
}

var _ = sdk.RegisterFunc(&go_hook.HookConfig{
	Queue: "/modules/node-manager/discover_standby_ng",
	Kubernetes: []go_hook.KubernetesConfig{
		{
			Name:       "node_groups",
			ApiVersion: "deckhouse.io/v1",
			Kind:       "NodeGroup",
			FilterFunc: standbyNodeGroupFilter,
		},
		{
			Name:       "nodes",
			ApiVersion: "v1",
			Kind:       "Node",
			FilterFunc: standbyNodeFilter,
		},
		{
			Name:       "standby_pods",
			ApiVersion: "v1",
			Kind:       "Pod",
			NamespaceSelector: &types.NamespaceSelector{
				NameSelector: &types.NameSelector{
					MatchNames: []string{"d8-cloud-instance-manager"},
				},
			},
			LabelSelector: &metav1.LabelSelector{
				MatchExpressions: []metav1.LabelSelectorRequirement{
					{
						Key:      "app",
						Operator: metav1.LabelSelectorOpIn,
						Values: []string{
							"standby-holder",
						},
					},
				},
			},
			FilterFunc: standbyPodFilter,
		},
	},
}, discoverStandbyNGHandler)

type StandbyNodeGroupForValues struct {
	Name          string     `json:"name"`
	Standby       int        `json:"standby"`
	ReserveCPU    string     `json:"reserveCPU"`
	ReserveMemory string     `json:"reserveMemory"`
	Taints        []v1.Taint `json:"taints"`
}

func discoverStandbyNGHandler(input *go_hook.HookInput) error {
	standbyNodeGroups := make([]StandbyNodeGroupForValues, 0)

	for _, node := range input.Snapshots["node_groups"] {
		ng := node.(StandbyNodeGroupInfo)

		// Ignore node groups without standby parameter, or having it zero.
		if ng.Standby == nil || ng.Standby.String() == "0" {
			setNodeGroupStandbyStatus(input.PatchCollector, ng.Name, nil)
			continue
		}

		actualStandby := 0
		for _, pod := range input.Snapshots["standby_pods"] {
			standbyPod := pod.(StandbyPodInfo)
			if standbyPod.Group == ng.Name && standbyPod.IsReady {
				actualStandby++
			}
		}
		setNodeGroupStandbyStatus(input.PatchCollector, ng.Name, &actualStandby)

		readyNodesCount := 0
		allocatableCPUList := make([]*resource.Quantity, 0)
		allocatableMemoryList := make([]*resource.Quantity, 0)
		for _, node := range input.Snapshots["nodes"] {
			standbyNode := node.(StandbyNodeInfo)
			if standbyNode.Group != ng.Name {
				continue
			}
			if standbyNode.IsReady && !standbyNode.IsUnschedulable {
				readyNodesCount++
			}
			if standbyNode.AllocatableCPU != nil {
				allocatableCPUList = append(allocatableCPUList, standbyNode.AllocatableCPU)
			}
			if standbyNode.AllocatableMemory != nil {
				allocatableMemoryList = append(allocatableMemoryList, standbyNode.AllocatableMemory)
			}
		}

		if ng.ZonesCount == 0 {
			if zones, ok := input.Values.GetOk("nodeManager.internal.cloudProvider.zones"); ok {
				ng.ZonesCount = len(zones.Array())
			} else {
				ng.ZonesCount = 1
			}
		}
		maxInstances := ng.MaxPerZone * ng.ZonesCount

		desiredStandby := intOrPercent(ng.Standby, maxInstances)
		totalNodesCount := readyNodesCount + desiredStandby - actualStandby
		if totalNodesCount > maxInstances {
			excessNodesCount := totalNodesCount - maxInstances
			desiredStandby -= excessNodesCount
		}

		// Always keep one Pending standby Pod to catch a Node on application scale down.
		if desiredStandby <= 0 {
			desiredStandby = 1
		}

		// Calculate CPU amount.
		standbyRequestCPU, err := calculateStandbyRequestCPU(input, allocatableCPUList, ng)
		if err != nil {
			return err
		}

		// Calculate Mem amount.
		standbyRequestMemory, err := calculateStandbyRequestMemory(input, allocatableMemoryList, ng)
		if err != nil {
			return err
		}

		standbyNodeGroups = append(standbyNodeGroups, StandbyNodeGroupForValues{
			Name:          ng.Name,
			Standby:       desiredStandby,
			ReserveCPU:    standbyRequestCPU,
			ReserveMemory: standbyRequestMemory,
			Taints:        ng.Taints,
		})
	}

	input.Values.Set("nodeManager.internal.standbyNodeGroups", standbyNodeGroups)
	return nil
}

func setNodeGroupStandbyStatus(patcher *object_patch.PatchCollector, nodeGroupName string, standby *int) {
	statusStandbyPatch := map[string]interface{}{
		"status": map[string]interface{}{
			"standby": standby,
		},
	}
	patcher.MergePatch(statusStandbyPatch, "deckhouse.io/v1", "NodeGroup", "", nodeGroupName, object_patch.WithSubresource("/status"))
}

var NumPercentRegex = regexp.MustCompile(`^([0-9]+)%$`)

func intOrPercent(val *intstr.IntOrString, max int) int {
	matches := NumPercentRegex.FindStringSubmatch(val.StrVal)
	if len(matches) > 1 {
		percent, _ := strconv.Atoi(matches[1])
		return int(float64(max) * float64(percent) / 100.0)
	}

	return val.IntValue()
}

func calculateStandbyRequestCPU(input *go_hook.HookInput, allocatableAmounts []*resource.Quantity, ng StandbyNodeGroupInfo) (string, error) {
	// minAllocatableForNow is a zero or the least quantity from allocatableList.
	minAllocatableForNow := resource.NewQuantity(0, resource.DecimalSI)
	if len(allocatableAmounts) > 0 {
		allocatableAmounts[0].DeepCopyInto(minAllocatableForNow)
	}
	for _, amount := range allocatableAmounts {
		if amount.Cmp(*minAllocatableForNow) < 0 {
			amount.DeepCopyInto(minAllocatableForNow)
		}
	}

	// Get reserved CPU for system components on every node from global values.
	reservedOnEveryNode, err := getQuantityFromValue(input, "global.modules.resourcesRequests.everyNode.cpu")

	// Get reserved CPU for system components on standby node from NodeGroup.
	reservedOnStandbyNode, err := resource.ParseQuantity(ng.StandbyNotHeldResources.CPU.String())
	if err != nil {
		return "", fmt.Errorf("nodegroup/%s: standbyNotHeldResoures.CPU '%s' is a malformed quantity: %v", ng.Name, ng.StandbyNotHeldResources.CPU.String(), err)
	}

	// Calculate milliCPUs available for Standby Pod.
	availableMillis := minAllocatableForNow.MilliValue() - reservedOnEveryNode.MilliValue() - reservedOnStandbyNode.MilliValue()

	// Request at least "cpu: 10m".
	if availableMillis < 10 {
		availableMillis = 10
	}
	return fmt.Sprintf("%dm", availableMillis), nil
}

func calculateStandbyRequestMemory(input *go_hook.HookInput, allocatableAmounts []*resource.Quantity, ng StandbyNodeGroupInfo) (string, error) {
	// minAllocatableForNow is a zero or the least quantity from allocatableList.
	minAllocatableForNow := resource.NewQuantity(0, resource.DecimalSI)
	if len(allocatableAmounts) > 0 {
		allocatableAmounts[0].DeepCopyInto(minAllocatableForNow)
	}
	for _, amount := range allocatableAmounts {
		if amount.Cmp(*minAllocatableForNow) < 0 {
			amount.DeepCopyInto(minAllocatableForNow)
		}
	}

	// Get reserved Memory for system components on every node from global values.
	reservedOnEveryNode, err := getQuantityFromValue(input, "global.modules.resourcesRequests.everyNode.memory")

	// Get reserved Memory for system components on standby node from NodeGroup.
	reservedOnStandbyNode, err := resource.ParseQuantity(ng.StandbyNotHeldResources.Memory.String())
	if err != nil {
		return "", fmt.Errorf("nodegroup/%s: standbyNotHeldResoures.Memory '%s' is a malformed quantity: %v", ng.Name, ng.StandbyNotHeldResources.Memory.String(), err)
	}

	// Calculate memory bytes available for Standby Pod and convert to Mi.
	availableBytes := minAllocatableForNow.Value() - reservedOnEveryNode.Value() - reservedOnStandbyNode.Value()
	availableMi := availableBytes / 1024 / 1024
	// Request at least "memory: 10Mi".
	if availableMi < 10 {
		availableMi = 10
	}
	return fmt.Sprintf("%dMi", availableMi), nil
}

func getQuantityFromValue(input *go_hook.HookInput, valuePath string) (*resource.Quantity, error) {
	value, ok := input.Values.GetOk(valuePath)
	if !ok {
		return nil, fmt.Errorf("value '%s' is required", valuePath)
	}
	str := value.String()
	q, err := resource.ParseQuantity(str)
	if err != nil {
		return nil, fmt.Errorf("value '%s' '%s' is a malformed quantity: %v", valuePath, str, err)
	}
	return &q, nil
}