cloud/pkg/controllermanager/edgeapplication/overridemanager/replicasoverrider.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

package overridemanager

import (
	"fmt"

	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"

	apppsv1alpha1 "github.com/kubeedge/api/apis/apps/v1alpha1"
)

const (
	deploymentReplicasPath = "/spec/replicas"
)

type ReplicasOverrider struct{}

func (o *ReplicasOverrider) ApplyOverrides(rawObj *unstructured.Unstructured, overriders OverriderInfo) error {
	switch rawObj.GetKind() {
	case DeploymentKind:
		if overriders.Overriders.Replicas == nil {
			return nil
		}
		patch := overrideOption{
			Op:    string(apppsv1alpha1.OverriderOpReplace),
			Path:  deploymentReplicasPath,
			Value: *overriders.Overriders.Replicas,
		}
		if err := applyJSONPatch(rawObj, []overrideOption{patch}); err != nil {
			return fmt.Errorf("failed to apply replicas override on deployment %s/%s, %v",
				rawObj.GetNamespace(), rawObj.GetName(), err)
		}
		return nil

	default:
		return fmt.Errorf("failed to apply replicas override on obj %s/%s, gvk: %s unsupported",
			rawObj.GetNamespace(), rawObj.GetName(), rawObj.GroupVersionKind())
	}
}