optimize webhook patchResponse function (#1334)

Signed-off-by: liheng.zms <[email protected]>
openkruise · Jul 18, 2023 · 4e35a1d · 4e35a1d
1 parent 6ec9dee
commit 4e35a1d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pkg/webhook/pod/mutating/pod_create_update_handler.go b/pkg/webhook/pod/mutating/pod_create_update_handler.go
@@ -58,7 +58,7 @@ func (h *PodCreateHandler) Handle(ctx context.Context, req admission.Request) ad
 	if obj.Namespace == "" {
 		obj.Namespace = req.Namespace
 	}
-
+	oriObj := obj.DeepCopy()
 	var changed bool
 
 	if skip := injectPodReadinessGate(req, obj); !skip {
@@ -110,7 +110,11 @@ func (h *PodCreateHandler) Handle(ctx context.Context, req admission.Request) ad
 	if err != nil {
 		return admission.Errored(http.StatusInternalServerError, err)
 	}
-	return admission.PatchResponseFromRaw(req.AdmissionRequest.Object.Raw, marshalled)
+	original, err := json.Marshal(oriObj)
+	if err != nil {
+		return admission.Errored(http.StatusInternalServerError, err)
+	}
+	return admission.PatchResponseFromRaw(original, marshalled)
 }
 
 var _ inject.Client = &PodCreateHandler{}