optimize webhook patchResponse function (openkruise#1334)

ppbits · Jul 18, 2023 · 9e717af · 9e717af
1 parent f762f95
commit 9e717af
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{}