From 1352070d647d0bf77fa38ca9e6568452ab54112a Mon Sep 17 00:00:00 2001 From: fengxsong Date: Fri, 16 Sep 2022 04:35:21 +0000 Subject: [PATCH] fix: set container after modification Signed-off-by: fengxsong --- pkg/app/master/inspectors/pod/pod_inspector.go | 1 + pkg/app/master/kubernetes/workload.go | 9 +++++++++ 2 files changed, 10 insertions(+) diff --git a/pkg/app/master/inspectors/pod/pod_inspector.go b/pkg/app/master/inspectors/pod/pod_inspector.go index fab2ca5724..5b7744fa8d 100644 --- a/pkg/app/master/inspectors/pod/pod_inspector.go +++ b/pkg/app/master/inspectors/pod/pod_inspector.go @@ -365,6 +365,7 @@ func (i *Inspector) prepareWorkload() error { targetCont.SecurityContext.Capabilities.Add, "SYS_ADMIN", ) + i.workload.SetContainer(targetCont) return nil } diff --git a/pkg/app/master/kubernetes/workload.go b/pkg/app/master/kubernetes/workload.go index 32af29775c..53e6f7295e 100644 --- a/pkg/app/master/kubernetes/workload.go +++ b/pkg/app/master/kubernetes/workload.go @@ -87,6 +87,15 @@ func (w *Workload) DefaultContainer() *corev1.Container { return nil } +func (w *Workload) SetContainer(ctr *corev1.Container) { + containers := w.Template().Spec.Containers + for i, c := range containers { + if c.Name == ctr.Name { + containers[i] = *ctr + } + } +} + func (w *Workload) TargetContainer() *corev1.Container { if w.targetContainerName == "" { return w.DefaultContainer()