1 files changed, 103 insertions, 0 deletions
diff --git a/cloud/pkg/controllermanager/edgeapplication/overridemanager/resourcesoverrider.go b/cloud/pkg/controllermanager/edgeapplication/overridemanager/resourcesoverrider.go
new file mode 100644
index 000000000..05c1562cf
--- /dev/null
+++ b/cloud/pkg/controllermanager/edgeapplication/overridemanager/resourcesoverrider.go
@@ -0,0 +1,103 @@
+/*
+Copyright 2024 The KubeEdge Authors.
+
+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 overridemanager
+
+import (
+	"fmt"
+	"strings"
+
+	corev1 "k8s.io/api/core/v1"
+	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
+	"k8s.io/klog/v2"
+
+	"github.com/kubeedge/kubeedge/pkg/apis/apps/v1alpha1"
+)
+
+type ResourcesOverrider struct{}
+
+func (o *ResourcesOverrider) ApplyOverrides(rawObj *unstructured.Unstructured, overriders OverriderInfo) error {
+	resourcesOverriders := overriders.Overriders.ResourcesOverrides
+	for index := range resourcesOverriders {
+		patches, err := buildResourcesPatches(rawObj, &resourcesOverriders[index])
+		if err != nil {
+			return err
+		}
+
+		klog.V(4).Infof("Parsed JSON patches by ResourcesOverrider(%+v): %+v", resourcesOverriders[index], patches)
+		if err = applyJSONPatch(rawObj, patches); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+// buildResourcesPatches build JSON patches for the resource object according to override declaration.
+func buildResourcesPatches(rawObj *unstructured.Unstructured, resourcesOverrider *v1alpha1.ResourcesOverrider) ([]overrideOption, error) {
+	switch rawObj.GetKind() {
+	case PodKind:
+		return buildResourcesPatchesWithPath("spec/containers", rawObj, resourcesOverrider)
+	case ReplicaSetKind:
+		fallthrough
+	case DeploymentKind:
+		fallthrough
+	case DaemonSetKind:
+		fallthrough
+	case JobKind:
+		fallthrough
+	case StatefulSetKind:
+		return buildResourcesPatchesWithPath("spec/template/spec/containers", rawObj, resourcesOverrider)
+	}
+	return nil, nil
+}
+
+func buildResourcesPatchesWithPath(specContainersPath string, rawObj *unstructured.Unstructured, resourcesOverrider *v1alpha1.ResourcesOverrider) ([]overrideOption, error) {
+	patches := make([]overrideOption, 0)
+	containers, ok, err := unstructured.NestedSlice(rawObj.Object, strings.Split(specContainersPath, pathSplit)...)
+	if err != nil {
+		return nil, fmt.Errorf("failed to retrieve path(%s) from rawObj, error: %v", specContainersPath, err)
+	}
+	if !ok || len(containers) == 0 {
+		return nil, nil
+	}
+	klog.V(4).Infof("buildResourcesPatchesWithPath containers info (%+v)", containers)
+	for index, container := range containers {
+		if container.(map[string]interface{})["name"] == resourcesOverrider.ContainerName {
+			resourcesPath := fmt.Sprintf("/%s/%d/resources", specContainersPath, index)
+			resourcesValue := resourcesOverrider.Value
+			var patch overrideOption
+			// Overridden with the current value regardless of whether the template value is empty
+			patch, _ = acquireOverride(resourcesPath, resourcesValue)
+
+			klog.V(4).Infof("[buildResourcesPatchesWithPath] containers patch info (%+v)", patch)
+			patches = append(patches, patch)
+		}
+	}
+	return patches, nil
+}
+
+// acquireOverrideOption for adding resources override.
+func acquireOverride(resourcesPath string, resourcesValue corev1.ResourceRequirements) (overrideOption, error) {
+	if !strings.HasPrefix(resourcesPath, pathSplit) {
+		return overrideOption{}, fmt.Errorf("internal error: [acquireOverride] resourcesPath should start with / character")
+	}
+	return overrideOption{
+		Op:    string(v1alpha1.OverriderOpReplace),
+		Path:  resourcesPath,
+		Value: resourcesValue,
+	}, nil
+}